from sqlalchemy import func

from calculators import Calculator
from commons.models.cement import Cement
from commons.models.fujian_survey import FujianSurvey
from commons.models.fuzhou_highway_bureau import FuzhouHighwayBureau
from commons.models.fuzhou_transportation_bureau import FuzhouTransportationBureau

from commons.models.price_result import PriceResult


class Cement325Calculator(Calculator):
    name = "32.5 水泥"
    material_id = "09.60.60.00"
    unit = "t"
    spec = ""

    def __init__(self, year, month):
        self.year = year
        self.month = month

    def _get_ftb_price(self):
        query = FuzhouTransportationBureau.get_query(self.year, self.month, name='32.5级水泥')
        query = query.with_entities(func.avg(FuzhouTransportationBureau.price))
        result = query.all()
        if not result[0][0]:
            return 0, 0
        price = int(result[0][0])
        fluctuating = price - int(getattr(self.previous_prices, 'ftb_price', price))
        return price, fluctuating

    def _get_fhb_price(self):
        query = FuzhouHighwayBureau.get_query(self.year, self.month, name='32.5级水泥', spec='旋窑（散装）')
        query = query.with_entities(func.avg(FuzhouHighwayBureau.price))
        result = query.all()
        if not result[0][0]:
            return 0, 0
        price = int(result[0][0])
        fluctuating = price - int(getattr(self.previous_prices, 'ftb_price', price))
        return price, fluctuating

    def _get_network_price(self):
        result = Cement.get_items(self.year, self.month, spec_in=('台泥', '金牛', '炼石牌'), name_in=('P.S.A32.5', 'M32.5', 'P.P 32.5R'), pack='散装')
        prices = result[0][0]
        if not result[0][0]:
            return 0, 0
        price = round(prices / 5) * 5
        fluctuating = price - int(getattr(self.previous_prices, 'price_network', price))
        return price, fluctuating

    def _get_survey_price(self):
        query = FujianSurvey.get_query(self.year, self.month, name='32.5级水泥', spec='旋窑')
        query = query.with_entities(func.avg(FujianSurvey.price))
        result = query.all()
        if not result[0][0]:
            return 0, 0
        price = int(result[0][0])
        fluctuating = price - int(getattr(self.previous_prices, 'price_survey', price))
        return price, fluctuating

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

    def save(self):
        result = self.result()
        PriceResult(**result).upsert()


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

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