97 lines
3.1 KiB
Python
97 lines
3.1 KiB
Python
import sqlite3
|
|
import os
|
|
import argparse
|
|
import json
|
|
import getpass
|
|
|
|
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)
|
|
|
|
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)
|
|
|
|
print('Inserting initial data')
|
|
conn.executescript("""
|
|
insert into project values(1, 'me', 'test', 'in progress', 'now', 'now')""")
|
|
|
|
parser = argparse.ArgumentParser()
|
|
subparsers = parser.add_subparsers()
|
|
|
|
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)
|
|
|
|
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)
|