"""Types for the adjustments validation logic."""

from typing import NotRequired, TypedDict


class Adjustment(TypedDict):
    account_id: str | None
    contract_id: str | None
    upc: NotRequired[str | None]
    amount: str | None
    currency: str | None
    activity_month: str | None
    activity_year: str | None
    statement_month: str | None
    statement_year: str | None
    adjustment_type: str | None
    client_facing_comments: str | None
    distribution_type: NotRequired[str | None]
    apply_to_flowthrough_payment: NotRequired[str | None]


class SnowflakeConfig(TypedDict):
    account: str
    role: str
    warehouse: str
    db: str
    schema: str
    user: str
    private_key: str | bytes