from datetime import datetime

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 default db models."""

    type_annotation_map = {
        str: sa.Text,
        datetime: NaiveUTCDateTime,
    }


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

    type_annotation_map = {
        str: sa.Text,
        datetime: NaiveUTCDateTime,
    }