from sqlalchemy import func

from calculators import Calculator
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
from commons.models.price_result import PriceResult
from commons.models.sanming_steel import SanmingSteel
from commons.models.steel_rebar import SteelRebar


class Reber400Calculator(Calculator):
    name = "带肋钢筋"
    material_id = "15.03.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='带肋钢筋')
        data = query.first()
        price = int(data.price) if data else 0
        fluctuating = int(self._get_fluctuating('price_ftb', price))
        return price, fluctuating

    def _get_ss_price(self):
        query = SanmingSteel.get_query(self.year, self.month, name='Ⅲ级螺纹钢筋', spec='HRB400Ф16-25')
        data = query.first()
        price = int(data.price) if data else 0
        fluctuating = int(self._get_fluctuating('price_ss', price))
        return price, fluctuating

    def _get_fhb_price(self):
        query = FuzhouHighwayBureau.get_query(self.year, self.month, name='带肋钢筋')
        data = query.first()
        price = int(data.price) if data else 0
        fluctuating = int(self._get_fluctuating('price_ftb', price))
        return price, fluctuating

    def _get_network_price(self):
        query = DataNetwork.get_query(self.year, self.month, name='带肋钢筋')  # spec='综合' 部分历史表格没有填写规格
        data = query.first()
        price = int(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='带肋钢筋')
        data = query.first()
        price = int(data.price) if data else 0
        fluctuating = int(self._get_fluctuating('price_survey', price))
        return price, fluctuating

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

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


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

    with ClientApp().app_context():
        calculator = Reber400Calculator(year=2024, month=5)
        result = calculator.run()
        calculator.save()
        print(result)