from datetime import datetime
from enum import IntEnum

from sqlalchemy.orm import Mapped, mapped_column

from models import BaseTable


class UserRole(IntEnum):
    REGULAR = 0
    ADMIN = 1


class User(BaseTable):
    __tablename__ = "user"

    id: Mapped[int] = mapped_column(primary_key=True, autoincrement=False)
    role: Mapped[UserRole] = mapped_column(default=UserRole.REGULAR)
    vpn_link: Mapped[str]
    datetime: Mapped[datetime]

    def is_regular(self) -> bool:
        return self.role >= UserRole.REGULAR

    def is_admin(self) -> bool:
        return self.role == UserRole.ADMIN