feat(model): Added correct data models

This commit is contained in:
Kita Trofimov
2025-10-26 00:11:20 +03:00
parent 7ccc9385e6
commit 79cac3d8cb
5 changed files with 55 additions and 14 deletions
+13 -14
View File
@@ -1,22 +1,21 @@
from dataclasses import dataclass
import json
from utils.token import generateKey
from datetime import datetime
@dataclass
class user:
class User:
id: int
email: str
password_hash: str
name: str
role: str
token: str
created_at: datetime
@classmethod
def initialize(cls, email:str, passwd:str):
#us = getUsModel() #возвращает словарь
id = 1#us['id']
name = 'Bob'#us['name']
role = 'Backend'#us['role']
token = generateKey(email, passwd)
return cls(id=id, name=name, role=role, token=token)
def is_admin(self) -> bool:
return self.role == 'admin'
def toJSON(self):
return json.dumps({"token": f'{self.token}', "user": {"id": self.id, "role": f'{self.role}', "name": f'{self.name}'}})
def is_operator(self) -> bool:
return self.role == 'operator'
def is_viewer(self) -> bool:
return self.role == 'viewer'