refactor: Split code into many files

auth.py

@@ -0,0 +1,116 @@
+# Native modules
+import logging
+import os
+from typing import Dict, Any, Optional
+
+# Third-party modules
+import requests
+from urllib.parse import urlencode
+
+# Configuration constants (will be moved from crossfit_booker.py)
+APPLICATION_ID = "81560887"
+DEVICE_TYPE = "3"
+APP_VERSION = "5.09.21"
+
+class AuthHandler:
+    """
+    A class for handling authentication with the CrossFit booking system.
+    This class is responsible for performing login, retrieving auth tokens,
+    and providing authentication headers.
+    """
+
+    def __init__(self, username: str, password: str) -> None:
+        """
+        Initialize the AuthHandler with credentials.
+
+        Args:
+            username (str): The username for authentication.
+            password (str): The password for authentication.
+        """
+        self.username = username
+        self.password = password
+        self.auth_token: Optional[str] = None
+        self.user_id: Optional[str] = None
+        self.session = requests.Session()
+        self.base_headers: Dict[str, str] = {
+            "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.15; rv:140.0) Gecko/20100101 Firefox/140.0",
+            "Content-Type": "application/x-www-form-urlencoded",
+            "Nubapp-Origin": "user_apps",
+        }
+        self.session.headers.update(self.base_headers)
+
+    def get_auth_headers(self) -> Dict[str, str]:
+        """
+        Return headers with authorization if available.
+
+        Returns:
+            Dict[str, str]: Headers dictionary with authorization if available.
+        """
+        headers: Dict[str, str] = self.base_headers.copy()
+        if self.auth_token:
+            headers["Authorization"] = f"Bearer {self.auth_token}"
+        return headers
+
+    def login(self) -> bool:
+        """
+        Authenticate and get the bearer token.
+
+        Returns:
+            bool: True if login is successful, False otherwise.
+        """
+        try:
+            # First login endpoint
+            login_params: Dict[str, str] = {
+                "app_version": APP_VERSION,
+                "device_type": DEVICE_TYPE,
+                "username": self.username,
+                "password": self.password
+            }
+
+            response: requests.Response = self.session.post(
+                "https://sport.nubapp.com/api/v4/users/checkUser.php",
+                headers={"Content-Type": "application/x-www-form-urlencoded"},
+                data=urlencode(login_params))
+
+            if not response.ok:
+                logging.error(f"First login step failed: {response.status_code} - {response.text[:100]}")
+                return False
+
+            try:
+                login_data: Dict[str, Any] = response.json()
+                self.user_id = str(login_data["data"]["user"]["id_user"])
+            except (KeyError, ValueError) as e:
+                logging.error(f"Error during login: {str(e)} - Response: {response.text}")
+                return False
+
+            # Second login endpoint
+            response: requests.Response = self.session.post(
+                "https://sport.nubapp.com/api/v4/login",
+                headers={"Content-Type": "application/x-www-form-urlencoded;charset=UTF-8"},
+                data=urlencode({
+                    "device_type": DEVICE_TYPE,
+                    "username": self.username,
+                    "password": self.password
+                }))
+
+            if response.ok:
+                try:
+                    login_data: Dict[str, Any] = response.json()
+                    self.auth_token = login_data.get("token")
+                except (KeyError, ValueError) as e:
+                    logging.error(f"Error during login: {str(e)} - Response: {response.text}")
+                    return False
+
+                if self.auth_token and self.user_id:
+                    logging.info("Successfully logged in")
+                    return True
+            else:
+                logging.error(f"Login failed: {response.status_code} - {response.text[:100]}")
+                return False
+
+        except requests.exceptions.RequestException as e:
+            logging.error(f"Request error during login: {str(e)}")
+            return False
+        except Exception as e:
+            logging.error(f"Unexpected error during login: {str(e)}")
+            return False