# import sqlite3 import os import argparse import json import getpass import datetime import logging import action def handle_project(args, last_action, conn): logging.info('> handle project') logging.debug('args: ' + str(args)) do_something = False if args.create_name: do_something = True create_project(args, conn) if args.delete_id: do_something = True delete_project(args.delete_id, conn) if args.edit_id: do_something = True edit_project(args, args.edit_id, conn) if not do_something: list_project(last_action.project_id, conn) def create_project(args, conn): # TODO Project is same name is forbidden logging.info('>> Create project') query = """ INSERT INTO project (username, name, status, created_at) VALUES (?, ?, ?, ?); """ status = 'active' if args.status: status = args.status cursor = conn.cursor() cursor.execute(query, (getpass.getuser(), args.create_name, status, datetime.datetime.now(),)) project_id = cursor.lastrowid action_message = '{} (status: {})'.format(args.create_name, status) action.create_action(cursor, project_id=project_id, message=action_message) print('created project {}: {} (status: {})'.format(project_id, args.create_name, status)) def edit_project(args, project_id, conn): logging.info('>> Edit project') update_args = {} if args.status: update_args['status'] = args.status if args.edit_name: update_args['name'] = args.edit_name query = 'UPDATE project SET {} WHERE id = ?' query = query.format(', '.join("%s = '%s'" % (k, v) for k, v in update_args.items())) logging.debug('update project query: ' + query) cursor = conn.cursor() if update_args: logging.debug('Do a project update') cursor.execute(query, (project_id,)) log_args = ', '.join("%s: '%s'" % (k, v) for k, v in update_args.items()) print('updated project {}: ({})'.format(project_id, log_args)) else: print('updated project {}: set active project') action.create_action(cursor, project_id, message = 'update ' + str(update_args)) def delete_project(project_id, conn): logging.info('>> Remove project') query = """ DELETE FROM project WHERE id = ?; """ cursor = conn.cursor() cursor.execute(query, (project_id,)) if cursor.rowcount != 1: logging.error('DELETE FAILED') print('deleted project {}: {}'.format(project_id, 'project_name')) def list_project(active_project_id, conn): logging.info('>> No arguments') query = """ SELECT id, username, name, status, created_at, (SELECT count(*) FROM task WHERE task.project_id = project.id) FROM project; """ cursor = conn.cursor() cursor.execute(query) for row in cursor.fetchall(): logging.debug('Project row: {}'.format(row)) project_id, username, name, status, date, nb_task = row print('{:1} {:2d}: ({:8}) | {} | {} ({} tasks)'.format('*' if active_project_id == project_id else '', project_id, username, status, name, nb_task))