|
|
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) |
