from database import MySqlDatabaseHelper

class MySqlDao:
    def __init__(self):
        self.db_helper = MySqlDatabaseHelper()
        
    def load_product_data(self, city_uuid):
        """从数据库中读取商品信息"""
        tablename = "tads_brandcul_product_info"
        
        query = f"SELECT * FROM {tablename} WHERE city_uuid = :city_uuid"
        params = {"city_uuid": city_uuid}
        
        data = self.db_helper.load_data_with_page(query, params)
        return data
        
    def load_cust_data(self, city_uuid):
        """从数据库中读取商户信息"""
        tablename = "tads_brandcul_cust_info"
        
        query = f"SELECT * FROM {tablename} WHERE BA_CITY_ORG_CODE = :city_uuid"
        params = {"city_uuid": city_uuid}
        
        data = self.db_helper.load_data_with_page(query, params)
        return data
    
    def load_order_data(self, city_uuid):
        """从数据库中读取订单信息"""
        tablename = "tads_brandcul_cust_order"

        query = f"SELECT * FROM {tablename} WHERE city_uuid = :city_uuid"
        params = {"city_uuid": city_uuid}
        
        data = self.db_helper.load_data_with_page(query, params)
        
        data.drop('stat_month', axis=1, inplace=True)
        data.drop('city_uuid', axis=1, inplace=True)
        
        # 去除重复值和填补缺失值
        data.drop_duplicates(inplace=True)
        data.fillna(0, inplace=True)
        data = data.infer_objects(copy=False)
        return data
    
    def load_mock_order_data(self, city_uuid):
        """从数据库中读取mock的订单信息"""
        tablename = "yunfu_mock_data"

        query = f"SELECT * FROM {tablename}"
        params = {"city_uuid": city_uuid}
        
        data = self.db_helper.load_data_with_page(query, params)
        
        # 去除重复值和填补缺失值
        data.drop_duplicates(inplace=True)
        data.fillna(0, inplace=True)
        data = data.infer_objects(copy=False)
        
        return data
        
if __name__ == "__main__":
    dao = MySqlDao()
    city_uuid = "00000000000000000000000011445301"
    # city_uuid = "00000000000000000000000011441801"
    dao.load_mock_order_data(city_uuid)