def do_task(task_name): """ "DO" tasks with current date. :param task_name: :return: """ conn = sqlite3.connect(DB_PATH) c = conn.cursor() # Do this instead t = (task_name,) c.execute('SELECT id_task FROM task WHERE name=?', t) result = c.fetchone() if result: task_id = result[0] #print task_id else: c.execute('SELECT id_task FROM task WHERE short_name=?', t) result = c.fetchone() if result: task_id = result[0] #print task_id else: c.execute('SELECT id_task FROM task WHERE id_task=?', t) result = c.fetchone() if result: task_id = result[0] #print task_id else: print "Brak takiego taska" if 'task_id' in locals(): #today = datetime.now().strftime("%Y-%m-%d") today = datetime.now() values = (task_id, today) c.execute("INSERT INTO registry (id_task, date) VALUES (?,?)", values) conn.commit() c.close() html_utils.update_output()
def create_task(task_name, short_name=None): """ Create task with given name and optional short_name :param task_name: :param short_name: :return: """ conn = sqlite3.connect(DB_PATH) c = conn.cursor() c.execute('SELECT id_color FROM colors') result = c.fetchall() id_color = random.choice(result) id_color = id_color[0] values = (task_name, id_color,short_name) try: c.execute("INSERT INTO task (name, color, short_name) VALUES (?,?,?)", values) except ValueError: print "Taki task jest juz w bazie" # http://stackoverflow.com/questions/22488763/sqlite-insert-query-not-working-with-python conn.commit() c.close() html_utils.update_output()
import html_utils import sys import subprocess # # Init # db_utils.create_db() # output = html_utils.json_get_time_all() # # #db_utils.get_tasks_info() # html_utils.json_every_task() # html_utils.html_every_task() # # # Render to image http://stackoverflow.com/questions/2192799/html-to-image-in-javascript-or-python # subprocess.check_call(['phantomjs', 'generate_image.js', 'html/']) html_utils.update_output() # Switch arguments = sys.argv print arguments if len(sys.argv) > 1: flag = arguments[1] print "Flag:" + flag if flag == "clear": try: db_utils.delete_db() db_utils.create_db() except: print "Failed cleaning db"