material-api/web/calculators/oil_92.py

import datetime

from calculators import Calculator, Helper
from commons.models.data_network import DataNetwork
from commons.models.fujian_survey import FujianSurvey
from commons.models.oil import Oil

from commons.models.price_result import PriceResult


class Oil92Calculator(Calculator):
    name = "汽油（92）"
    material_id = "69.60.61.00"
    unit = "kg"
    spec = "92#"

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

    def _get_survey_price(self):
        query = FujianSurvey.get_query(self.year, self.month, name='汽油', spec='92#')
        data = query.first()
        price = data.price if data else 0
        fluctuating = self._get_fluctuating('price_survey', price)
        return round(price, 2), round(fluctuating, 2)

    def _get_network_price(self):
        query = DataNetwork.get_query(self.year, self.month, name='汽油', spec='92号')
        data = query.first()
        price = data.price if data else 0
        fluctuating = self._get_fluctuating('price_network', price)
        return round(price, 2), round(fluctuating, 2)

    def _get_calculate_price(self, round_dit=2):
        return super()._get_calculate_price(round_dit)

    def _get_recommend_price(self):
        return self.price_network, self.fluctuating_network

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


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

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