104 lines
3.2 KiB
Python
104 lines
3.2 KiB
Python
# import sqlite3
|
|
import os
|
|
import argparse
|
|
import json
|
|
import getpass
|
|
import datetime
|
|
import logging
|
|
import sys
|
|
|
|
import action
|
|
import project
|
|
|
|
class Note:
|
|
def __init__(self):
|
|
self.message = ''
|
|
|
|
def handle_note(args, last_action, conn):
|
|
logging.info('> handle note')
|
|
logging.debug('args: ' + str(args))
|
|
|
|
if not args:
|
|
list_note(last_action.project_id, last_action.task_id, last_action.note_id, conn)
|
|
sys.exit(0)
|
|
|
|
# if arg[1] is digit ==> bad parameter
|
|
|
|
if args[0] == '-c':
|
|
if len(args) == 1:
|
|
print('missing note message')
|
|
sys.exit(1)
|
|
|
|
note = Note()
|
|
note.message = project.arg_string(args[1], 'note message')
|
|
create_note(note, last_action.project_id, last_action.task_id, conn)
|
|
|
|
def create_note(note, active_project_id, active_task_id, conn):
|
|
# TODO Don't create note if no project (nothing selected?) ==> foreign key is not a solution: there is no project ID :)
|
|
logging.info('>> Create note')
|
|
|
|
query = """
|
|
INSERT INTO note (project_id, task_id, username, message)
|
|
VALUES (?, ?, ?, ?);
|
|
"""
|
|
|
|
cursor = conn.cursor()
|
|
cursor.execute(query, (active_project_id, active_task_id, getpass.getuser(), note.message))
|
|
|
|
note_id = cursor.lastrowid
|
|
|
|
action.create_action(cursor, project_id=active_project_id, task_id = active_task_id, note_id=note_id)
|
|
|
|
print('created note {}: {} (task {})'.format(note_id, note.message, active_task_id))
|
|
|
|
def edit_note(args, note_id, conn):
|
|
logging.info('>> Edit note')
|
|
|
|
update_args = {}
|
|
if args.status:
|
|
update_args['status'] = args.status
|
|
if args.edit_name:
|
|
update_args['name'] = args.edit_name
|
|
|
|
query = 'UPDATE note SET {} WHERE id = ?'
|
|
query = query.format(', '.join("%s = '%s'" % (k, v) for k, v in update_args.items()))
|
|
logging.debug('update note query: ' + query)
|
|
|
|
cursor = conn.cursor()
|
|
if update_args:
|
|
logging.debug('Do a note update')
|
|
cursor.execute(query, (note_id,))
|
|
|
|
log_args = ', '.join("%s: '%s'" % (k, v) for k, v in update_args.items())
|
|
print('updated note {}: ({})'.format(note_id, log_args))
|
|
else:
|
|
print('updated note {}: set active note')
|
|
|
|
action.create_action(cursor, note_id, message = 'update ' + str(update_args))
|
|
|
|
def delete_note(note_id, conn):
|
|
logging.info('>> Remove note')
|
|
|
|
query = """
|
|
DELETE FROM note
|
|
WHERE id = ?;
|
|
"""
|
|
|
|
cursor = conn.cursor()
|
|
cursor.execute(query, (note_id,))
|
|
if cursor.rowcount != 1:
|
|
logging.error('DELETE FAILED')
|
|
print('deleted note {}: {}'.format(note_id, 'note_name'))
|
|
|
|
def list_note(active_project_id, active_task_id, active_note_id, conn):
|
|
logging.info('>> No arguments')
|
|
query = "SELECT id, username, message FROM note WHERE project_id = ? AND task_id = ?;" #TODO Date & time
|
|
|
|
cursor = conn.cursor()
|
|
cursor.execute(query, (active_project_id, active_task_id))
|
|
for row in cursor.fetchall():
|
|
note_id, username, message = row
|
|
print('{:1} {:2d}: ({:8}) {}'.format('*' if active_note_id == note_id else '', note_id, username, message))
|
|
|
|
# 2: (budd) |open| |normal| test (0 notes)
|
|
# * 3: (budd) |open| |normal| Dec 31, 2018 tet (0 notes) |