mirror of
https://github.com/kennethreitz-archive/plac.git
synced 2026-06-05 15:40:17 +00:00
Kenneth Reitz
30 lines
911 B
Python
30 lines
911 B
Python
# dbcli.py
|
|
import plac
|
|
from sqlalchemy.ext.sqlsoup import SqlSoup
|
|
|
|
@plac.annotations(
|
|
db=("Connection string", 'positional', None, SqlSoup),
|
|
header=("Header", 'flag', 'H'),
|
|
sqlcmd=("SQL command", 'option', 'c', str, None, "SQL"),
|
|
delimiter=("Column separator", 'option', 'd'),
|
|
scripts="SQL scripts",
|
|
)
|
|
def main(db, header, sqlcmd, delimiter="|", *scripts):
|
|
"A script to run queries and SQL scripts on a database"
|
|
yield 'Working on %s' % db.bind.url
|
|
|
|
if sqlcmd:
|
|
result = db.bind.execute(sqlcmd)
|
|
if header: # print the header
|
|
yield delimiter.join(result.keys())
|
|
for row in result: # print the rows
|
|
yield delimiter.join(map(str, row))
|
|
|
|
for script in scripts:
|
|
db.bind.execute(file(script).read())
|
|
yield 'executed %s' % script
|
|
|
|
if __name__ == '__main__':
|
|
for output in plac.call(main):
|
|
print(output)
|