diff options
Diffstat (limited to 'plac/doc/sql_interface.py')
-rw-r--r-- | plac/doc/sql_interface.py | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/plac/doc/sql_interface.py b/plac/doc/sql_interface.py index d045d03..ccedb97 100644 --- a/plac/doc/sql_interface.py +++ b/plac/doc/sql_interface.py @@ -11,8 +11,8 @@ class SqlInterface(object): commands = ['SELECT'] def __init__(self, dsn): self.soup = SqlSoup(dsn) - def SELECT(self, *args): - sql = 'SELECT ' + ' '.join(args) + def SELECT(self, argstring): + sql = 'SELECT ' + argstring for row in self.soup.bind.execute(sql): yield str(row) # the formatting can be much improved @@ -21,5 +21,10 @@ rl_input = plac.ReadlineInput( histfile=os.path.expanduser('~/.sql_interface.history'), case_sensitive=False) +def split_on_first_space(line, commentchar): + return line.strip().split(' ', 1) # ignoring comments + if __name__ == '__main__': - plac.Interpreter(plac.call(SqlInterface)).interact(rl_input) + si = plac.call(SqlInterface) + i = plac.Interpreter(si, split=split_on_first_space) + i.interact(rl_input) |