feat(model): Added correct data models
This commit is contained in:
+14
-2
@@ -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'
|
||||
|
||||
Reference in New Issue
Block a user