Start to handle project

Beginning of configuration

pits.py

@@ -1,50 +1,96 @@
 import sqlite3
 import os
+import argparse
+import json
+import getpass
 
-db_filename = 'pits.sql'
+def info():
+    print('info')
+
+def handle_project(args):
+    print('> handle project')
+    print(args)
+    if args.create_name:
+        # TODO Project is same name is forbidden
+        print('>> Create project')
+
+        query = """
+        INSERT INTO project (username, name, status)
+        VALUES (?, ?, 'active');
+        """
+
+        with sqlite3.connect(db_filename) as conn:
+            cursor = conn.cursor()
+            cursor.execute(query, (getpass.getuser(), args.c,))
+
+            project_id = cursor.lastrowid
+
+            print('created project {}: {} (status: {})'.format(project_id, args.c, 'active'))
+    if args.delete_id:
+        print('>> Remove project')
+
+        query = """
+        DELETE FROM project
+        WHERE id = ?;
+        """
+
+        with sqlite3.connect(db_filename) as conn:
+            cursor = conn.cursor()
+            cursor.execute(query, (args.delete_id,))
+            if cursor.rowcount != 1:
+                print('DELETE FAILED')
+            print('deleted project {}: {}'.format(args.delete_id, 'project_name'))
+    else:
+        print('>> No arguments')
+        query = "SELECT id, username, name, status FROM project;"
+        with sqlite3.connect(db_filename) as conn:
+            cursor = conn.cursor()
+            cursor.execute(query)
+            for row in cursor.fetchall():
+                project_id, username, name, status = row
+                nb_task = 0
+                print('{:2d}: ({:16}) | {} | {} ({} tasks)'.format(project_id, username, status, name, nb_task))
+
+        # desired output
+        #created project 3: toto (status: active)
+
+def handle_note(args):
+    print('>> handle note')
+
+db_filename = 'pits.db'
+schema_filename = 'pit_schema.sql'
 
 db_is_new = not os.path.exists(db_filename)
 
-conn = sqlite3.connect(db_filename)
+with sqlite3.connect(db_filename) as conn:
+    if db_is_new:
+        print('Creating schema')
+        with open(schema_filename, 'rt') as f:
+            schema = f.read()
+        conn.executescript(schema)
 
-if db_is_new:
-    conn.execute('''CREATE TABLE project (
-        id         int,
-        username   text,
-        name       text,
-        status     text,
-        created_at timestring,
-        updatet_at timestring
-    )''')
+        print('Inserting initial data')
+        conn.executescript("""
+        insert into project values(1, 'me', 'test', 'in progress', 'now', 'now')""")
 
-    conn.execute('''CREATE TABLE task (
-        id         int,
-        project_id int,
-        username   text,
-        name       text,
-        status     text,
-        priority   text,
-        date       timestring,
-        time       timestring,
-        created_at timestring,
-        updated_at timestring
-    )''')
+parser = argparse.ArgumentParser()
+subparsers = parser.add_subparsers()
 
-    conn.execute('''CREATE TABLE note (
-        id         int,
-        project_id int,
-        task_id    int,
-        username   text,
-        message    text,
-        created_at timestring,
-        updated_at timestring
-    )''')
+parser_note = subparsers.add_parser('project')#, usage='toto')
+parser_note.add_argument('-c', type=str, dest='create_name', metavar='name', help='name of project')
+parser_note.add_argument('-e', type=int, dest='edit_id', metavar='number', help='Edit a project')
+parser_note.add_argument('-d', type=int, dest='delete_id', nargs='?', const=-1, metavar='number', help='Delete a project')
+parser_note.add_argument('-s', type=str, dest='status', metavar='status')
+parser_note.add_argument('-q', type=int)
+parser_note.set_defaults(func=handle_project)
 
-    conn.execute('''CREATE TABLE action (
-        project_id int,
-        task_id    int,
-        note_id    int,
-        username   text,
-        message    text,
-        created_at timestring
-    )''')
+parser_note = subparsers.add_parser('note')
+parser_note.add_argument('-c', type=str)
+parser_note.add_argument('-e', type=int)
+parser_note.add_argument('-d', type=int)
+parser_note.set_defaults(func=handle_note)
+
+parser.add_argument('--version', action='version', version='%(prog)s 1.0')
+
+args = parser.parse_args()
+args.func(args)