otf-api 0.3.0__py3-none-any.whl → 0.5.0__py3-none-any.whl

otf_api/models/auth.py

@@ -1,147 +0,0 @@
-import json
-from pathlib import Path
-from typing import ClassVar
-
-from pycognito import Cognito, TokenVerificationException
-from pydantic import Field
-
-from otf_api.models.base import OtfItemBase
-
-CLIENT_ID = "65knvqta6p37efc2l3eh26pl5o"  # from otlive
-USER_POOL_ID = "us-east-1_dYDxUeyL1"
-
-
-class IdClaimsData(OtfItemBase):
-    sub: str
-    email_verified: bool
-    iss: str
-    cognito_username: str = Field(alias="cognito:username")
-    given_name: str
-    locale: str
-    home_studio_id: str = Field(alias="custom:home_studio_id")
-    aud: str
-    event_id: str
-    token_use: str
-    auth_time: int
-    exp: int
-    is_migration: str = Field(alias="custom:isMigration")
-    iat: int
-    family_name: str
-    email: str
-    koji_person_id: str = Field(alias="custom:koji_person_id")
-
-    @property
-    def member_uuid(self) -> str:
-        return self.cognito_username
-
-    @property
-    def full_name(self) -> str:
-        return f"{self.given_name} {self.family_name}"
-
-
-class AccessClaimsData(OtfItemBase):
-    sub: str
-    device_key: str
-    iss: str
-    client_id: str
-    event_id: str
-    token_use: str
-    scope: str
-    auth_time: int
-    exp: int
-    iat: int
-    jti: str
-    username: str
-
-    @property
-    def member_uuid(self) -> str:
-        return self.username
-
-
-class User:
-    token_path: ClassVar[Path] = Path("~/.otf/.tokens").expanduser()
-    cognito: Cognito
-
-    def __init__(self, cognito: Cognito):
-        self.cognito = cognito
-
-    @property
-    def member_id(self) -> str:
-        return self.id_claims_data.cognito_username
-
-    @property
-    def member_uuid(self) -> str:
-        return self.access_claims_data.sub
-
-    @property
-    def access_claims_data(self) -> AccessClaimsData:
-        return AccessClaimsData(**self.cognito.access_claims)
-
-    @property
-    def id_claims_data(self) -> IdClaimsData:
-        return IdClaimsData(**self.cognito.id_claims)
-
-    def save_to_disk(self) -> None:
-        self.token_path.parent.mkdir(parents=True, exist_ok=True)
-        data = {
-            "username": self.cognito.username,
-            "id_token": self.cognito.id_token,
-            "access_token": self.cognito.access_token,
-            "refresh_token": self.cognito.refresh_token,
-        }
-        self.token_path.write_text(json.dumps(data))
-
-    @classmethod
-    def cache_file_exists(cls) -> bool:
-        return cls.token_path.exists()
-
-    @classmethod
-    def username_from_disk(cls) -> str:
-        val: str = json.loads(cls.token_path.read_text())["username"]
-        return val
-
-    @classmethod
-    def load_from_disk(cls, username: str, password: str) -> "User":
-        """Load a User instance from disk. If the token is invalid, reauthenticate with the provided credentials.
-
-        Args:
-            username (str): The username to reauthenticate with.
-            password (str): The password to reauthenticate with.
-
-        Returns:
-            User: The loaded user.
-
-        """
-        attr_dict = json.loads(cls.token_path.read_text())
-
-        cognito_user = Cognito(USER_POOL_ID, CLIENT_ID, **attr_dict)
-        try:
-            cognito_user.verify_tokens()
-            return cls(cognito=cognito_user)
-        except TokenVerificationException:
-            user = cls.login(username, password)
-            return user
-
-    @classmethod
-    def login(cls, username: str, password: str) -> "User":
-        """Login and return a User instance. After a successful login, the user is saved to disk.
-
-        Args:
-            username (str): The username to login with.
-            password (str): The password to login with.
-
-        Returns:
-            User: The logged in user.
-        """
-        cognito_user = Cognito(USER_POOL_ID, CLIENT_ID, username=username)
-        cognito_user.authenticate(password)
-        cognito_user.check_token()
-        user = cls(cognito=cognito_user)
-        user.save_to_disk()
-        return user
-
-    def refresh_token(self) -> "User":
-        """Refresh the user's access token."""
-        self.cognito.check_token()
-        self.save_to_disk()
-        return self

