run.py

"""
Main file
"""
import sys
import json
import os
from strava_oauth import StravaOauth
from strava_workouts import StravaWorkouts
from helpers import Helpers
from config import Config

helpers = Helpers()
config = Config()
strava_workouts = StravaWorkouts()
strava_oauth = StravaOauth()

helpers.welcome()

#region #? Read config, secret handling, & do oauth
workdir = f"{os.path.dirname(os.path.realpath(__file__))}"
secrets_file, config_file = f"{workdir}/settings/secrets.json", f"{workdir}/settings/config.json"
workout_db_file = f"{workdir}/settings/downloaded_workouts.json"

# Config file management ======================================================
if not os.path.exists(config_file):
  tracks_dir = config.ask_for_path(message="\033[93m⚠️  [Optional] Please, provide a full path to a folder for storing your tracks on",
                                    prompt="\033[95m📂 Tracks Folder: \033[0m")

  workouts_dir = config.ask_for_path(message="\033[93m⚠️  [Optional] Please, provide a full path to a folder for storing your workouts on",
                                    prompt="\033[95m📂 Workouts Folder: \033[0m")

  if tracks_dir == "":
    tracks_dir = f"{workdir}/tracks"
    print(f"\033[94mℹ️  Custom tracks output dir not set. Defaulting to \"{tracks_dir}\"\033[0m")

  if workouts_dir == "":
    workouts_dir = f"{workdir}/workouts"
    print(f"\033[94mℹ️  Custom workouts output dir not set. Defaulting to \"{workouts_dir}\"\033[0m")

  config.write_config_file(config_file=config_file,
                            tracks_output_path=tracks_dir,
                            workouts_output_path=workouts_dir)

else:
  tracks_dir, workouts_dir = config.read_config_file(config_file)
# =============================================================================

# Activities DB file management ===============================================
if not os.path.exists(workout_db_file):
  downloaded_workouts_db = {}
  with open(workout_db_file, mode="w", encoding="utf8") as f:
    f.write(json.dumps(downloaded_workouts_db))
else:
  downloaded_workouts_db = config.read_downloaded_workouts(db_file=workout_db_file)
# =============================================================================

# Secrets file management =====================================================
strava_access_token, strava_refresh_token = "", ""
if not os.path.exists(secrets_file):
  # There's no secrets file. Ask user for client ID & Secret
  strava_client_id, strava_client_secret = strava_oauth.ask_for_secrets()
  if strava_client_secret == "" or strava_client_id == "":
    print("\033[91m❌ Either strava's \"Client Secret\" or \"Client ID\" provided are empty. Check them, then try again.\033[0m")
    sys.exit(1)
  else:
    # Write client info to secrets file
    config.write_secrets_file(secrets_file=secrets_file, \
                            strava_client_id=strava_client_id, \
                            strava_client_secret=strava_client_secret)
else:
  # Get all credentials from file
  strava_access_token, \
  strava_refresh_token, \
  strava_client_id, \
  strava_client_secret = config.read_secrets_file(secrets_file)

if strava_access_token == "":
  # No access token present. Let's retrieve them
  strava_access_token, \
  strava_refresh_token = strava_oauth.do_oauth_flow(client_id=strava_client_id, \
                                                    client_secret=strava_client_secret)
else:
  if not strava_oauth.check_access_token(access_token=strava_access_token):
    # Refresh invalid access_token, so we don't bother user
    strava_access_token = strava_oauth.refresh_access_token(client_id=strava_client_id, \
                                              client_secret=strava_client_secret, \
                                              refresh_token=strava_refresh_token)

if strava_access_token == "":
  # If at this point we still have no access token, we've failed and can't do anything about it,
  # so we exit with error
  print("\033[91m❌ Unable to retrieve tokens. Check provided strava's \"Client ID\" & \"Secret\", then try again\033[0m")
  sys.exit(1)
else:
  config.write_secrets_file(secrets_file=secrets_file, \
                          strava_client_id=strava_client_id, \
                          strava_client_secret=strava_client_secret, \
                          strava_access_token=strava_access_token, \
                          strava_refresh_token=strava_refresh_token)

print("\033[92m🔐 Authentication successful!\n\033[0m")
# =============================================================================

#endregion

# =============================================================================
# Main program flow ===========================================================
# =============================================================================

# Get full workouts' list to download
workout_list = strava_workouts.get_workout_list(access_token=strava_access_token)

# Download all workouts
strava_workouts.download_all_workouts(workdir=workouts_dir, \
                              workout_list=workout_list, \
                              access_token=strava_access_token, \
                              downloaded_workouts_db=downloaded_workouts_db, \
                              workout_db_file=workout_db_file)

# Extract tracks and convert them to gpx
strava_workouts.extract_all_tracks(workouts_dir=workouts_dir, tracks_dir=tracks_dir)