180 lines
4.7 KiB
Python
180 lines
4.7 KiB
Python
# All the functionality needed to connect to Runkeeper
|
|
from rowers.imports import *
|
|
|
|
# Python
|
|
import gzip
|
|
|
|
import base64
|
|
from io import BytesIO
|
|
|
|
|
|
from rowsandall_app.settings import (
|
|
C2_CLIENT_ID, C2_REDIRECT_URI, C2_CLIENT_SECRET,
|
|
STRAVA_CLIENT_ID, STRAVA_REDIRECT_URI, STRAVA_CLIENT_SECRET,
|
|
TP_CLIENT_ID, TP_CLIENT_SECRET,
|
|
TP_REDIRECT_URI,TP_CLIENT_KEY,
|
|
)
|
|
|
|
tpapilocation = "https://api.trainingpeaks.com"
|
|
|
|
from celery import Celery,app
|
|
from django_rq import job
|
|
import time
|
|
from async_messages import message_user,messages
|
|
|
|
oauth_data = {
|
|
'client_id': TP_CLIENT_ID,
|
|
'client_secret': TP_CLIENT_SECRET,
|
|
'redirect_uri': TP_REDIRECT_URI,
|
|
'autorization_uri': "https://oauth.trainingpeaks.com/oauth/authorize?",
|
|
'content_type': 'application/x-www-form-urlencoded',
|
|
'tokenname': 'tptoken',
|
|
'refreshtokenname': 'tprefreshtoken',
|
|
'expirydatename': 'tptokenexpirydate',
|
|
'bearer_auth': False,
|
|
'base_url': "https://oauth.trainingpeaks.com/oauth/token",
|
|
}
|
|
|
|
|
|
# Checks if user has UnderArmour token, renews them if they are expired
|
|
def tp_open(user):
|
|
return imports_open(user, oauth_data)
|
|
|
|
# Refresh ST token using refresh token
|
|
def do_refresh_token(refreshtoken):
|
|
return imports_do_refresh_token(refreshtoken, oauth_data)
|
|
|
|
# Exchange access code for long-lived access token
|
|
def get_token(code):
|
|
return imports_get_token(code, oauth_data)
|
|
|
|
# Make authorization URL including random string
|
|
def make_authorization_url(request):
|
|
return imports_make_authorization_url(oauth_data)
|
|
|
|
|
|
def getidfromresponse(response):
|
|
t = json.loads(response.text)
|
|
|
|
links = t["_links"]
|
|
|
|
id = links["self"][0]["id"]
|
|
|
|
return int(id)
|
|
|
|
def createtpworkoutdata(w):
|
|
filename = w.csvfilename
|
|
row = rowingdata(filename)
|
|
tcxfilename = filename[:-4]+'.tcx'
|
|
try:
|
|
newnotes = w.notes+'\n from '+w.workoutsource+' via rowsandall.com'
|
|
except TypeError:
|
|
newnotes = 'from '+w.workoutsource+' via rowsandall.com'
|
|
|
|
row.exporttotcx(tcxfilename,notes=newnotes)
|
|
|
|
return tcxfilename
|
|
|
|
|
|
def tp_check(access_token):
|
|
headers = {
|
|
"Content-Type": "application/json",
|
|
'Accept': 'application/json',
|
|
'authorization': 'Bearer %s' % access_token
|
|
}
|
|
|
|
resp = requests.post(tpapilocation+"/v1/info/version",
|
|
headers=headers)
|
|
|
|
return resp
|
|
|
|
def uploadactivity(access_token,filename,description='',
|
|
name='Rowsandall.com workout'):
|
|
|
|
data_gz = BytesIO()
|
|
with file(filename,'rb') as inF:
|
|
s = inF.read()
|
|
with gzip.GzipFile(fileobj=data_gz,mode="w") as gzf:
|
|
gzf.write(s)
|
|
|
|
|
|
headers = {
|
|
'Content-Type': 'application/json',
|
|
'Accept': 'application/json',
|
|
'Authorization': 'Bearer %s' % access_token
|
|
}
|
|
|
|
|
|
|
|
data = {
|
|
"UploadClient": "rowsandall",
|
|
"Filename": filename,
|
|
"SetWorkoutPublic": True,
|
|
"Title":name,
|
|
"Type": "rowing",
|
|
"Comment": description,
|
|
"Data": base64.b64encode(data_gz.getvalue()).decode("ascii")
|
|
}
|
|
|
|
resp = requests.post(tpapilocation+"/v1/file",
|
|
data = json.dumps(data),
|
|
headers=headers)
|
|
|
|
if resp.status_code != 200:
|
|
if settings.DEBUG:
|
|
print resp.status_code
|
|
print resp.reason
|
|
print ""
|
|
print headers
|
|
print ""
|
|
return 0,resp.reason,resp.status_code,headers
|
|
else:
|
|
return resp.json()[0]["Id"],"ok",200,""
|
|
|
|
return 0,0,0,0
|
|
|
|
|
|
def workout_tp_upload(user,w):
|
|
message = "Uploading to TrainingPeaks"
|
|
tpid = 0
|
|
r = w.user
|
|
|
|
thetoken = tp_open(r.user)
|
|
|
|
if (checkworkoutuser(user,w)):
|
|
tcxfile = createtpworkoutdata(w)
|
|
if tcxfile:
|
|
res,reason,status_code,headers = uploadactivity(
|
|
thetoken,tcxfile,
|
|
name=w.name
|
|
)
|
|
if res == 0:
|
|
message = "Upload to TrainingPeaks failed with status code "+str(status_code)+": "+reason
|
|
w.tpid = -1
|
|
try:
|
|
os.remove(tcxfile)
|
|
except WindowsError:
|
|
pass
|
|
|
|
return message,tpid
|
|
|
|
else: # res != 0
|
|
w.uploadedtotp = res
|
|
tpid = res
|
|
w.save()
|
|
os.remove(tcxfile)
|
|
return 'Successfully synchronized to TrainingPeaks',tpid
|
|
|
|
else: # no tcxfile
|
|
message = "Upload to TrainingPeaks failed"
|
|
w.uploadedtotp = -1
|
|
tpid = -1
|
|
w.save()
|
|
return message,tpid
|
|
else: # not allowed to upload
|
|
message = "You are not allowed to export this workout to TP"
|
|
tpid = 0
|
|
return message,tpid
|
|
|
|
return message,tpid
|