41 lines
1.4 KiB
Python
41 lines
1.4 KiB
Python
import sqlite3
|
|
from django.db import transaction
|
|
from lifting.models import Exercise, Set
|
|
|
|
|
|
def _clean_name(name):
|
|
return name.lower()
|
|
|
|
|
|
def import_fitnotes_db(filename, user):
|
|
# exercise names to db ids
|
|
exercises = {}
|
|
for e in Exercise.objects.all():
|
|
for n in e.names:
|
|
exercises[_clean_name(n)] = e.id
|
|
|
|
# build mapping FitNotes exercise id => our exercise id
|
|
exercise_id_mapping = {}
|
|
|
|
conn = sqlite3.connect(filename)
|
|
cur = conn.cursor()
|
|
for fnid, ename in cur.execute('SELECT _id, name FROM exercise WHERE exercise_type_id=0'):
|
|
cleaned = _clean_name(ename)
|
|
# map to an Exercise id or str
|
|
exercise_id_mapping[fnid] = exercises[cleaned] if cleaned in exercises else cleaned
|
|
|
|
with transaction.atomic():
|
|
Set.objects.filter(source='fitnotes').delete()
|
|
for fnid, date, weight_kg, reps in cur.execute(
|
|
'SELECT exercise_id, date, metric_weight, reps FROM training_log'):
|
|
|
|
# create Exercise if it wasn't found and there's a workout using it
|
|
if isinstance(exercise_id_mapping[fnid], str):
|
|
exercise_id_mapping[fnid] = Exercise.objects.create(
|
|
names=[exercise_id_mapping[fnid]]).id
|
|
|
|
exercise_id = exercise_id_mapping[fnid]
|
|
|
|
Set.objects.create(exercise_id=exercise_id, date=date, weight_kg=weight_kg, reps=reps,
|
|
source='fitnotes', user=user)
|