Add habit duplicate checking
This commit is contained in:
parent
52fa6474ad
commit
8d4c32a652
2 changed files with 41 additions and 16 deletions
|
@ -1,32 +1,56 @@
|
|||
def migrate(db, trackers):
|
||||
c = db.cursor()
|
||||
tracker = trackers["cigarette"]
|
||||
habit_dict = {
|
||||
"archived": 0,
|
||||
"color": 11,
|
||||
"name": tracker["label"],
|
||||
"description": "smoking",
|
||||
"freq_den": 1,
|
||||
"freq_num": 1,
|
||||
"highlight": 0,
|
||||
"position": 4,
|
||||
"question": "Did you smoke today punk?",
|
||||
"uuid": tracker["id"],
|
||||
}
|
||||
add_to_database(c, habit_dict)
|
||||
habits = trackers_to_habits(trackers)
|
||||
for habit in habits:
|
||||
existing_habit = check_habit_duplicate(c, habit)
|
||||
if not existing_habit:
|
||||
add_to_database(c, habit)
|
||||
else:
|
||||
print(f"Found duplicate Habit: {existing_habit} - skipping.")
|
||||
|
||||
# DEBUGGING
|
||||
c.execute("""select * from Habits""")
|
||||
print(c.fetchall())
|
||||
|
||||
|
||||
def trackers_to_habits(trackers):
|
||||
habits = []
|
||||
# for each list entry tracker dict transform to habit dict
|
||||
for tracker_name in trackers.keys():
|
||||
t = trackers[tracker_name]
|
||||
habits.append(
|
||||
{
|
||||
"archived": 0 if t["hidden"] == False else 1,
|
||||
"color": 11,
|
||||
"description": t["label"],
|
||||
"freq_den": 1,
|
||||
"freq_num": 1,
|
||||
"highlight": 0,
|
||||
"name": f"{t['emoji']} {t['label']}",
|
||||
"question": "",
|
||||
"unit": "" if t["uom"] == "num" else t["uom"],
|
||||
"uuid": t["id"],
|
||||
}
|
||||
# FIXME conditional for ranges (if type = range > habits[-1]['type'] = intforrangetype; habits[-1]['max'] =)
|
||||
)
|
||||
return habits
|
||||
|
||||
|
||||
def check_habit_duplicate(cursor, habit):
|
||||
cursor.execute("select name from Habits where uuid = ?", [habit["uuid"]])
|
||||
name = cursor.fetchone()
|
||||
if name:
|
||||
return name[0]
|
||||
return False
|
||||
|
||||
|
||||
def add_to_database(cursor, habit_dict):
|
||||
placeholder = ", ".join("?" * len(habit_dict))
|
||||
columns = ", ".join(habit_dict.keys())
|
||||
values = list(habit_dict.values())
|
||||
sql = "insert into `{table}` ({columns}) values ({values});".format(
|
||||
table="Habits", columns=columns, values=placeholder
|
||||
)
|
||||
values = list(habit_dict.values())
|
||||
cursor.execute(sql, values)
|
||||
|
||||
|
||||
|
|
|
@ -2,6 +2,8 @@
|
|||
|
||||
import json
|
||||
import sys
|
||||
|
||||
# TODO is pandas necessary? should only be for csv, not built for now
|
||||
import pandas as pd
|
||||
import loop.migration as loop
|
||||
|
||||
|
@ -45,7 +47,6 @@ def verify_continue(data):
|
|||
if __name__ == "__main__":
|
||||
# load nomie json
|
||||
nomie_data = load_file(sys.argv[1])
|
||||
# DISABLED FOR DEBUGGING
|
||||
verify_continue(nomie_data)
|
||||
trackers = nomie_data["trackers"]
|
||||
events = nomie_data["events"]
|
||||
|
|
Loading…
Reference in a new issue