diff --git a/rowers/polarstuff.py b/rowers/polarstuff.py index a45842bf..3efd6d3d 100644 --- a/rowers/polarstuff.py +++ b/rowers/polarstuff.py @@ -118,8 +118,11 @@ def make_authorization_url(): return HttpResponseRedirect(url) -def get_polar_workout_list(user): +def get_polar_workouts(user): r = Rower.objects.get(user=user) + + exercise_list = [] + if (r.polartoken == '') or (r.polartoken is None): s = "Token doesn't exist. Need to authorize" return custom_exception_handler(401,s) @@ -131,19 +134,52 @@ def get_polar_workout_list(user): headers = {'Authorization':authorizationstring, 'Accept': 'application/json'} + headers2 = { + 'Authorization':authorizationstring, + } + url = 'https://polaraccesslink.com/v3/users/{userid}/exercise-transactions'.format( userid = r.polaruserid ) -# url = 'https://polaraccesslink.com/v3/users/{userid}/activity-transactions'.format( -# userid = r.polaruserid -# ) - - print url response = requests.post(url, headers=headers) + print url + print r.polartoken + print response.status_code - return response + if response.status_code == 201: + transactionid = response.json()['transaction-id'] + url = 'https://polaraccesslink.com/v3/users/{userid}/exercise-transactions/{transactionid}'.format( + transactionid = transactionid, + userid = r.polaruserid + ) + + response = requests.get(url, headers=headers) + if response.status_code == 200: + exerciseurls = response.json()['exercises'] + for exerciseurl in exerciseurls: + response = requests.get(exerciseurl,headers=headers) + if response.status_code == 200: + exercise_dict = response.json() + tcxuri = exerciseurl+'/tcx' + response = requests.get(tcxuri,headers=headers2) + if response.status_code == 200: + filename = 'media/polarimport{id}.tcx'.format( + id = exercise_dict['id'] + ) + with open(filename,'w') as fop: + fop.write(response.text) + exercise_dict['filename'] = filename + else: + exercise_dict['filename'] = None + + exercise_list.append(exercise_dict) + + # commit transaction + requests.put(url, headers=headers) + + return exercise_list def get_polar_user_info(user,physical=False): r = Rower.objects.get(user=user) @@ -160,6 +196,7 @@ def get_polar_user_info(user,physical=False): 'Accept': 'application/json' } + params = { 'user-id': r.polaruserid } @@ -174,12 +211,78 @@ def get_polar_user_info(user,physical=False): ) - print url - if physical: response = requests.post(url, headers=headers) else: response = requests.get(url, headers=headers) return response + + +def get_polar_workout(user,id,transactionid): + r = Rower.objects.get(user=user) + if (r.polartoken == '') or (r.polartoken is None): + s = "Token doesn't exist. Need to authorize" + return custom_exception_handler(401,s) + elif (timezone.now()>r.polartokenexpirydate): + s = "Token expired. Needs to refresh" + return custom_exception_handler(401,s) + else: + authorizationstring = str('Bearer ' + r.polartoken) + headers = { + 'Authorization':authorizationstring, + 'Accept': 'application/json' + } + + + url = 'https://polaraccesslink.com/v3/users/{userid}/exercise-transactions'.format( + userid = r.polaruserid + ) + + + response = requests.post(url, headers=headers) + + if response.status_code == 201: + transactionid = response.json()['transaction-id'] + url = 'https://polaraccesslink.com/v3/users/{userid}/exercise-transactions/{transactionid}'.format( + transactionid = transactionid, + userid = r.polaruserid + ) + + response = requests.get(url, headers=headers) + if response.status_code == 200: + exerciseurls = response.json()['exercises'] + for exerciseurl in exerciseurls: + response = requests.gedt(exerciseurl,headers=headers) + if response.status_code == 200: + exercise_dict = response.json() + thisid = exercise_dict['id'] + if thisid == id: + url = 'https://polaraccesslink.com/v3/users/{userid}/exercise-transactions/{transactionid}/exercises/{exerciseid}/tcx'.format( + userid = r.polaruserid, + transactionid = transactionid, + exerciseid = id + ) + + print "Get TCX" + print url + response = requests.get(url,headers = headers2) + print response.status_code + + if response.status_code == 200: + print response.text + + result = response.text + # commit transaction + response = requests.put(url,headers=headers) + else: + result = None + + return result + + return None + + + + diff --git a/rowers/templates/polar_list_import.html b/rowers/templates/polar_list_import.html new file mode 100644 index 00000000..0792fd58 --- /dev/null +++ b/rowers/templates/polar_list_import.html @@ -0,0 +1,42 @@ +{% extends "base.html" %} +{% load staticfiles %} +{% load rowerfilters %} + +{% block title %}Workouts{% endblock %} + +{% block content %} +

