File size: 1,502 Bytes
9d4bd7c e558c69 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
from tortoise import fields, models
from tortoise.contrib.pydantic.creator import pydantic_model_creator, pydantic_queryset_creator
from tortoise.queryset import QuerySet
class User(models.Model):
id = fields.IntField(pk=True)
username = fields.CharField(max_length=50, unique=True)
email = fields.CharField(max_length=100, unique=True)
hashed_password = fields.CharField(max_length=128)
created_at = fields.DatetimeField(auto_now_add=True)
@staticmethod
async def get_list(data):
if type(data) == QuerySet:
parser=pydantic_queryset_creator(User)
return await parser.from_queryset(data)
async def to_dict(self):
if type(self) == User:
parser=pydantic_model_creator(User)
return await parser.from_tortoise_orm(self)
class Watchlist(models.Model):
id = fields.IntField(pk=True)
user = fields.ForeignKeyField("models.User", related_name="watchlist")
stock = fields.ForeignKeyField("models.Stock", null=True, related_name="watching")
utt = fields.ForeignKeyField("models.UTTFund", null=True, related_name="watching")
@staticmethod
async def get_list(data):
if type(data) == QuerySet:
parser=pydantic_queryset_creator(Watchlist)
return await parser.from_queryset(data)
async def to_dict(self):
if type(self) == models.Model:
parser=pydantic_model_creator(Watchlist)
return await parser.from_tortoise_orm(self) |