| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061 |
- import logging
- from typing import Optional
- from pydantic import BaseModel
- from .config import get_auth_path
- logger = logging.getLogger("fastapi_cli")
- class AuthConfig(BaseModel):
- access_token: str
- def write_auth_config(auth_data: AuthConfig) -> None:
- auth_path = get_auth_path()
- logger.debug("Writing auth config to: %s", auth_path)
- auth_path.write_text(auth_data.model_dump_json(), encoding="utf-8")
- logger.debug("Auth config written successfully")
- def delete_auth_config() -> None:
- auth_path = get_auth_path()
- logger.debug("Deleting auth config at: %s", auth_path)
- if auth_path.exists():
- auth_path.unlink()
- logger.debug("Auth config deleted successfully")
- else:
- logger.debug("Auth config file doesn't exist, nothing to delete")
- def read_auth_config() -> Optional[AuthConfig]:
- auth_path = get_auth_path()
- logger.debug("Reading auth config from: %s", auth_path)
- if not auth_path.exists():
- logger.debug("Auth config file doesn't exist")
- return None
- logger.debug("Auth config loaded successfully")
- return AuthConfig.model_validate_json(auth_path.read_text(encoding="utf-8"))
- def get_auth_token() -> Optional[str]:
- logger.debug("Getting auth token")
- auth_data = read_auth_config()
- if auth_data is None:
- logger.debug("No auth data found")
- return None
- logger.debug("Auth token retrieved successfully")
- return auth_data.access_token
- def is_logged_in() -> bool:
- result = get_auth_token() is not None
- logger.debug("Login status: %s", result)
- return result
|
