from sqlalchemy import func

from calculators import Calculator
from commons.models.data_fujian import DataFujian
from commons.models.data_network import DataNetwork
from commons.models.fujian_survey import FujianSurvey
from commons.models.fuzhou_highway_bureau import FuzhouHighwayBureau
from commons.models.fuzhou_transportation_bureau import FuzhouTransportationBureau


class AsphaltDomesticCalculator(Calculator):
    name = "国产沥青"
    material_id = "3001001001"
    unit = "t"
    spec = ""

    def _get_ftb_price(self):
        query = FuzhouTransportationBureau.get_query(self.year, self.month, name='石油沥青', spec='国产')
        query = query.with_entities(func.avg(FuzhouTransportationBureau.price))
        data = query.first()
        price = int(data[0]) if data[0] else 0

        fluctuating = int(self._get_fluctuating('price_ftb', price))
        return price, fluctuating

    def _get_fhb_price(self):
        query = FuzhouHighwayBureau.get_query(self.year, self.month, name='石油沥青', spec='国产')
        query = query.with_entities(func.avg(FuzhouHighwayBureau.price))
        data = query.first()
        price = int(data[0]) if data[0] else 0
        fluctuating = int(self._get_fluctuating('price_fhb', price))
        return price, fluctuating

    def _get_network_price(self):
        query = DataNetwork.get_query(self.year, self.month, name='石油沥青', spec='国产')
        data = query.first()
        price = data.price if data else 0
        fluctuating = int(self._get_fluctuating('price_network', price))
        return price, fluctuating

    def _get_survey_price(self):
        query = FujianSurvey.get_query(self.year, self.month, name='石油沥青', spec='国产')
        query = query.with_entities(func.avg(FujianSurvey.price))
        data = query.first()
        price = int(data[0]) if data[0] else 0
        fluctuating = int(self._get_fluctuating('price_survey', price))
        return price, fluctuating

    def _get_fujian_price(self):
        query = DataFujian.get_query(self.year, self.month, name='石油沥青')
        data = query.first()
        price = data.price * 1000 if data else 0
        fluctuating = self._get_fluctuating('price_fujian', price)
        return price, fluctuating

    def _get_recommend_price(self):
        self._fluctuatings = [self.fluctuating_network, self.fluctuating_survey]
        return super()._get_recommend_price()


if __name__ == '__main__':
    from core.factory import ClientApp

    with ClientApp().app_context():
        calculator = AsphaltDomesticCalculator(year=2023, month=12)
        _result = calculator.run()
        calculator.save()
        print(_result)