diff --git a/project.py b/project.py index 325e2de..d968e6a 100644 --- a/project.py +++ b/project.py @@ -10,9 +10,13 @@ import sys import action class Project: - name = '' - status = 'active' + def __init__(self, status = ''): + self.id = 0 + self.name = '' + self.status = status + def __str__(self): + return str(self.__dict__) def handle_project(args, last_action, conn): logging.info('> handle project') @@ -33,24 +37,84 @@ def handle_project(args, last_action, conn): print('missing project name') sys.exit(1) - project = Project() - project.name = arg_string(args[1], 'project name') + project = Project('active') + project.name = get_string_arg(args[1:], 'project name') + parse_project_args(project, args[2:]) create_project(project, conn) elif args[0] == '-e': - print('TODO Update project') + if len(args) == 1: + print('nothing to update') + sys.exit(1) + + project = Project() + project.id = last_action.project_id + + if args[1].isdigit(): + if int(args[1]) == project.id: + print('nothing to update') + sys.exit(1) + project.id = int(args[1]) + parse_project_args(project, args[2:]) + else: + parse_project_args(project, args[1:]) + logging.debug('Project: {}'.format(project)) + + edit_project(project, conn) + # TODO To set as active project, us a -a option? + elif args[0] == '-d': if len(args) > 1: delete_project(arg_number(args[1]), conn) else: delete_project(last_action.project_id, conn) + else: + print('Nothing to do... And this is not normalous') + +def parse_project_args(project, args): + if not args: + return + + i = 0 + while i < len(args): + logging.debug(args[i]) + + if args[i] == '-s': + i += 1 + project.status = get_string_arg(args[i:i+1], 'project status') + elif args[i] == '-n': + i += 1 + project.name = get_string_arg(args[i:i+1], 'project name') + else: + print(f'Invalid project option: {args[i]}') + sys.exit(1) + i += 1 + +def get_string_arg(args, required): + """ + Take an array of element, take first element and return it stripped. + If array is empty or None, print an error message with required and exit + in error. + + :param args: an array of arguments. Can be empty or None + :param required: the message to print in no args (or None) + :return: The arg stripped + """ + if not args: + print(f'Missing {required}') + sys.exit(1) + + logging.debug(args) + return args[0].strip() # TODO Move this function def arg_string(arg, required = ''): + # TODO To meld with get_string_args 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 @@ -76,14 +140,14 @@ def create_project(project, conn): print('created project {}: {} (status: {})'.format(project_id, project.name, project.status)) -def edit_project(args, project_id, conn): +def edit_project(project, 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 + if project.status: + update_args['status'] = project.status + if project.name: + update_args['name'] = project.name query = 'UPDATE project SET {} WHERE id = ?' query = query.format(', '.join("%s = '%s'" % (k, v) for k, v in update_args.items())) @@ -92,14 +156,15 @@ def edit_project(args, project_id, conn): cursor = conn.cursor() if update_args: logging.debug('Do a project update') - cursor.execute(query, (project_id,)) + 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)) + print('updated project {}: ({})'.format(project.id, log_args)) else: - print('updated project {}: set active project') + print('updated project {}: set as active project'.format(project.id)) + # TODO If not project ID to set active - action.create_action(cursor, project_id, message = 'update ' + str(update_args)) + action.create_action(cursor, project.id, message = 'update ' + str(update_args)) def delete_project(project_id, conn): logging.info('>> Remove project') @@ -113,12 +178,14 @@ def delete_project(project_id, conn): cursor.execute(query, (project_id,)) if cursor.rowcount != 1: logging.error('DELETE FAILED') - print('deleted project {}: {}'.format(project_id, 'project_name')) + print('could not find current project') + else: + 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 id, username, name, status, (SELECT count(*) FROM task WHERE task.project_id = project.id) FROM project; """ @@ -127,7 +194,7 @@ def list_project(active_project_id, conn): cursor.execute(query) for row in cursor.fetchall(): logging.debug('Project row: {}'.format(row)) - project_id, username, name, status, date, nb_task = row + project_id, username, name, status, nb_task = row print('{:1} {:2d}: ({:8}) | {} | {} ({} tasks)'.format('*' if active_project_id == project_id else '', project_id, username, status, name, nb_task))