New Workouts Available on Polar Flow

+ +

Due to a limitation in Polar Flow's API, we can only access new workouts.

+ +{% if workouts %} +
+ + + + + + + + + + + + {% for workout in workouts %} + + + + + + + + + {% endfor %} + +
Import Date Duration Distance Type
+ Import{{ workout|lookup:'starttime' }}{{ workout|lookup:'duration' }} {{ workout|lookup:'distance' }} m{{ workout|lookup:'type' }}
+
+{% else %} +

No new workouts found

+{% endif %} +{% endblock %} diff --git a/rowers/urls.py b/rowers/urls.py index 5e29e8b5..2550da97 100644 --- a/rowers/urls.py +++ b/rowers/urls.py @@ -328,6 +328,8 @@ urlpatterns = [ url(r'^workout/sporttracksimport/$',views.workout_sporttracksimport_view), url(r'^workout/sporttracksimport/(?P\d+)/$',views.workout_getsporttracksworkout_view), url(r'^workout/sporttracksimport/all/$',views.workout_getsporttracksworkout_all), + url(r'^workout/polarimport/$',views.workout_polarimport_view), + url(r'^workout/polarimport/(?P\d+)/(?P\d+)$',views.workout_getpolarworkout_view), url(r'^workout/runkeeperimport/$',views.workout_runkeeperimport_view), url(r'^workout/runkeeperimport/(?P\d+)/$',views.workout_getrunkeeperworkout_view), url(r'^workout/underarmourimport/$',views.workout_underarmourimport_view), diff --git a/rowers/views.py b/rowers/views.py index e36866b7..32e25a3a 100644 --- a/rowers/views.py +++ b/rowers/views.py @@ -23,6 +23,8 @@ import gc from pyparsing import ParseException from uuid import uuid4 +import isodate + from django.shortcuts import render from django.http import ( HttpResponse, HttpResponseRedirect, @@ -2399,10 +2401,9 @@ def rower_polar_authorize(request): "response_type": "code", "redirect_uri": POLAR_REDIRECT_URI, "state": state, - "scope":"accesslink.read_all" +# "scope":"accesslink.read_all" } url = "https://flow.polar.com/oauth2/authorization?" +urllib.urlencode(params) - print url return HttpResponseRedirect(url) @@ -9644,6 +9645,37 @@ def workout_underarmourimport_view(request,message=""): return HttpResponse(res) +# the page where you select which Polar workout to Import +@login_required() +def workout_polarimport_view(request): + exercises = polarstuff.get_polar_workouts(request.user) + workouts = [] + for exercise in exercises: + try: + d = exercise['distance'] + except KeyError: + d = 0 + + i = exercise['id'] + transactionid = exercise['transaction-id'] + starttime = exercise['start-time'] + rowtype = exercise['sport'] + durationstring = exercise['duration'] + duration = isodate.parse_duration(durationstring) + keys = ['id','distance','duration','starttime','type','transactionid'] + values = [i,d,duration,starttime,rowtype,transactionid] + res = dict(zip(keys,values)) + workouts.append(res) + + return render(request, 'polar_list_import.html', + { + 'workouts':workouts, + 'teams':get_my_teams(request.user), + }) + + + + # The page where you select which SportTracks workout to import @login_required() def workout_sporttracksimport_view(request,message=""): @@ -9820,6 +9852,13 @@ def workout_c2import_view(request,page=1,message=""): 'page':page, }) +# Import a workout from Polar +@login_required() +def workout_getpolarworkout_view(request,polarid,transactionid): + result = polarstuff.get_polar_workout(request.user,polarid,transactionid) + + return HttpResponse(result) + # Import a workout from Strava @login_required() def workout_getstravaworkout_view(request,stravaid):