Refactor repetition creation function
This commit is contained in:
parent
b9c08758f9
commit
db0121966c
1 changed files with 28 additions and 17 deletions
|
@ -5,7 +5,7 @@ import re
|
|||
def migrate(db, habitlist, events):
|
||||
c = db.cursor()
|
||||
habits = simple_habits_list(c, habitlist)
|
||||
repetitions = extract_all_repetitions(habits, events)
|
||||
repetitions = get_all_repetitions(habits, events)
|
||||
for rep in repetitions:
|
||||
add_to_database(
|
||||
c, rep["id"], rep["timestamp"], 2 if "value" not in rep else rep["value"]
|
||||
|
@ -15,7 +15,7 @@ def migrate(db, habitlist, events):
|
|||
LOOP_RANGE_VALUE_MULTIPLIER = 1000
|
||||
|
||||
|
||||
def extract_all_repetitions(habits, events):
|
||||
def get_all_repetitions(habits, events):
|
||||
"""Return list of all repetitions found of habits in events passed in.
|
||||
Parameters:
|
||||
habits (list): Collection of habits, with minimum necessary fields description and id.
|
||||
|
@ -26,25 +26,36 @@ def extract_all_repetitions(habits, events):
|
|||
"""
|
||||
repetitions = []
|
||||
for event in events:
|
||||
text = event["note"]
|
||||
tags = extract_tags(text)
|
||||
for habit in habits:
|
||||
for tag in tags:
|
||||
if habit["description"] in tag[0]:
|
||||
print(
|
||||
f"found repetition: {habit['name']} in {text} at {event['end']}"
|
||||
)
|
||||
repetitions.append({"id": habit["id"], "timestamp": event["end"]})
|
||||
if tag[1]:
|
||||
print(f"repetition contains counter: {tag[0]} ({tag[1]})")
|
||||
if "type" in habit and habit["type"] == 1:
|
||||
repetitions[-1]["value"] = (
|
||||
tag[1] * LOOP_RANGE_VALUE_MULTIPLIER
|
||||
)
|
||||
print(f"added counter to ranged habit.")
|
||||
reps = tags_to_repetitions(habit, extract_tags(event["note"]), event["end"])
|
||||
if reps:
|
||||
repetitions.extend(reps)
|
||||
|
||||
return repetitions
|
||||
|
||||
|
||||
def tags_to_repetitions(habit, tags, timestamp):
|
||||
"""Return a list of all repetitions generated from the tags and habits passed in.
|
||||
Parameters:
|
||||
habits (list): Collection of habits, with minimum necessary fields description and id.
|
||||
tags (list): Collection of tag tuples.
|
||||
Returns:
|
||||
repetitions (list): Collection of habits for which a corresponding tag has
|
||||
been found. If they correspond that means at the timestamp
|
||||
the habit has been checked in, and a repetition is created.
|
||||
Contains fields id, timestamp, value (for ranges).
|
||||
"""
|
||||
reps = []
|
||||
for tag in tags:
|
||||
if habit["description"] in tag[0]:
|
||||
repetition = {"id": habit["id"], "timestamp": timestamp}
|
||||
if tag[1]:
|
||||
if "type" in habit and habit["type"] == 1:
|
||||
repetition["value"] = tag[1] * LOOP_RANGE_VALUE_MULTIPLIER
|
||||
reps.append(repetition)
|
||||
return reps
|
||||
|
||||
|
||||
def extract_tags(text, tagmarker="#"):
|
||||
"""Return lists of tuples of all event tags found in text.
|
||||
Parameters:
|
||||
|
|
Loading…
Reference in a new issue