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"
