feat(model): Added correct data models

This commit is contained in:
Kita Trofimov
2025-10-26 00:11:20 +03:00
committed by Sweetbread
parent 70041fa58e
commit e80c5645f1
5 changed files with 56 additions and 2 deletions
+14 -2
View File
@@ -1,13 +1,16 @@
from dataclasses import dataclass
import json
from datetime import datetime
from utils.token import generateKey
@dataclass
class User:
id: int
email: str
password_hash: str
name: str
role: str
token: str
created_at: datetime
def __init__(self, email: str, passwd: str):
#us = getUsModel() #возвращает словарь
@@ -18,3 +21,12 @@ class User:
def toJson(self):
return {"user": {"id": self.id, "name": self.name, "role": self.role}, "token": self.token}
def is_admin(self) -> bool:
return self.role == 'admin'
def is_operator(self) -> bool:
return self.role == 'operator'
def is_viewer(self) -> bool:
return self.role == 'viewer'