config.py

import os

from datetime import timedelta
from enum import Enum
from functools import lru_cache
from typing import Dict

import yaml

from pydantic import BaseSettings, NameEmail, SecretStr


class Modalidad(str, Enum):
    GRUPAL = "g"
    INDIVIDUAL = "i"
    PARCIALITO = "p"


class Settings(BaseSettings):
    test: bool
    title: str
    sender: NameEmail
    job_queue: str = "default"

    spreadsheet_id: str
    planilla_ttl: timedelta

    cuatri: str
    entregas: Dict[str, Modalidad]  # TODO: introducir clase Entrega.

    oauth_client_id: str
    oauth_client_secret: SecretStr
    oauth_refresh_token: SecretStr
    service_account_jsonfile: str

    github_app_id: int
    github_app_keyfile: str

    # TODO: borrar este token cuando todo se migre a Github App.
    github_token: SecretStr


@lru_cache
def load_config():
    """Carga la configuración de un archivo YAML.

    Returns:
       un objeto de tipo Settings, validado.
    """
    # FIXME: estaría mejor si las variables de entorno pudieran tomar
    # precedencia sobre la configuración en YAML (ver el orden de prioridad en
    # https://pydantic-docs.helpmanual.io/usage/settings/#field-value-priority).
    with open(os.environ.get("ENTREGAS_CONFIG", "entregas.yml")) as yml:
        return Settings(**yaml.safe_load(yml), _env_file=".secrets")