otf_api/models/responses/workouts.py

@@ -1,78 +0,0 @@
-from ast import literal_eval
-from datetime import datetime
-from typing import Any
-
-from pydantic import Field, PrivateAttr
-
-from otf_api.models.base import OtfItemBase
-
-
-class WorkoutType(OtfItemBase):
-    id: int
-    display_name: str = Field(..., alias="displayName")
-    icon: str
-
-
-class Workout(OtfItemBase):
-    studio_number: str = Field(..., alias="studioNumber")
-    studio_name: str = Field(..., alias="studioName")
-    class_type: str = Field(..., alias="classType")
-    active_time: int = Field(..., alias="activeTime")
-    coach: str
-    member_uuid: str = Field(..., alias="memberUuId")
-    class_date: datetime = Field(..., alias="classDate")
-    total_calories: int = Field(..., alias="totalCalories")
-    avg_hr: int = Field(..., alias="avgHr")
-    max_hr: int = Field(..., alias="maxHr")
-    avg_percent_hr: int = Field(..., alias="avgPercentHr")
-    max_percent_hr: int = Field(..., alias="maxPercentHr")
-    total_splat_points: int = Field(..., alias="totalSplatPoints")
-    red_zone_time_second: int = Field(..., alias="redZoneTimeSecond")
-    orange_zone_time_second: int = Field(..., alias="orangeZoneTimeSecond")
-    green_zone_time_second: int = Field(..., alias="greenZoneTimeSecond")
-    blue_zone_time_second: int = Field(..., alias="blueZoneTimeSecond")
-    black_zone_time_second: int = Field(..., alias="blackZoneTimeSecond")
-    step_count: int = Field(..., alias="stepCount")
-    class_history_uuid: str = Field(..., alias="classHistoryUuId")
-    class_id: str = Field(..., alias="classId")
-    date_created: datetime = Field(..., alias="dateCreated")
-    date_updated: datetime = Field(..., alias="dateUpdated")
-    is_intro: bool = Field(..., alias="isIntro")
-    is_leader: bool = Field(..., alias="isLeader")
-    member_email: None = Field(..., alias="memberEmail")
-    member_name: str = Field(..., alias="memberName")
-    member_performance_id: str = Field(..., alias="memberPerformanceId")
-    minute_by_minute_hr: list[int] = Field(
-        ...,
-        alias="minuteByMinuteHr",
-        description="HR data for each minute of the workout. It is returned as a string literal, so it needs to be "
-        "evaluated to a list. If can't be parsed, it will return an empty list.",
-    )
-    source: str
-    studio_account_uuid: str = Field(..., alias="studioAccountUuId")
-    version: str
-    workout_type: WorkoutType = Field(..., alias="workoutType")
-    _minute_by_minute_raw: str | None = PrivateAttr(None)
-
-    @property
-    def active_time_minutes(self) -> int:
-        """Get the active time in minutes."""
-        return self.active_time // 60
-
-    def __init__(self, **data: Any):
-        if "minuteByMinuteHr" in data:
-            try:
-                data["minuteByMinuteHr"] = literal_eval(data["minuteByMinuteHr"])
-            except (ValueError, SyntaxError):
-                data["minuteByMinuteHr"] = []
-
-        super().__init__(**data)
-        self._minute_by_minute_raw = data.get("minuteByMinuteHr")
-
-
-class WorkoutList(OtfItemBase):
-    workouts: list[Workout]
-
-    @property
-    def by_class_history_uuid(self) -> dict[str, Workout]:
-        return {workout.class_history_uuid: workout for workout in self.workouts}