from tortoise import fields, models
from pydantic import BaseModel, ConfigDict
from tortoise.contrib.pydantic.creator import pydantic_model_creator, pydantic_queryset_creator
from tortoise.queryset import QuerySet

class UTTFund(models.Model):
    id = fields.IntField(pk=True)
    symbol = fields.CharField(max_length=20, unique=True)
    name = fields.CharField(max_length=100)

    @staticmethod
    async def get_list(data):
        if type(data) == type([UTTFund]):
            parser=pydantic_queryset_creator(UTTFund)
            return await parser.from_queryset(data)

    async def to_dict(self):
        if type(self) == UTTFund:
            parser=pydantic_model_creator(UTTFund)
            return await parser.from_tortoise_orm(self)


class UTTFundData(models.Model):
    id = fields.IntField(pk=True)
    fund = fields.ForeignKeyField("models.UTTFund", related_name="data")
    date = fields.DateField()
    nav_per_unit = fields.FloatField()
    sale_price_per_unit = fields.FloatField()
    repurchase_price_per_unit = fields.FloatField()
    outstanding_number_of_units = fields.BigIntField()
    net_asset_value = fields.BigIntField()

    @classmethod
    async def get_list(data):
        if type(data) == QuerySet:
            parser=pydantic_queryset_creator(UTTFundData)
            return await parser.from_queryset(data)


    async def to_dict(self):
        if type(self) == models.Model:
            parser=pydantic_model_creator(UTTFundData)
            return await parser.from_tortoise_orm(self)



    class Meta:
        unique_together = ("fund", "date")