metric_by_sql_general.py 652 B

1234567891011121314151617181920212223242526
  1. # -*- coding: utf-8 -*-
  2. """
  3. @author: yq
  4. @time: 2024/10/31
  5. @desc: sql直接统计指标直出
  6. """
  7. import pandas as pd
  8. from data import DataLoaderBase
  9. from .metric_base import MetricBase
  10. class MetricBySqlGeneral(MetricBase):
  11. def __init__(self, metric_name: str, metric_code: str):
  12. super().__init__(metric_name, metric_code)
  13. def validate_data(self):
  14. pass
  15. def load_data(self, data_loader: DataLoaderBase, sql: str) -> pd.DataFrame:
  16. data = data_loader.get_data(sql)
  17. return data
  18. def calculate(self, data_loader: DataLoaderBase, sql: str) -> pd.DataFrame:
  19. return self.load_data(data_loader, sql)