from typing import Protocol
from uuid import UUID, uuid4

import sqlalchemy as sa
from sqlalchemy.orm import DeclarativeBase, Mapped, declared_attr


class ModelBase(DeclarativeBase):
    pass


class CompareProtocol(Protocol):
    id: Mapped[int | UUID]  # Or `Mapped[int] | Mapped[UUID]`


class CompareMixin:
    def compare(self: CompareProtocol, other: CompareProtocol) -> bool:
        return self.id == other.id  # Simplified example


class IntIdMixin:
    @declared_attr
    def id(cls) -> Mapped[int]:
        return sa.orm.mapped_column(sa.Integer, primary_key=True)


class UuidIdMixin:
    @declared_attr
    def id(cls) -> Mapped[UUID]:
        return sa.orm.mapped_column(sa.UUID, primary_key=True, default=uuid4)


class MyModel(CompareMixin, IntIdMixin, ModelBase):
    pass


m1 = MyModel()
m2 = MyModel()
m1.compare(m2)  # Mypy reports two errors here