Pārlūkot izejas kodu

fix: 统一 get_cust_list 和 get_product_from_order 的接口

- 改用 load_data_with_page 替代直接调用 fetch_all
- 与其他查询方法保持一致
Sherlock 2 dienas atpakaļ
vecāks
revīzija
dc63195bc8
1 mainītis faili ar 6 papildinājumiem un 6 dzēšanām
  1. 6 6
      database/dao/mysql_dao.py

+ 6 - 6
database/dao/mysql_dao.py

@@ -252,10 +252,10 @@ class MySqlDao:
         query = f"SELECT DISTINCT product_code FROM {self._order_tablename} WHERE city_uuid = :city_uuid"
         params = {"city_uuid": city_uuid}
         
-        data = pd.DataFrame(self.db_helper.fetch_all(text(query), params))
-        
+        data = self.db_helper.load_data_with_page(query, params)
+
         return data
-    
+
     # def get_product_from_order(self, city_uuid):
     #     query = f"SELECT cust_code, product_code FROM {self._order_tablename} WHERE city_uuid = :city_uuid"
     #     params = {"city_uuid": city_uuid}
@@ -273,10 +273,10 @@ class MySqlDao:
         query = f"SELECT DISTINCT BB_RETAIL_CUSTOMER_CODE FROM {self._cust_tablename} WHERE BA_CITY_ORG_CODE = :city_uuid"
         params = {"city_uuid": city_uuid}
         
-        data = pd.DataFrame(self.db_helper.fetch_all(text(query), params))
-        
+        data = self.db_helper.load_data_with_page(query, params)
+
         return data
-    
+
     def data_preprocess(self, data: pd.DataFrame):
         """数据预处理"""
         data.drop(["cust_uuid", "longitude", "latitude", "range_radius"], axis=1, inplace=True)