def setup_smashrun_code(options, config):
""" Setup Smashrun API explicit code flow (for applications)
"""
if options.key is None:
print(
"To set a connection with Smashrun you need to request an API key at https://api.smashrun.com/register ."
)
options.key = input('Please the client id: ')
if options.secret is None:
options.secret = input('Please enter the client secret: ')
client = Smashrun(client_id=options.key,
client_secret=options.secret,
redirect_uri='urn:ietf:wg:oauth:2.0:auto')
auth_url = client.get_auth_url()
print("Go to '%s' and authorize this application." % auth_url[0])
code = input('Please enter your the code provided: ')
resp = client.fetch_token(code=code)
if not config.has_section('smashrun'):
config.add_section('smashrun')
config.set('smashrun', 'client_id', options.key)
config.set('smashrun', 'client_secret', options.secret)
config.set('smashrun', 'refresh_token', resp['refresh_token'])
config.set('smashrun', 'type', 'code')
def _getClient(self, serviceRec=None):
redirect_uri = None
if not serviceRec:
# reverse() doesn't work in the worker and we only really need the redirect
# uri in the ui...
redirect_uri = WEB_ROOT + reverse('oauth_return', kwargs={'service': 'smashrun'})
client = SmashrunClient(client_id=SMASHRUN_CLIENT_ID,
client_secret=SMASHRUN_CLIENT_SECRET,
redirect_uri=redirect_uri)
if serviceRec:
client.refresh_token(refresh_token=serviceRec.Authorization['refresh_token'])
return client
def setup_smashrun( options, config ):
""" Setup Smashrun API implicit user level authentication
"""
mobile = MobileApplicationClient('client') # implicit flow
client = Smashrun(client_id='client',client=mobile,client_secret='my_secret',redirect_uri='https://httpbin.org/get')
auth_url = client.get_auth_url()
print("Go to '%s' and log into Smashrun. After redirection, copy the access_token from the url." % auth_url[0])
print("Example url: https://httpbin.org/get#access_token=____01234-abcdefghijklmnopABCDEFGHIJLMNOP01234567890&token_type=[...]")
print("Example access_token: ____01234-abcdefghijklmnopABCDEFGHIJLMNOP01234567890")
token = input("Please enter your access token: " )
if not config.has_section('smashrun'):
config.add_section('smashrun')
config.set('smashrun', 'token', urllib.parse.unquote(token))
config.set('smashrun', 'type', 'implicit')
def _getClient(self, serviceRec=None):
redirect_uri = None
if not serviceRec:
# reverse() doesn't work in the worker and we only really need the redirect
# uri in the ui...
redirect_uri = WEB_ROOT + reverse('oauth_return',
kwargs={'service': 'smashrun'})
client = SmashrunClient(client_id=SMASHRUN_CLIENT_ID,
client_secret=SMASHRUN_CLIENT_SECRET,
redirect_uri=redirect_uri)
if serviceRec:
client.refresh_token(
refresh_token=serviceRec.Authorization['refresh_token'])
return client
def auth_smashrun( config ):
""" Authenticate client with Smashrun
"""
if config.get('smashrun', 'type') == 'code':
client = Smashrun(client_id=config.get('smashrun', 'client_id'),
client_secret=config.get('smashrun', 'client_secret'))
client.refresh_token(refresh_token=config.get('smashrun', 'refresh_token'))
else:
mobile = MobileApplicationClient('client') # implicit flow
client = Smashrun(client_id='client', client=mobile,
token={'access_token':config.get('smashrun', 'token'),'token_type':'Bearer'})
return client
def setUp(self):
self.client = Smashrun(client_id=client_id,
client_secret=client_secret,
redirect_uri='http://localhost')
self.client.refresh_token(refresh_token=refresh_token)
class TestSmashrun(TestCase):
"""Tests for the smashrun client.
Note that these tests are *not* unit tests and actually run against the
live Smashrun API, so be nice and don't repeatedly run them a ridiculous
amount.
"""
@classmethod
def setUpClass(cls):
if not all((client_id, client_secret, refresh_token)):
raise SkipTest("Missing SMASHRUN_REFRESH_TOKEN, "
"SMASHRUN_CLIENT_ID, and SMASHRUN_CLIENT_SECRET "
"environment variables")
def setUp(self):
self.client = Smashrun(client_id=client_id,
client_secret=client_secret,
redirect_uri='http://localhost')
self.client.refresh_token(refresh_token=refresh_token)
def test_get_activities(self):
activities = list(self.client.get_activities())
self.assertNotEqual(activities, []) # small chance this will fail...
def test_get_activities_style(self):
for style in ('summary', 'briefs', 'ids'):
activities = self.client.get_activities(count=2, style='ids')
activity = next(activities)
self.assertIsInstance(activity, int)
def test_get_activities_invalid_style(self):
with self.assertRaises(requests.HTTPError):
list(self.client.get_activities(style='foobar'))
def test_get_activities_limit(self):
activities = list(self.client.get_activities(style='ids', limit=3))
self.assertEqual(len(activities), 3)
def test_get_activities_since(self):
one_month_ago = (datetime.datetime.now() - datetime.timedelta(days=30))
# TODO: it would be easier to have a test for this if each date field
# was converted to a datetime...
list(self.client.get_activities(since=one_month_ago))
def test_get_activity(self):
activities = self.client.get_activities()
activity = next(activities)
id_num = activity['activityId']
fetched_activity = self.client.get_activity(id_num)
self.assertEqual(activity['activityId'],
fetched_activity['activityId'])
self.assertIn('recordingValues', fetched_activity)
def test_get_badges(self):
# if it doesn't raise an exception, that's good enough for me
self.client.get_badges()
def test_get_current_weight(self):
self.client.get_current_weight()
def test_get_weight_history(self):
self.client.get_weight_history()
def test_get_stats(self):
self.client.get_stats()
def test_get_stats_year(self):
self.client.get_stats(year=2014)
def test_get_stats_year_month(self):
self.client.get_stats(year=2015, month=2)
def test_get_stats_month_and_no_year(self):
with self.assertRaises(ValueError):
self.client.get_stats(month=1)
def test_get_auth_url(self):
url = self.client.get_auth_url()[0]
self.assertIn('client_id', url)
self.assertIn('client_secret', url)
self.assertIn('redirect_uri', url)
def test_get_userinfo(self):
r = self.client.get_userinfo()
self.assertIn('id', r)
def setUp(self):
self.client = Smashrun(client_id=client_id,
client_secret=client_secret,
redirect_uri='http://localhost')
self.client.refresh_token(refresh_token=refresh_token)
class TestSmashrun(TestCase):
"""Tests for the smashrun client.
Note that these tests are *not* unit tests and actually run against the
live Smashrun API, so be nice and don't repeatedly run them a ridiculous
amount.
"""
@classmethod
def setUpClass(cls):
if not all((client_id, client_secret, refresh_token)):
raise SkipTest("Missing SMASHRUN_REFRESH_TOKEN, "
"SMASHRUN_CLIENT_ID, and SMASHRUN_CLIENT_SECRET "
"environment variables")
def setUp(self):
self.client = Smashrun(client_id=client_id,
client_secret=client_secret,
redirect_uri='http://localhost')
self.client.refresh_token(refresh_token=refresh_token)
def test_get_activities(self):
activities = list(self.client.get_activities())
self.assertNotEqual(activities, []) # small chance this will fail...
def test_get_activities_style(self):
for style in ('summary', 'briefs', 'ids'):
activities = self.client.get_activities(count=2, style='ids')
activity = next(activities)
self.assertIsInstance(activity, int)
def test_get_activities_invalid_style(self):
with self.assertRaises(requests.HTTPError):
list(self.client.get_activities(style='foobar'))
def test_get_activities_limit(self):
activities = list(self.client.get_activities(style='ids', limit=3))
self.assertEqual(len(activities), 3)
def test_get_activities_since(self):
one_month_ago = (datetime.datetime.now() -
datetime.timedelta(days=30))
# TODO: it would be easier to have a test for this if each date field
# was converted to a datetime...
list(self.client.get_activities(since=one_month_ago))
def test_get_activity(self):
activities = self.client.get_activities()
activity = next(activities)
id_num = activity['activityId']
fetched_activity = self.client.get_activity(id_num)
self.assertEqual(activity['activityId'],
fetched_activity['activityId'])
self.assertIn('recordingValues', fetched_activity)
def test_get_badges(self):
# if it doesn't raise an exception, that's good enough for me
self.client.get_badges()
def test_get_current_weight(self):
self.client.get_current_weight()
def test_get_weight_history(self):
self.client.get_weight_history()
def test_get_stats(self):
self.client.get_stats()
def test_get_stats_year(self):
self.client.get_stats(year=2014)
def test_get_stats_year_month(self):
self.client.get_stats(year=2015, month=2)
def test_get_stats_month_and_no_year(self):
with self.assertRaises(ValueError):
self.client.get_stats(month=1)
def test_get_auth_url(self):
url = self.client.get_auth_url()[0]
self.assertIn('client_id', url)
self.assertIn('client_secret', url)
self.assertIn('redirect_uri', url)
def test_get_userinfo(self):
r = self.client.get_userinfo()
self.assertIn('id', r)
