Files
material-api/web/calculators/cement_325.py
2024-07-12 17:13:02 +08:00

78 lines
2.8 KiB
Python

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
from commons.models.price_result import PriceResult
class Cement325Calculator(Calculator):
name = "32.5 水泥"
material_id = "09.60.60.00"
unit = "t"
spec = ""
def _get_ftb_price(self):
query = FuzhouTransportationBureau.get_query(self.year, self.month, name='32.5级水泥')
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='32.5级水泥', 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='32.5')
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='32.5级水泥', 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='水泥', spec='32.5')
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, 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)