init

web/calculators/steel_rebar_300.py

@@ -0,0 +1,88 @@
+from sqlalchemy import func
+
+from calculators import Calculator
+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 Reber300Calculator(Calculator):
+    name = "光圆钢筋"
+    material_id = "15.05.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='光圆钢筋')
+        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 - getattr(self.previous_prices, 'ftb_price', price)
+        return price, fluctuating
+
+    def _get_ss_price(self):
+        query = SanmingSteel.get_query(self.year, self.month, name='高线', spec='φ10mm')
+        query = query.with_entities(func.avg(SanmingSteel.price))
+        result = query.all()
+        if not result[0][0]:
+            return 0, 0
+        price = int(result[0][0])
+        fluctuating = price - getattr(self.previous_prices, 'ss_price', price)
+        return price, fluctuating
+
+    def _get_fhb_price(self):
+        query = FuzhouHighwayBureau.get_query(self.year, self.month, name='光圆钢筋')
+        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 - getattr(self.previous_prices, 'ftb_price', price)
+        return price, fluctuating
+
+    def _get_network_price(self):
+        result = SteelRebar.get_items(self.year, self.month, 'HPB300')
+        price_dict = {spec: price for spec, material, price in result}
+        price = 0.3 * float(price_dict.get('Φ6', 0)) + 0.7 * float(price_dict.get('Φ8-10', 0))
+        price = round(price)
+
+        fluctuating = price - getattr(self.previous_prices, 'price_network', price)
+        return price, fluctuating
+
+    def _get_survey_price(self):
+        query = FujianSurvey.get_query(self.year, self.month, name='光圆钢筋')
+        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 - getattr(self.previous_prices, '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 = Reber300Calculator(year=2023, month=10)
+        result = calculator.run()
+        calculator.save()
+        print(result)
+