mirror of
https://github.com/dancojocaru2000/foxbank.git
synced 2025-02-23 12:29:35 +02:00
84 lines
2.1 KiB
Python
84 lines
2.1 KiB
Python
from dataclasses import dataclass
|
|
from marshmallow import Schema, fields
|
|
|
|
@dataclass
|
|
class User:
|
|
id: int
|
|
username: str
|
|
email: str
|
|
otp: str
|
|
fullname: str
|
|
|
|
class UserSchema(Schema):
|
|
id = fields.Int(required=False)
|
|
username = fields.String()
|
|
email = fields.String()
|
|
otp = fields.String(load_only=True, required=False)
|
|
fullname = fields.String()
|
|
|
|
@staticmethod
|
|
def new_user(username: str, email: str, fullname: str) -> 'User':
|
|
return User(
|
|
id=-1,
|
|
username=username,
|
|
email=email,
|
|
otp='',
|
|
fullname=fullname,
|
|
)
|
|
|
|
def to_json(self, include_otp=False, include_id=False):
|
|
result = {
|
|
'username': self.username,
|
|
'email': self.email,
|
|
'fullname': self.fullname,
|
|
}
|
|
if include_id:
|
|
result['id'] = self.id
|
|
if include_otp:
|
|
result['otp'] = self.otp
|
|
return result
|
|
|
|
@classmethod
|
|
def from_query(cls, query_result):
|
|
return cls(*query_result)
|
|
|
|
|
|
@dataclass
|
|
class Account:
|
|
id: int
|
|
iban: str
|
|
currency: str
|
|
account_type: str
|
|
custom_name: str
|
|
|
|
class Schema(Schema):
|
|
id = fields.Int(required=False)
|
|
iban = fields.Str()
|
|
currency = fields.Str()
|
|
account_type = fields.Str(data_key='accountType')
|
|
custom_name = fields.Str(data_key='customName')
|
|
|
|
@staticmethod
|
|
def new_account(currency: str, account_type: str, custom_name: str = '') -> 'Account':
|
|
return Account(
|
|
id=-1,
|
|
iban='',
|
|
currency=currency,
|
|
account_type=account_type,
|
|
custom_name=custom_name,
|
|
)
|
|
|
|
def to_json(self, include_id=True):
|
|
result = {
|
|
'iban': self.iban,
|
|
'currency': self.currency,
|
|
'accountType': self.account_type,
|
|
'customName': self.custom_name,
|
|
}
|
|
if include_id:
|
|
result['id'] = self.id
|
|
return result
|
|
|
|
@classmethod
|
|
def from_query(cls, query_result):
|
|
return cls(*query_result)
|