sander/rowsandall
Private
Public Access
1
0
Fork 0
Code Activity

commit

Browse Source
This commit is contained in:
Sander Roosendaal
2022-02-15 08:05:12 +01:00
parent 5b3d7fcf2c
commit 8af7ac8af4
71 changed files with 19992 additions and 19476 deletions
Download Patch File Download Diff File Expand all files Collapse all files

146
rowers/rp3stuff.py
View File

@@ -1,4 +1,18 @@
from __future__ import absolute_import
from celery import Celery, app
from rowers.rower_rules import is_workout_user
import time
from django_rq import job
from rowers.tasks import handle_rp3_async_workout
from rowsandall_app.settings import (
C2_CLIENT_ID, C2_REDIRECT_URI, C2_CLIENT_SECRET,
STRAVA_CLIENT_ID, STRAVA_REDIRECT_URI, STRAVA_CLIENT_SECRET,
RP3_CLIENT_ID, RP3_CLIENT_SECRET,
RP3_REDIRECT_URI, RP3_CLIENT_KEY,
RP3_CLIENT_ID, RP3_CLIENT_KEY, RP3_REDIRECT_URI, RP3_CLIENT_SECRET,
UPLOAD_SERVICE_URL, UPLOAD_SERVICE_SECRET
)
from rowers.utils import myqueue
from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals
@@ -19,23 +33,7 @@ queue = django_rq.get_queue('default')
queuelow = django_rq.get_queue('low')
queuehigh = django_rq.get_queue('high')
from rowers.utils import myqueue
from rowsandall_app.settings import (
C2_CLIENT_ID, C2_REDIRECT_URI, C2_CLIENT_SECRET,
STRAVA_CLIENT_ID, STRAVA_REDIRECT_URI, STRAVA_CLIENT_SECRET,
RP3_CLIENT_ID, RP3_CLIENT_SECRET,
RP3_REDIRECT_URI,RP3_CLIENT_KEY,
RP3_CLIENT_ID, RP3_CLIENT_KEY, RP3_REDIRECT_URI, RP3_CLIENT_SECRET,
UPLOAD_SERVICE_URL, UPLOAD_SERVICE_SECRET
)
from rowers.tasks import handle_rp3_async_workout
from celery import Celery,app
from django_rq import job
import time
#from async_messages import message_user,messages
oauth_data = {
@@ -44,16 +42,15 @@ oauth_data = {
'redirect_uri': RP3_REDIRECT_URI,
'autorization_uri': "https://rp3rowing-app.com/oauth/authorize?",
'content_type': 'application/x-www-form-urlencoded',
# 'content_type': 'application/json',
# 'content_type': 'application/json',
'tokenname': 'rp3token',
'refreshtokenname': 'rp3refreshtoken',
'expirydatename': 'rp3tokenexpirydate',
'bearer_auth': False,
'base_url': "https://rp3rowing-app.com/oauth/token",
'scope':'read,write',
}
'scope': 'read,write',
}
from rowers.rower_rules import is_workout_user
graphql_url = "https://rp3rowing-app.com/graphql"
@@ -63,29 +60,33 @@ def rp3_open(user):
return imports_open(user, oauth_data)
# Refresh ST token using refresh token
def do_refresh_token(refreshtoken): # pragma: no cover
def do_refresh_token(refreshtoken): # pragma: no cover
return imports_do_refresh_token(refreshtoken, oauth_data)
# Exchange access code for long-lived access token
def get_token(code): # pragma: no cover
client_auth = requests.auth.HTTPBasicAuth(RP3_CLIENT_KEY, RP3_CLIENT_SECRET)
def get_token(code): # pragma: no cover
client_auth = requests.auth.HTTPBasicAuth(
RP3_CLIENT_KEY, RP3_CLIENT_SECRET)
post_data = {
"client_id":RP3_CLIENT_KEY,
"client_id": RP3_CLIENT_KEY,
"grant_type": "authorization_code",
"code": code,
"redirect_uri":RP3_REDIRECT_URI,
"client_secret": RP3_CLIENT_SECRET,
"redirect_uri": RP3_REDIRECT_URI,
"client_secret": RP3_CLIENT_SECRET,
}
headers = {
'Content-Type': 'application/x-www-form-urlencoded',
'Content-Type': 'application/x-www-form-urlencoded',
}
response = requests.post(
"https://rp3rowing-app.com/oauth/token",
data=post_data,verify=False,
"https://rp3rowing-app.com/oauth/token",
data=post_data, verify=False,
)
try:
token_json = response.json()
thetoken = token_json['access_token']
@@ -96,10 +97,12 @@ def get_token(code): # pragma: no cover
expires_in = 0
refresh_token = 0
return thetoken,expires_in,refresh_token
return thetoken, expires_in, refresh_token
# Make authorization URL including random string
def make_authorization_url(request): # pragma: no cover
def make_authorization_url(request): # pragma: no cover
return imports_make_authorization_url(oauth_data)
@@ -108,7 +111,7 @@ def get_rp3_workout_list(user):
auth_token = rp3_open(user)
headers = {'Authorization': 'Bearer ' + auth_token }
headers = {'Authorization': 'Bearer ' + auth_token}
get_workouts_list = """{
workouts{
@@ -118,15 +121,15 @@ def get_rp3_workout_list(user):
}"""
response = requests.post(
url=graphql_url,
headers=headers,
json={'query': get_workouts_list}
url=graphql_url,
headers=headers,
json={'query': get_workouts_list}
)
return response
def get_rp3_workouts(rower,do_async=True): # pragma: no cover
def get_rp3_workouts(rower, do_async=True): # pragma: no cover
try:
auth_token = rp3_open(rower.user)
except NoTokenError:
@@ -138,11 +141,11 @@ def get_rp3_workouts(rower,do_async=True): # pragma: no cover
return 0
s = '{d}'.format(d=res.json())
dologging('rp3_import.log',s)
dologging('rp3_import.log', s)
workouts_list = pd.json_normalize(res.json()['data']['workouts'])
try:
rp3ids = workouts_list['id'].values
workouts_list.set_index('id',inplace=True)
workouts_list.set_index('id', inplace=True)
except (KeyError, IndexError):
return 0
@@ -153,34 +156,36 @@ def get_rp3_workouts(rower,do_async=True): # pragma: no cover
newids = [rp3id for rp3id in rp3ids if not rp3id in knownrp3ids]
for id in newids:
startdatetime = workouts_list.loc[id,'executed_at']
dologging('rp3_import.log',startdatetime)
startdatetime = workouts_list.loc[id, 'executed_at']
dologging('rp3_import.log', startdatetime)
job = myqueue(
queuehigh,
handle_rp3_async_workout,
rower.user.id,
auth_token,
id,
startdatetime,
10,
queuehigh,
handle_rp3_async_workout,
rower.user.id,
auth_token,
id,
startdatetime,
10,
)
return 1
def download_rp3_file(url,auth_token,filename): # pragma: no cover
headers = {'Authorization': 'Bearer ' + auth_token }
res = requests.get(url,headers=headers)
def download_rp3_file(url, auth_token, filename): # pragma: no cover
headers = {'Authorization': 'Bearer ' + auth_token}
res = requests.get(url, headers=headers)
if res.status_code == 200:
with open(filename,'wb') as f:
with open(filename, 'wb') as f:
f.write(res.content)
return res.status_code
def get_rp3_workout_token(workout_id,auth_token,waittime=3,max_attempts=20): # pragma: no cover
headers = {'Authorization': 'Bearer ' + auth_token }
def get_rp3_workout_token(workout_id, auth_token, waittime=3, max_attempts=20): # pragma: no cover
headers = {'Authorization': 'Bearer ' + auth_token}
get_download_link = """{
download(workout_id: """ + str(workout_id) + """, type:csv){
@@ -198,20 +203,21 @@ def get_rp3_workout_token(workout_id,auth_token,waittime=3,max_attempts=20): # p
url=graphql_url,
headers=headers,
json={'query': get_download_link}
)
)
if response.status_code != 200:
have_link = True
workout_download_details = pd.json_normalize(response.json()['data']['download'])
workout_download_details = pd.json_normalize(
response.json()['data']['download'])
if workout_download_details.iat[0,1] == 'ready':
download_url = workout_download_details.iat[0,2]
if workout_download_details.iat[0, 1] == 'ready':
download_url = workout_download_details.iat[0, 2]
have_link = True
counter += 1
if counter>max_attempts:
if counter > max_attempts:
have_link = True
time.sleep(waittime)
@@ -219,14 +225,15 @@ def get_rp3_workout_token(workout_id,auth_token,waittime=3,max_attempts=20): # p
return download_url
def get_rp3_workout_link(user,workout_id,waittime=3,max_attempts=20): # pragma: no cover
def get_rp3_workout_link(user, workout_id, waittime=3, max_attempts=20): # pragma: no cover
r = Rower.objects.get(user=user)
auth_token = rp3_open(user)
return get_rp3_workout_token(workout_id,auth_token,waittime=waittime,max_attempts=max_attempts)
return get_rp3_workout_token(workout_id, auth_token, waittime=waittime, max_attempts=max_attempts)
def get_rp3_workout(user,workout_id,startdatetime=None): # pragma: no cover
url = get_rp3_workout_link(user,workout_id)
def get_rp3_workout(user, workout_id, startdatetime=None): # pragma: no cover
url = get_rp3_workout_link(user, workout_id)
filename = 'media/RP3Import_'+str(workout_id)+'.csv'
r = Rower.objects.get(user=user)
@@ -235,7 +242,6 @@ def get_rp3_workout(user,workout_id,startdatetime=None): # pragma: no cover
if not startdatetime:
startdatetime = str(timezone.now())
status_code = download_rp3_file(url, auth_token, filename)
if status_code != 200:
@@ -247,17 +253,17 @@ def get_rp3_workout(user,workout_id,startdatetime=None): # pragma: no cover
'secret': UPLOAD_SERVICE_SECRET,
'user': userid,
'file': filename,
'workouttype':'dynamic',
'boattype':'1x',
'rp3id':workout_id,
'startdatetime':startdatetime,
'workouttype': 'dynamic',
'boattype': '1x',
'rp3id': workout_id,
'startdatetime': startdatetime,
}
session = requests.session()
newHeaders = {'Content-type': 'application/json', 'Accept': 'text/plain'}
session.headers.update(newHeaders)
response = session.post(UPLOAD_SERVICE_URL,json=uploadoptions)
response = session.post(UPLOAD_SERVICE_URL, json=uploadoptions)
if response.status_code != 200:
return 0