def __init__(self, access_token=None, mongodb=None):
self.logger = logging.getLogger(__name__)
self.client = None
if access_token:
self.set_access_token(access_token)
self.__mongoh = mongohelper(mongodb)
def update_athlete(athlete):
capture_datetime = timezone.now()
capture_timestamp = int(capture_datetime.strftime('%s'))
logger.debug("Starting update for athlete {}. Capture datetime = {} timestamp = {}".format(athlete.id, capture_datetime.strftime('%c'), capture_timestamp))
mongoh = mongohelper()
collection = mongoh.get_collection('raw_activities')
token = athlete.strava_api_token
stravahelper = strava(token)
after_timestamp = None
if athlete.last_strava_sync:
filters = [
{'field': 'athlete.id', 'query': athlete.strava_id}
]
max_record = stravahelper.aggregate_activities_mongo(filters, {
'_id': None,
'max_start_date': {'$max': '$start_date'},
})
record = None
for agg in max_record:
if not record:
record = agg
if record:
after_timestamp = int(record['max_start_date'].strftime('%s'))
else:
after_timestamp = capture_timestamp - 604800
logger.debug("After dbstring = {} timestamp = {}".format(athlete.last_strava_sync.strftime('%c'), after_timestamp))
activities = stravahelper.get_athlete_activities_api(after=after_timestamp)
for activity in activities:
activity['_id'] = activity['id']
activity['capture_timestamp'] = capture_timestamp
activity['start_date'] = datetime.strptime(activity['start_date'],'%Y-%m-%dT%H:%M:%SZ')
activity['strava_api_version'] = 3
collection.replace_one({'_id': activity['id']}, activity, upsert=True)
athlete.last_strava_sync = capture_datetime
bikes,shoes = stravahelper.get_athlete_gear_api()
for api_bike in bikes:
db_bike = bike()
db_bike.strava_id = api_bike.id
db_bike.athlete = athlete
db_bike.primary = api_bike.primary
db_bike.name = api_bike.name
db_bike.description = api_bike.description
db_bike.resource_state = api_bike.resource_state
db_bike.brand_name = api_bike.brand_name
db_bike.model_name = api_bike.model_name
db_bike.frame_type = api_bike.frame_type
db_bike.save()
athlete.save()
def aggregate_activities_mongo(self, filters=[], aggregate={}):
filter_dict = self._assemble_mongo_filters(filters)
mongoh = mongohelper()
mongo = mongoh.get_collection('raw_activities')
return mongo.aggregate([{'$match': filter_dict}, {'$group': aggregate}])
def get_activities_mongo(self, filters=[]):
filter_dict = self._assemble_mongo_filters(filters)
mongoh = mongohelper()
mongo = mongoh.get_collection('raw_activities')
return mongo.find(filter_dict)
