aimbat

class Settings(EventParametersValidatorMixin, BaseSettings):
    model_config = SettingsConfigDict(env_prefix="aimbat_", env_file=".env")

    project: Path = Field(
        default=Path("aimbat.db"),
        description="AIMBAT project file location (ignored if `db_url` is specified).",
    )
    """AIMBAT project file location."""

    db_url: str = Field(
        default="",
        description="AIMBAT database url (default value is derived from `project`).",
    )
    """AIMBAT database url (default is derived from `project`)."""

    logfile: Path = Field(default=Path("aimbat.log"), description="Log file location.")
    """Log file location."""

    debug: bool = Field(default=False, description="Enable debug logging.")
    """Enable debug logging."""

    window_pre: timedelta = Field(
        default=ICCS_DEFAULTS.window_pre,
        lt=0,
        description="Initial relative begin time of window.",
    )
    """Initial relative begin time of window."""

    window_post: timedelta = Field(
        default=ICCS_DEFAULTS.window_post,
        ge=0,
        description="Initial relative end time of window.",
    )
    """Initial relative end time of window."""

    context_width: timedelta = Field(
        default=ICCS_DEFAULTS.context_width,
        gt=0,
        description="Context padding to apply before and after the time window.",
    )
    """Context padding to apply before and after the time window."""

    min_ccnorm: float | np.floating = Field(
        default=ICCS_DEFAULTS.min_ccnorm,
        ge=0,
        le=1,
        description="Initial minimum cross correlation coefficient.",
    )
    """Initial minimum cross correlation coefficient."""

    bandpass_apply: bool = Field(
        default=ICCS_DEFAULTS.bandpass_apply,
        description="Whether to apply bandpass filter to seismograms.",
    )
    """Whether to apply bandpass filter to seismograms."""

    bandpass_fmin: float = Field(
        default=ICCS_DEFAULTS.bandpass_fmin,
        ge=0,
        description="Minimum frequency for bandpass filter (ignored if `bandpass_apply` is False).",
    )
    """Minimum frequency for bandpass filter (ignored if `bandpass_apply` is False)."""

    bandpass_fmax: float = Field(
        default=ICCS_DEFAULTS.bandpass_fmax,
        gt=0,
        description="Maximum frequency for bandpass filter (ignored if `bandpass_apply` is False).",
    )
    """Maximum frequency for bandpass filter (ignored if `bandpass_apply` is False)."""

    sac_pick_header: str = Field(
        default="t0", description="SAC header field where initial pick is stored."
    )
    """SAC header field where initial pick is stored."""

    sampledata_src: str = Field(
        default="https://github.com/pysmo/data-example/archive/refs/heads/aimbat_v2.zip",
        description="URL where sample data is downloaded from.",
    )
    """URL where sample data is downloaded from."""

    sampledata_dir: Path = Field(
        default=Path("sample-data"),
        description="Directory to store downloaded sample data.",
    )
    """Directory to store downloaded sample data."""

    min_id_length: int = Field(
        default=2, ge=1, description="Minimum length of ID string."
    )
    """Minimum length of truncated UUID string."""

    @model_validator(mode="after")
    def set_computed_defaults(self) -> Self:
        """Sets defaults that depend on other fields."""
        # 1. Handle db_url dependency on project
        if self.db_url == "":
            self.db_url = f"sqlite+pysqlite:///{self.project}"
        return self