Skip to content

Instantly share code, notes, and snippets.

@pepl
Forked from jkreileder/whoop-goldencheetah.py
Created April 27, 2020 16:54
Show Gist options
  • Save pepl/40b8d9515ad5b5a80b728cadb651289a to your computer and use it in GitHub Desktop.
Save pepl/40b8d9515ad5b5a80b728cadb651289a to your computer and use it in GitHub Desktop.
Python 3 script to export WHOOP Strap recovery data for use with Golden Cheetah
#!/usr/bin/env python
import requests # for getting URL
import json # for parsing json
from datetime import datetime # datetime parsing
import pytz # timezone adjusting
import csv # for making csv files
import os
#################################################################
# USER VARIABLES
username = "user@gmail.com"
password = "password123"
save_directory = "~/" # keep trailing slash
#################################################################
# GET ACCESS TOKEN
# Post credentials
r = requests.post("https://api-7.whoop.com/oauth/token", json={
"grant_type": "password",
"issueRefresh": False,
"password": password,
"username": username
})
# Exit if fail
if r.status_code != 200:
print("Fail - Credentials rejected.")
exit()
else:
print("Success - Credentials accepted")
# Set userid/token variables
userid = r.json()['user']['id']
access_token = r.json()['access_token']
#################################################################
# GET DATA
# Download data
url = 'https://api-7.whoop.com/users/{}/cycles'.format(userid)
params = {
'start': '2000-01-01T00:00:00.000Z',
'end': '2030-01-01T00:00:00.000Z'
}
headers = {
'Authorization': 'bearer {}'.format(access_token)
}
r = requests.get(url, params=params, headers=headers)
# Check if user/auth are accepted
if r.status_code != 200:
print("Fail - User ID / auth token rejected.")
exit()
else:
print("Success - User ID / auth token accepted")
#################################################################
# PARSE/TRANSFORM DATA
# Convert data to json
data_raw = r.json()
# Takes a time and offset string and returns a timezone-corrected datetime string
def time_parse(time_string, offset_string):
# Switch sign on offset
offset_string = offset_string.replace(
'-', '+') if offset_string.count('-') else offset_string.replace('+', '-')
# Remove tz from time and add offset, get to 19 characters
time_string = time_string[:-(len(time_string) - 19)] + offset_string
# Parse and format
oldformat = '%Y-%m-%dT%H:%M:%S%z'
newformat = '%Y-%m-%d %H:%M:%S'
return datetime.strptime(time_string, oldformat).astimezone(pytz.utc).strftime(newformat)
# Make data object
data_summary = []
# Iterate through data
for d in data_raw:
# Make record object with default values
record = {
'timestamp_measurement': None,
'HR': None,
'AVNN': None,
'SDNN': None,
'rMSSD': None,
'pNN50': None,
'LF': None,
'HF': None,
'HRV4T_Recovery_Points': None
}
# Recovery
if (d['recovery'] and
'timestamp' in d['recovery'] and
'heartRateVariabilityRmssd' in d['recovery'] and
isinstance(d['recovery']['heartRateVariabilityRmssd'], (int, float)) and
d['sleep'] and
d['sleep']['sleeps'] and
d['sleep']['sleeps'][0]['timezoneOffset']):
# This is the timestamp when Whoop processed sleep -
# not the time of measurement
record['timestamp_measurement'] = time_parse(
d['recovery']['timestamp'],
d['sleep']['sleeps'][0]['timezoneOffset'])
record['rMSSD'] = d['recovery']['heartRateVariabilityRmssd'] * 1000.0
if ('restingHeartRate' in d['recovery'] and
isinstance(d['recovery']['restingHeartRate'], (int, float))):
record['HR'] = d['recovery']['restingHeartRate']
# Recovery score
if ('score' in d['recovery'] and
isinstance(d['recovery']['score'], (int, float))):
record['HRV4T_Recovery_Points'] = d['recovery']['score'] / 10.0
# Append record to data dictionary
data_summary.append(record)
#################################################################
# WRITE JSON RAW DATA FILE
'''
# Write json file
with open(save_directory + 'whoop_raw.json', 'w') as outfile:
json.dump(data_raw, outfile)
print("Success - JSON raw data saved.")
'''
#################################################################
# WRITE CSV SUMMARY DATA FILE
# Write to CSV file
with open(os.path.expanduser(save_directory + 'whoop-goldencheetah.csv'), 'w', newline='') as f:
writer = csv.DictWriter(f, fieldnames=data_summary[0].keys())
# Write header
writer.writeheader()
# Write rows
for row in data_summary:
writer.writerow(row)
print("Success - CSV summary data saved.")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment