podusowski/spartan

import logging
import endoapi.endomondo
from django.db import transaction

from training import models
from . import gpx
from . import gps_workout
from . import activity

@transaction.atomic
def _import_endomondo_workout(user, endomondo_workout):
    workout = models.Workout.objects.create(user=user,
                                            activity_type=activity.TYPE,
                                            started=endomondo_workout.start_time,
                                            finished=endomondo_workout.start_time + endomondo_workout.duration)

    models.EndomondoWorkout.objects.create(workout=workout,
                                           endomondo_id=endomondo_workout.id)

    gpx = models.Gpx.objects.create(workout=workout,
                                    name=endomondo_workout.sport,
                                    distance=endomondo_workout.distance)

    for point in endomondo_workout.points:
        gpx.gpxtrackpoint_set.create(lat=point['lat'],
                                     lon=point['lon'],
                                     hr=point.get('hr', None),
                                     cad=point.get('cad', None),
                                     time=point['time'])


def _endomondo_time_bounds(user):
    workouts = models.Workout.objects.filter(user=user,
                                             endomondoworkout__isnull=False)

    try:
        latest_workout = workouts.latest("started")
        earliest_workout = workouts.earliest("started")

        return latest_workout.started, earliest_workout.started

    except Exception as e:
        logging.warning(str(e))
        return None, None


def synchronize_endomondo(user, max_results=None):
    key = models.AuthKeys.objects.get(user=user, name="endomondo")
    endomondo = endoapi.endomondo.Endomondo(token=key.key)

    newest, oldest = _endomondo_time_bounds(user)

    endomondo_workouts = endomondo.fetch(max_results=max_results, before=oldest, after=newest)

    count = 0
    for endomondo_workout in endomondo_workouts:
        if not gps_workout.exists(user,
                                  endomondo_workout.start_time,
                                  endomondo_workout.start_time + endomondo_workout.duration):
            try:
                _import_endomondo_workout(user, endomondo_workout)
                count += 1
            except Exception as e:
                logging.exception('error during workout import')

    logging.debug('imported {} workouts'.format(count))
    return count


def connect_to_endomondo(user, email, password):
    endomondo = endoapi.endomondo.Endomondo(email=email, password=password)
    token = endomondo.token
    models.AuthKeys.objects.update_or_create(defaults={'key': token}, user=user, name="endomondo")


def endomondo_key(user):
    try:
        return models.AuthKeys.objects.get(user=user, name="endomondo")
    except:
        return None


def disconnect_endomondo(user):
    models.AuthKeys.objects.get(user=user, name="endomondo").delete()


def purge_endomondo_workouts(user):
    models.Workout.objects.filter(user=user, endomondoworkout__isnull=False).delete()