mdbq 4.0.64__tar.gz → 4.0.66__tar.gz

{mdbq-4.0.64 → mdbq-4.0.66}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: mdbq
-Version: 4.0.64
+Version: 4.0.66
 Home-page: https://pypi.org/project/mdbq
 Author: xigua, 
 Author-email: 2587125111@qq.com

mdbq-4.0.66/mdbq/__version__.py

@@ -0,0 +1 @@
+VERSION = '4.0.66'

{mdbq-4.0.64 → mdbq-4.0.66}/mdbq/mysql/s_query.py

@@ -779,6 +779,82 @@ class QueryDatas:
                 df[col] = df[col].astype(float)
         return df
 
+    def _convert_columns_to_lowercase(self, data: Union[pd.DataFrame, List[Dict[str, Any]]], 
+                                    lower_col: Optional[List[str]], 
+                                    return_format: str) -> Union[pd.DataFrame, List[Dict[str, Any]]]:
+        """
+        将指定列的值转换为小写。
+        
+        Args:
+            data: 原始数据，可以是DataFrame或列表字典
+            lower_col: 需要转换为小写的列名列表
+            return_format: 返回数据格式
+            
+        Returns:
+            转换后的数据
+        """
+        # 参数验证
+        if not lower_col:
+            return data
+        
+        # 检查data是否为空
+        if return_format == 'df':
+            if data is None or data.empty:
+                return data
+        else:  # list_dict格式
+            if data is None or len(data) == 0:
+                return data
+            
+        # 确保 lower_col 是列表类型
+        if not isinstance(lower_col, list):
+            logger.warning('lower_col 参数类型错误，应为列表', {'传入类型': type(lower_col).__name__})
+            return data
+            
+        try:
+            if return_format == 'df':
+                df = data.copy()
+                for col in lower_col:
+                    if not isinstance(col, str):
+                        logger.warning('列名必须是字符串类型', {'列名': col, '类型': type(col).__name__})
+                        continue
+                        
+                    if col in df.columns:
+                        # 只对字符串类型的列进行小写转换
+                        if df[col].dtype == 'object':
+                            # 更安全的空值处理：保持 None/NaN 不变，只转换非空字符串
+                            df[col] = df[col].apply(lambda x: str(x).lower() if pd.notna(x) and x is not None else x)
+                        else:
+                            logger.debug('列不是字符串类型，跳过小写转换', {'列名': col, '数据类型': df[col].dtype})
+                        logger.debug('列转换为小写', {'列名': col, '数据类型': df[col].dtype})
+                    else:
+                        logger.debug('指定的列不存在，跳过小写转换', {'列名': col, '可用列': list(df.columns)})
+                return df
+            else:  # list_dict格式
+                result = []
+                for row in data:
+                    new_row = row.copy()
+                    for col in lower_col:
+                        if not isinstance(col, str):
+                            logger.warning('列名必须是字符串类型', {'列名': col, '类型': type(col).__name__})
+                            continue
+                            
+                        if col in new_row and new_row[col] is not None:
+                            # 确保值是字符串类型再转换为小写
+                            try:
+                                new_row[col] = str(new_row[col]).lower()
+                            except Exception as e:
+                                logger.debug('值转换为小写失败', {'列名': col, '值': new_row[col], '错误': str(e)})
+                                # 保持原值不变
+                    result.append(new_row)
+                return result
+        except Exception as e:
+            logger.warning('小写转换失败', {
+                '错误类型': type(e).__name__,
+                '错误信息': str(e),
+                '指定列': lower_col
+            })
+            return data
+
     def __enter__(self):
         """上下文管理器入口"""
         return self
@@ -831,7 +907,8 @@ class QueryDatas:
             limit: Optional[int] = None,
             page_size: Optional[int] = None,
             date_column: Optional[str] = None,
-            return_format: Literal['df', 'list_dict'] = 'df'
+            return_format: Literal['df', 'list_dict'] = 'df',
+            lower_col: Optional[List[str]] = ['店铺名称']
             ) -> Union[pd.DataFrame, List[Dict[str, Any]]]:
         """
         从数据库表获取数据，支持列筛选、日期范围过滤和行数限制。
@@ -855,6 +932,7 @@ class QueryDatas:
             return_format: 返回数据格式
                          - 'df': 返回pandas DataFrame（默认）
                          - 'list_dict': 返回列表字典格式 [{列1:值, 列2:值, ...}, ...]
+            lower_col: 需要转换为小写的列名列表，默认['店铺名称']。如果为None或空列表，则不进行小写转换。
             
         Returns:
             根据return_format参数返回不同格式的数据：
@@ -874,6 +952,18 @@ class QueryDatas:
             logger.error('无效的return_format值', {'库': db_name, '表': table_name, '指定返回数据格式, 有效值应为: ': ', '.join(valid_formats)})
             return [] if return_format == 'list_dict' else pd.DataFrame()
 
+        # 验证lower_col参数
+        if lower_col is not None:
+            if not isinstance(lower_col, list):
+                logger.warning('lower_col 参数类型错误，应为列表，将使用默认值', {'传入类型': type(lower_col).__name__})
+                lower_col = ['店铺名称']
+            elif len(lower_col) > 0:
+                # 验证列表中的每个元素都是字符串
+                invalid_items = [item for item in lower_col if not isinstance(item, str)]
+                if invalid_items:
+                    logger.warning('lower_col 列表包含非字符串元素，将过滤掉', {'无效元素': invalid_items})
+                    lower_col = [item for item in lower_col if isinstance(item, str)]
+
         # 验证日期范围
         start_date, end_date = self._validate_date_range(start_date, end_date, db_name, table_name)
         
@@ -928,10 +1018,12 @@ class QueryDatas:
                         
                         if result:
                             if return_format == 'list_dict':
+                                result = self._convert_columns_to_lowercase(result, lower_col, return_format)
                                 return result
                             else:
                                 df = pd.DataFrame(result)
                                 df = self._convert_decimal_columns(df)
+                                df = self._convert_columns_to_lowercase(df, lower_col, return_format)
                                 return df
                         return [] if return_format == 'list_dict' else pd.DataFrame()
 
@@ -975,11 +1067,13 @@ class QueryDatas:
                     })
 
                     if return_format == 'list_dict':
+                        all_results = self._convert_columns_to_lowercase(all_results, lower_col, return_format)
                         return all_results
                     else:
                         df = pd.DataFrame(all_results)
                         if not df.empty:
                             df = self._convert_decimal_columns(df)
+                            df = self._convert_columns_to_lowercase(df, lower_col, return_format)
                         return df
 
         except Exception as e:

{mdbq-4.0.64 → mdbq-4.0.66}/mdbq.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: mdbq
-Version: 4.0.64
+Version: 4.0.66
 Home-page: https://pypi.org/project/mdbq
 Author: xigua, 
 Author-email: 2587125111@qq.com

mdbq-4.0.64/mdbq/__version__.py

@@ -1 +0,0 @@
-VERSION = '4.0.64'