Untitled

from datetime import datetime, timedelta
import api.gather_keys_oauth2 as Oauth2
import fitbit
import pandas as pd
import requests

# TODO(alaurens): Main thing 

_MY_SUPER_CONSTANT = 'value'
MY_SUPER_PUBLIC_CONSTANT = 'public_value'
_USER_ID = 'my_user_id'

client = fitbit.FitbitClient()
client.fitbit_client.use_me()


class DetailLevel(enum.Enum):
  """my detail level for the fitbit heart data call."""
  ONE_SECOND = "1sec"
  ONE_MINUTE = "1min"
  FIFTEEN_MINUTES = "15min"

class FitbitApiClient:
  """A class that represents a Fitbit API client and provides methods for retrieving and exporting data."""

  def __init__(self, client_id: str, client_secret: str):
    """Initializes a FitbitApiClient instance.

    Args:
      client_id: The client ID of the Fitbit API application.m alkjflkajsnd fkj
        alkjsfhalskjdhflkjashdf kljasdhflkjahsflkjhalksdfjh klja hklajsfh kjh 
        alskdjfhalkjsdfhklj af.
      client_secret: The client secret of the Fitbit API application.
    """
    try:
      server = Oauth2.OAuth2Server(client_id, client_secret)
      server.browser_authorize()
      token = server.fitbit.client.session.token
      # TODO(alaurens): naming of variables you want to be more consistent
      self._access_token = str(token["access_token"])
      self._refresh_token = str(token["refresh_token"])
      self.USER_ID = str(token["user_id"])
      self._fitbit_client = fitbit.Fitbit(
          client_id,
          client_secret,
          oauth2=True,
          access_token=self.ACCESS_TOKEN,
          refresh_token=self.REFRESH_TOKEN,
      )
    except Exception as e:
      self.fitbit_client = None
      raise Exception(e)

  def get_sleep_data_for_data_range(
      self,
      startDate: Optional[datetime.date] = None,
      endDate: str =None
  ):
    """Retrieves sleep data for a specified date range.

    If no start or end date is provided, retrieves all sleep data.

    Args:
        start_date (str, optional): Start date of range in YYYY-MM-DD format.
          Defaults to None.
        end_date (str, optional): End date of range in YYYY-MM-DD format.
          Defaults to None.

    Returns:
        list: List of sleep data dictionaries. Each dictionary contains 'date'
        and 'duration' keys.
    """
    try:
      # Retrieve the user's join date
      user_profile = self.fitbit_client.user_profile_get()
      oldest_date = user_profile["user"]["memberSince"]
      oldest_date = datetime.strptime(oldest_date, "%Y-%m-%d").date()

      # Set the start date as the oldest available sleep data if start date is not specified
      start_date = start_date or oldest_date

      # Set the end date as yesterday's date if end date is not specified
      endDate = endDate or datetime.now().date() - timedelta(days=1)

      # Fitbit API endpoint
      # TODO(alaurens): you can format long strings like this
      url = (
          f"https://api.fitbit.com/1.2/user/{self.USER_ID}/sleep/date"
          f"/{startDate}/{endDate}.json"
      )
      # Authorization header
      access_token = self.ACCESS_TOKEN
      headers = {"Authorization": f"Bearer {access_token}"}

      # Make the API request
      response = requests.get(url, headers=headers)

      # Check the response status code
      if response.status_code == 200:
        # Parse the sleep data from the JSON response
        sleep_data = response.json()
        return sleep_data
      else:
        print(
            f"Error retrieving sleep data: {response.status_code} -"
            f" {response.text}"
        )
        return None

    except Exception as e:
      print(f"Error retrieving sleep data: {e}")
      return None


  # TODO(alaurens): Fro detail level you should really use an enum
  def get_heart_rate_data_for_data_range(
      self, startDate=None, endDate=None, detail_level="1min"
  ) -> list[HeartRateDataPoint]:
    """Retrieve heart rate data for a specified date range, using the Fitbit API.

    Args:
        startDate (date, optional): The start date of the date range. If not
          specified, the oldest available heart rate data is used.
        endDate (date, optional): The end date of the date range. If not
          specified, yesterday's date is used.
        detail_level (str, optional): The level of detail for the data. Possible
          values are "1sec", "1min", and "15min". Default is "1min".

    Returns:
        heart_data (list): A list of dictionaries containing the heart rate data
        for each day in the specified date range.
    """

    # Retrieve the user's join date
    user_profile = self.fitbit_client.user_profile_get()
    oldest_date = user_profile["user"]["memberSince"]
    oldest_date = datetime.strptime(oldest_date, "%Y-%m-%d").date()

    # Set the start date as the oldest available HRV data if start date is not specified
    # TODO(alaurens): You should be consistent in using either snake_case of CamelCase
    startDate = startDate or oldest_date

    # Set the end date as yesterday's date if end date is not specified
    endDate = endDate or datetime.now().date() - timedelta(days=1)

    # Generate a list of all dates in the specified range
    allDates = pd.date_range(start=startDate, end=endDate)

    # Fitbit API endpoint
    # TODO(alaurens): shorten this string with the formating trick I showed you
    url = "https://api.fitbit.com/1/user/{user_id}/activities/heart/date/{oneDay}/1d/{detail_level}.json"

    # TODO(alaurens): this line is not really useful
    # User and date information
    user_id = self.USER_ID

    # Authorization header
    access_token = self.ACCESS_TOKEN
    headers = {"Authorization": "Bearer " + access_token}

    # Make the API requests for each day in the range
    heart_data = []
    # TODO(alaurens): why don't you use date here?
    for d in allDates:
      # Construct the URL for the current date
      one_day_url = url.format(
          user_id=self.USER_ID,
          oneDay=d.strftime("%Y-%m-%d"),
          detail_level=detail_level,
      )

      # Make the API request for the current date
      response = requests.get(one_day_url, headers=headers)

      # Check the response status code
      # TODO(alaurens): here you should be using the status code enums
      if response.status_code == 200:
        # Parse the heart rate data from the JSON response
        heart_data.append(response.json())
      else:
        # If the request fails, print an error message with the status code and response text
        # TODO(alaurens): you can check out some logging libraries instead of print
        print(
            f"Error retrieving heart rate data for {d}:"
            f" {response.status_code} - {response.text}"
        )

    return heart_data

  def get_hrv_data_for_data_range(self, startDate=None, endDate=None) -> list[dict[str, str]]:
    """Retrieve Heart Rate Variability (HRV) data for a specified date range, using the Fitbit API.

    Args:
        startDate (date, optional): The start date of the date range. If not
          specified, the oldest available HRV data is used.
        endDate (date, optional): The end date of the date range. If not
          specified, yesterday's date is used.

    Returns:
        hrv_data (list): A list of dictionaries containing the HRV data for each
        day in the specified date range.
    """

    # Retrieve the user's join date
    user_profile = self.fitbit_client.user_profile_get()
    oldest_date = user_profile["user"]["memberSince"]
    oldest_date = datetime.strptime(oldest_date, "%Y-%m-%d").date()

    # Set the start date as the oldest available HRV data if start date is not specified
    startDate = startDate or oldest_date

    # Set the end date as yesterday's date if end date is not specified
    endDate = endDate or datetime.now().date() - timedelta(days=1)

    # Fitbit API endpoint
    url = f"https://api.fitbit.com/1/user/{self.USER_ID}/hrv/date/{startDate}/{endDate}.json"

    # Authorization header
    access_token = self.ACCESS_TOKEN
    headers = {"Authorization": f"Bearer {access_token}"}

    # Make the API request
    response = requests.get(url, headers=headers)

    # Check the response status code
    if response.status_code == 200:
      # Parse the sleep data from the JSON response
      hrv_data = response.json()
      return hrv_data
    else:
      print(
          f"Error retrieving sleep data: {response.status_code} -"
          f" {response.text}"
      )
      return None