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