Make the commands to be ran a list.

master
Simon Quigley 4 years ago
parent 0fbe46ecdc
commit 11a23f817f

@ -29,22 +29,29 @@ def sqlite_run(command, db=":memory:", return_output=False):
""" """
conn = sqlite3.connect(db) conn = sqlite3.connect(db)
c = conn.cursor() c = conn.cursor()
c.execute(command) for cmd in command:
c.execute(cmd)
conn.commit() conn.commit()
# Make sure we return an output if requested # Make sure we return an output if requested
if return_output: try:
rows = c.fetchall() if return_output:
return rows rows = c.fetchall()
return rows
conn.close() except Exception as e:
print(e)
finally:
conn.close()
def main(module): def main(module):
"""Given a specific module, set it up and insert recent values""" """Given a specific module, set it up and insert recent values"""
module = module() module = module()
sqlite_run(module.sqlite_setup()) run = []
sqlite_run(module.sqlite_add()) run.append(module.sqlite_setup())
run.append(module.sqlite_add())
run.append(module.sqlite_time_range(days=10))
print(sqlite_run(run, return_output=True))
if __name__ == "__main__": if __name__ == "__main__":

Loading…
Cancel
Save