mdbq 1.1.8__tar.gz → 1.1.9__tar.gz

{mdbq-1.1.8 → mdbq-1.1.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mdbq
-Version: 1.1.8
+Version: 1.1.9
 Home-page: https://pypi.org/project/mdbsql
 Author: xigua, 
 Author-email: 2587125111@qq.com

{mdbq-1.1.8 → mdbq-1.1.9}/mdbq/aggregation/aggregation.py

@@ -935,7 +935,7 @@ def one_file_to_mysql(file, db_name, table_name, target_service, database):
     m.df_to_mysql(df=df, db_name=db_name, table_name=table_name, filename=filename)
 
 
-def file_dir():
+def file_dir(one_file=True):
     """
     按照文件记录对照表
     批量上传数据库
@@ -954,20 +954,26 @@ def file_dir():
             if platform.system() == 'Windows':
                 sub_path = sub_path.replace('/', '\\')
             # print(os.path.join(path, sub_path), db_name, table_name)
-            # 从每个文件夹中取出一个文件
-            real_path_list = []
-            for root, dirs, files in os.walk(os.path.join(path, sub_path), topdown=False):
-                for name in files:
-                    if name.endswith('.csv') and 'baidu' not in name and '~' not in name:
-                        real_path_list.append(os.path.join(root, name))
-                        break
-            for real_path in real_path_list:
-                one_file_to_mysql(
-                    file=real_path,
-                    db_name=db_name,
-                    table_name=table_name,
-                    target_service='home_lx',
-                    database='mysql'
+            if one_file:  # 从每个文件夹中取出一个文件
+                real_path_list = []
+                for root, dirs, files in os.walk(os.path.join(path, sub_path), topdown=False):
+                    for name in files:
+                        if name.endswith('.csv') and 'baidu' not in name and '~' not in name:
+                            real_path_list.append(os.path.join(root, name))
+                            break
+                for real_path in real_path_list:
+                    one_file_to_mysql(
+                        file=real_path,
+                        db_name=db_name,
+                        table_name=table_name,
+                        target_service='home_lx',
+                        database='mysql'
+                    )
+            else:  # 上传全部文件夹
+                upload(
+                    path=os.path.join(path, sub_path),
+                    db_name = db_name,
+                    collection_name = table_name,
                 )
             data.update({'入库进度': 1})  # 更新进度
     df = pd.DataFrame.from_dict(datas, orient='columns')
@@ -995,4 +1001,4 @@ if __name__ == '__main__':
     #     db_name = '生意经2',
     #     collection_name = 'e3_零售明细统计',
     # )
-    file_dir()
+    file_dir(one_file=True)

{mdbq-1.1.8 → mdbq-1.1.9}/mdbq.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mdbq
-Version: 1.1.8
+Version: 1.1.9
 Home-page: https://pypi.org/project/mdbsql
 Author: xigua, 
 Author-email: 2587125111@qq.com

{mdbq-1.1.8 → mdbq-1.1.9}/setup.py

@@ -3,7 +3,7 @@
 from setuptools import setup, find_packages
 
 setup(name='mdbq',
-      version='1.1.8',
+      version='1.1.9',
       author='xigua, ',
       author_email="2587125111@qq.com",
       url='https://pypi.org/project/mdbsql',