Add initial database migration

1 files changed, 15 insertions, 0 deletions

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