mdbq 4.1.6__tar.gz → 4.1.7__tar.gz

{mdbq-4.1.6 → mdbq-4.1.7}/PKG-INFO

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

mdbq-4.1.7/mdbq/__version__.py

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

{mdbq-4.1.6 → mdbq-4.1.7}/mdbq/mysql/uploader.py

@@ -2297,6 +2297,47 @@ class MySQLUploader:
         return validated_keys
 
 
+def process_df_columns(
+        df: pd.DataFrame, 
+        columns: List[str], 
+        default_value: Any = 0
+) -> pd.DataFrame:
+    """
+    处理DataFrame的列，补齐缺失的列并丢弃多余的列
+    
+    :param df: 要处理的DataFrame
+    :param columns: 所需的列名列表，注意不处理大小写
+    :param default_value: 缺失列的填充值，默认为None
+    :return: 处理后的DataFrame
+    """
+    if df is None or not isinstance(df, pd.DataFrame) or not isinstance(columns, list) or not columns:
+        return df
+    
+    # 获取当前列名
+    current_columns = list(df.columns)
+    
+    # 找出需要添加的列和需要删除的列
+    missing_columns = [col for col in columns if col not in current_columns]
+    extra_columns = [col for col in current_columns if col not in columns]
+    
+    # 复制DataFrame
+    result_df = df.copy()
+    
+    # 删除多余的列
+    if extra_columns:
+        result_df = result_df.drop(columns=extra_columns)
+    
+    # 添加缺失的列
+    if missing_columns:
+        for col in missing_columns:
+            result_df[col] = default_value
+    
+    # 按照指定顺序重新排列列
+    result_df = result_df.reindex(columns=columns)
+    
+    return result_df
+
+
 def main():
     dir_path = os.path.expanduser("~")
     parser = myconf.ConfigParser()

{mdbq-4.1.6 → mdbq-4.1.7}/mdbq.egg-info/PKG-INFO

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

mdbq-4.1.6/mdbq/__version__.py

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