From 4dc654a68c37ed7da0b725b2a75b286aaebfdf08 Mon Sep 17 00:00:00 2001 From: "Maxence G. de Montauzan" Date: Thu, 12 Jul 2018 23:47:14 +0200 Subject: [PATCH] Move main part in a function --- pits.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/pits.py b/pits.py index 5dec705..8188496 100644 --- a/pits.py +++ b/pits.py @@ -85,7 +85,7 @@ def help(args, last_action, conn): print('TODO help') # TODO help -if __name__ == '__main__': +def main(): command = [ 'project', 'task', 'note', 'log', 'info', 'help', 'version', 'init' ] handler = [ project.handle_project, @@ -120,11 +120,14 @@ if __name__ == '__main__': sys.exit(0) conn = create_connection(DB_FILENAME) - last_action = action.read_last_action(conn) + last_action = action.read_current(conn) logging.debug('Last action: {}'.format(last_action)) logging.debug(argv) logging.debug(argv[2:]) handler[candidate](argv[2:], last_action, conn) conn.commit() - conn.close() \ No newline at end of file + conn.close() + +if __name__ == '__main__': + main() \ No newline at end of file