rowsandall/rowers/rojabo_stuff.py

from rowers.models import Rower, Workout, TombStone
from rowers import utils
import datetime

from rowers.imports import *
from rowsandall_app.settings import (
    ROJABO_CLIENT_ID, ROJABO_REDIRECT_URI, ROJABO_CLIENT_SECRET,
    SITE_URL, ROJABO_OAUTH_LOCATION,
    UPLOAD_SERVICE_URL, UPLOAD_SERVICE_SECRET,
)
import gzip
import rowers.mytypes as mytypes
from rowers.utils import myqueue

import requests
import base64

from rowers.utils import dologging
from json.decoder import JSONDecodeError

import django_rq
queue = django_rq.get_queue('default')
queuelow = django_rq.get_queue('low')
queuehigh = django_rq.get_queue('low')

requests.packages.urllib3.disable_warnings()

oauth_data = {
    'client_id': ROJABO_CLIENT_ID,
    'client_secret': ROJABO_CLIENT_SECRET,
    'redirect_uri': ROJABO_REDIRECT_URI,
    'autorization_uri': ROJABO_OAUTH_LOCATION+"oauth/authorize",
    'content_type': 'application/json',
    'tokenname': 'nktoken',
    'refreshtokenname': 'nkrefreshtoken',
    'expirydatename': 'nktokenexpirydate',
    'bearer_auth': True,
    'base_url': ROJABO_OAUTH_LOCATION+"oauth/token",
    'scope': 'read',
}

def get_token(code):  # pragma: no cover

    post_data = {
        "grant_type": "authorization_code",
        "code": code,
        "redirect_uri": oauth_data['redirect_uri'],
    }


    auth_string = '{id}:{secret}'.format(
        id=ROJABO_CLIENT_ID,
        secret=ROJABO_CLIENT_SECRET
    )



    try:
        headers = {'Authorization': 'Basic %s' % base64.b64encode(auth_string),
                   'Content-Type': 'application/x-www-form-urlencoded'}
    except TypeError:
        headers = {'Authorization': 'Basic %s' % base64.b64encode(
            bytes(auth_string, 'utf-8')).decode('utf-8'),
                   'Content-Type': 'application/x-www-form-urlencoded'}



    response = requests.post(oauth_data['base_url'],
                             data=post_data,
                             headers=headers)

    if response.status_code != 200:
        return (0,response.reason)

    try:
        token_json = response.json()
        thetoken = token_json['access_token']
        expires_in = token_json['expires_in']
        refresh_token = token_json['refresh_token']
    except (KeyError, JSONDecodeError) as e:  # pragma: no cover
        thetoken = 0
        expires_in = 0
        refresh_token = 0

    return [thetoken, expires_in, refresh_token]

def rojabo_open(user):
    r = Rower.objects.get(user=user)
    if (r.rojabo_token == '') or (r.rojabo_token is None):
        raise NoTokenError("User has no token")
    else:
        if (timezone.now() > r.rojabo_tokenexpirydate):
            res = rower_rojabo_token_refresh(user)
            if res is None:  # pragma: no cover
                raise NoTokenError("User has no token")
            if res[0] is not None:
                thetoken = res[0]
            else:  # pragma: no cover
                raise NoTokenError("User has no token")
        else:
            thetoken = r.rojabo_token

    return thetoken

def rower_rojabo_token_refresh(user):
    r = Rower.objects.get(user=user)
    res = do_refresh_token(r.rojabo_refreshtoken)
    if res[0]:
        access_token = res[0]
        expires_in = res[1]
        refresh_token = res[2]
        expirydatetime = timezone.now()+timedelta(seconds=expires_in)

        r = Rower.objects.get(user=user)
        r.rojabo_token = access_token
        r.rojabo_tokenexpirydate = expirydatetime
        r.rojabo_refreshtoken = refresh_token

        r.save()
        return r.rojabo_token
    else:  # pragma: no cover
        return None

def do_refresh_token(refreshtoken):
    post_data = {
        "grant_type": "refresh_token",
        "refresh_token": refreshtoken,
        "redirect_uri": oauth_data['redirect_uri'],
    }


    auth_string = '{id}:{secret}'.format(
        id=ROJABO_CLIENT_ID,
        secret=ROJABO_CLIENT_SECRET
    )



    try:
        headers = {'Authorization': 'Basic %s' % base64.b64encode(auth_string),
                   'Content-Type': 'application/x-www-form-urlencoded'}
    except TypeError:
        headers = {'Authorization': 'Basic %s' % base64.b64encode(
            bytes(auth_string, 'utf-8')).decode('utf-8'),
                   'Content-Type': 'application/x-www-form-urlencoded'}


    response = requests.post(oauth_data['base_url'],
                             data=post_data,
                             headers=headers,
                             verify=False)

    if response.status_code != 200:
        return (0,response.reason)

    try:
        token_json = response.json()
        thetoken = token_json['access_token']
        expires_in = token_json['expires_in']
        refresh_token = token_json['refresh_token']
    except (KeyError, JSONDecodeError) as e:  # pragma: no cover
        thetoken = 0
        expires_in = 0
        refresh_token = 0

    return [thetoken, expires_in, refresh_token]

aweekago = timezone.now()-timedelta(days=7)
today = timezone.now()
a_week_from_now = timezone.now()+timedelta(days=7)

def get_rojabo_workout_list(user,startdate=aweekago,enddate=a_week_from_now):
    r = Rower.objects.get(user=user)
    if (r.rojabo_token == '') or (r.rojabo_token is None):  # pragma: no cover
        s = "Token doesn't exist. Need to authorize"
        return custom_exception_handler(401, s)
    elif (timezone.now() > r.rojabo_tokenexpirydate):  # pragma: no cover
        s = "Token expired. Needs to refresh."

        return custom_exception_handler(401, s)

    _ = rojabo_open(user)

    authorizationstring = str('Bearer ' + r.rojabo_token)

    headers = {'Authorization': authorizationstring,
               'Content-Type': 'application/json'}

    date1 = startdate.strftime('%Y-%m-%d')
    date2 = enddate.strftime('%Y-%m-%d')


    url = ROJABO_OAUTH_LOCATION+'api/v1/training_sessions?from={date1}&to={date2}'.format(date1=date1,date2=date2)

    response = requests.get(url, headers=headers)

    return response