From bc7f486aa7b543a934f4cf23dc80a95f44afcb64 Mon Sep 17 00:00:00 2001 From: Tolmachev Igor Date: Mon, 23 Mar 2026 00:07:08 +0300 Subject: Add initial database migration --- models/user.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) (limited to 'models/user.py') diff --git a/models/user.py b/models/user.py index 7118725..4983a13 100644 --- a/models/user.py +++ b/models/user.py @@ -1,9 +1,24 @@ +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] + + def is_regular(self) -> bool: + return self.role >= UserRole.REGULAR + + def is_admin(self) -> bool: + return self.role == UserRole.ADMIN -- cgit v1.3