135 lines
3.8 KiB
Python
135 lines
3.8 KiB
Python
# import sqlite3
|
|
import os
|
|
import argparse
|
|
import json
|
|
import getpass
|
|
import datetime
|
|
import logging
|
|
import sys
|
|
|
|
import action
|
|
|
|
class Project:
|
|
name = ''
|
|
status = 'active'
|
|
|
|
|
|
def handle_project(args, last_action, conn):
|
|
logging.info('> handle project')
|
|
logging.debug('args: ' + str(args))
|
|
|
|
if not args:
|
|
list_project(last_action.project_id, conn)
|
|
sys.exit(0)
|
|
|
|
if args[0].isdigit():
|
|
view_project()
|
|
sys.exit(0)
|
|
|
|
# Use sys.exit when no add/update in Database
|
|
|
|
if args[0] == '-c':
|
|
if len(args) == 1:
|
|
print('required project name')
|
|
sys.exit(1)
|
|
|
|
project = Project()
|
|
project.name = arg_string(args[1], 'project name')
|
|
create_project(project, conn)
|
|
elif args[0] == '-e':
|
|
print('Update project')
|
|
elif args[0] == '-d':
|
|
if len(args) > 1:
|
|
delete_project(arg_number(args[1]), conn)
|
|
else:
|
|
delete_project(last_action.project_id, conn)
|
|
|
|
def arg_string(arg, required = ''):
|
|
if required and arg.startswith('-'):
|
|
print('required ', required)
|
|
sys.exit('1')
|
|
else:
|
|
return arg.strip()
|
|
def arg_number(arg, required = ''):
|
|
if not arg.isdigit():
|
|
print('Invalid value') # TODO
|
|
sys.exit('1')
|
|
|
|
return int(arg)
|
|
|
|
def create_project(project, conn):
|
|
# TODO Project is same name is forbidden
|
|
logging.info('>> Create project')
|
|
|
|
query = """
|
|
INSERT INTO project (username, name, status, created_at)
|
|
VALUES (?, ?, ?, ?);
|
|
"""
|
|
|
|
cursor = conn.cursor()
|
|
cursor.execute(query, (getpass.getuser(), project.name, project.status, datetime.datetime.now(),))
|
|
project_id = cursor.lastrowid
|
|
action_message = '{} (status: {})'.format(project.name, project.status)
|
|
action.create_action(cursor, project_id=project_id, message=action_message)
|
|
logging.debug(action_message)
|
|
|
|
print('created project {}: {} (status: {})'.format(project_id, project.name, project.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))
|
|
|
|
def view_project():
|
|
print("TODO View Project")
|
|
#TODO |