import datetime
from typing import Any, ClassVar

import sqlalchemy as sa
from fansifter_common.adapters.db.types import NaiveUTCDateTime
from sqlalchemy.orm import DeclarativeBase, MappedAsDataclass


class Model(MappedAsDataclass, DeclarativeBase):
    """Base class for all db models."""

    type_annotation_map: ClassVar[dict[Any, Any]] = {
        str: sa.Text,
        datetime.datetime: NaiveUTCDateTime,
    }


class PgModel(MappedAsDataclass, DeclarativeBase):
    """Base class for all postgres db models."""

    type_annotation_map: ClassVar[dict[Any, Any]] = {
        str: sa.Text,
        datetime.datetime: NaiveUTCDateTime,
    }