71 lines
2.5 KiB
Python
71 lines
2.5 KiB
Python
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
|
|
from commons.models.price_result import PriceResult
|
|
|
|
|
|
class SteelSectionCalculator(Calculator):
|
|
name = "型钢"
|
|
material_id = "2003004"
|
|
unit = "t"
|
|
spec = ""
|
|
|
|
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):
|
|
# return 4000, -10
|
|
|
|
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_fhb', price))
|
|
return price, fluctuating
|
|
|
|
def _get_network_price(self):
|
|
query = DataNetwork.get_query(self.year, self.month, name='型钢')
|
|
data = query.first()
|
|
price = round(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_fujian_price(self):
|
|
query = DataFujian.get_query(self.year, self.month, name='型钢')
|
|
data = query.first()
|
|
price = data.price 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, self.fluctuating_fhb, self.fluctuating_ftb]
|
|
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 = SteelSectionCalculator(year=2024, month=5)
|
|
result = calculator.run()
|
|
calculator.save()
|
|
print(result)
|