Share same connection

pits.py

@@ -31,19 +31,14 @@ def handle_action(args, last_action):
             result = row
             logging.debug(result)
 
-def read_last_action():
-    logging.info('> last_action')
-    # query = 'SELECT * FROM action WHERE id = (SELECT MAX(id) FROM action'
-    # query = 'SELECT project_id, task_id, note_id FROM action'
-    query = 'SELECT project_id, task_id, note_id FROM action ORDER BY created_at DESC LIMIT 1;'
+def create_connection(db_filename):
+    try:
+        conn = sqlite3.connect(db_filename)
+        return conn
+    except Error as e:
+        print(e)
 
-    with sqlite3.connect(db_filename) as conn:
-        cursor = conn.cursor()
-        cursor.execute(query)
-        logging.debug('DEEEEEEEEBU')
-        last_action = cursor.fetchone()
-        logging.debug(last_action)
-    return last_action
+    return None
 
 db_filename = 'pits.db'
 schema_filename = 'pits_schema.sql'
@@ -93,9 +88,11 @@ parser.add_argument('--version', action='version', version='%(prog)s 1.0')
 if __name__ == '__main__':
     args = parser.parse_args()
 
-    last_action = read_last_action()
-    logging.debug('Last action: {}'.format(last_action))
+    conn = create_connection(db_filename)
+    with conn:
+        last_action = action.read_last_action(conn)
+        logging.debug('Last action: {}'.format(last_action))
 
-    args.func(args, last_action)
+        args.func(args, last_action, conn)
 
     print('END')