mdbq 1.0.1__tar.gz → 1.0.2__tar.gz

{mdbq-1.0.1 → mdbq-1.0.2}/PKG-INFO

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

{mdbq-1.0.1 → mdbq-1.0.2}/mdbq/aggregation/mysql_types.py

@@ -136,19 +136,19 @@ class DataTypes:
                     return self.datas[cl][db_name][tabel_name]
                 else:
                     print(f'不存在的集合名信息: {tabel_name}, 文件位置: {self.json_file}')
-                    mysql_all_dtypes()  # 更新 mysql 中所有数据库的 dtypes 信息到本地 json
+                    mysql_all_dtypes(db_name=db_name, table_name=tabel_name)   # 更新一个表的 dtypes
                     return {}
             else:
                 print(f'不存在的数据库信息: {db_name}, 文件位置: {self.json_file}')
-                mysql_all_dtypes()  # 更新 mysql 中所有数据库的 dtypes 信息到本地 json
+                mysql_all_dtypes(db_name=db_name)  # 更新一个数据库的 dtypes
                 return {}
         else:
             print(f'不存在的数据分类: {cl}, 文件位置: {self.json_file}')
-            mysql_all_dtypes()  # 更新 mysql 中所有数据库的 dtypes 信息到本地 json
+            mysql_all_dtypes()  # 更新所有数据库所有数据表的 dtypes 信息到本地 json
             return {}
 
 
-def mysql_all_dtypes(path=None):
+def mysql_all_dtypes(db_name=None, table_name=None, path=None):
     """
     更新笔记本 mysql 中所有数据库的 dtypes 信息到本地 json
     """
@@ -186,18 +186,18 @@ def mysql_all_dtypes(path=None):
     #     '聚合数据',
     # ]
     results = []
-    for db_name in db_name_lists:
-        config.update({'database': db_name})  # 添加更新 config 字段
+    for db_ in db_name_lists:
+        config.update({'database': db_})  # 添加更新 config 字段
         connection = pymysql.connect(**config)  # 连接数据库
         try:
             with connection.cursor() as cursor:
-                sql = f"SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = '{db_name}';"
+                sql = f"SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = '{db_}';"
                 sql = "SHOW TABLES;"
                 cursor.execute(sql)
                 table_name = cursor.fetchall()
                 for table in table_name:
                     for k, v in table.items():
-                        results.append({db_name: v})
+                        results.append({db_: v})
         except:
             pass
         finally:
@@ -206,21 +206,28 @@ def mysql_all_dtypes(path=None):
 
     d = DataTypes()
     for result in results:
-        for db_name, tabel_name in result.items():
-            print(f'获取列信息 数据库: < {db_name} >, 数据表: < {tabel_name} >')
-            # d.mysql_dtypes_to_json(db_name=db_name, tabel_name=tabel_name, path=path)
+        for db_n, tabel_n in result.items():
+            if db_name and table_name:  # 下载一个指定的数据表
+                if db_name != db_n or tabel_name != tabel_n:
+                    continue
+            elif db_name:  # 下载一个数据库的所有数据表
+                if db_name != db_n:
+                    continue
+            # 如果 db_name 和 table_name 都不指定，则下载所有数据库的所有数据表
+            print(f'获取列信息 数据库: < {db_n} >, 数据表: < {tabel_n} >')
+            # d.mysql_dtypes_to_json(db_name=db_n, tabel_name=tabel_n, path=path)
             sq = s_query.QueryDatas(username=username, password=password, host=host, port=port)
             # 获取数据表的指定列, 返回列表
             # [{'视频bv号': 'BV1Dm4y1S7BU', '下载进度': 1}, {'视频bv号': 'BV1ov411c7US', '下载进度': 1}]
-            name_type = sq.dtypes_to_list(db_name=db_name, tabel_name=tabel_name)
+            name_type = sq.dtypes_to_list(db_name=db_n, tabel_name=tabel_n)
             if name_type:
                 dtypes = {item['COLUMN_NAME']: item['COLUMN_TYPE'] for item in name_type}
-                dtypes = {'mysql': {db_name: {tabel_name: dtypes}}}
+                dtypes = {'mysql': {db_n: {tabel_n: dtypes}}}
                 d.get_mysql_types(
                     dtypes=dtypes, 
                     cl='mysql', 
-                    db_name=db_name, 
-                    tabel_name=tabel_name,
+                    db_name=db_n,
+                    tabel_name=tabel_n,
                     is_file_dtype=True
                 )
             else:
@@ -228,4 +235,5 @@ def mysql_all_dtypes(path=None):
     d.as_json_file()
 
 if __name__ == '__main__':
-    mysql_all_dtypes()  # 更新 mysql 中所有数据库的 dtypes 信息到本地 json
+    # mysql_all_dtypes()  # 更新 mysql 中所有数据库的 dtypes 信息到本地 json
+    get_one_dtypes(db_name='京东数据2', tabel_name='sku_商品明细')

{mdbq-1.0.1 → mdbq-1.0.2}/mdbq/aggregation/query_data.py

@@ -119,8 +119,8 @@ class MysqlDatasQuery:
             '订单数': 1,
             '退货量': 1,
             '退款额': 1,
-            '退款额_发货后': 1,
-            '退货量_发货后': 1,
+            '退款额（发货后）': 1,
+            '退货量（发货后）': 1,
         }
         df = self.download.data_to_df(
             db_name='生意经2',
@@ -279,8 +279,8 @@ class GroupBy:
                    '订单数': ('订单数', np.min),
                    '退货量': ('退货量', np.max),
                    '退款额': ('退款额', np.max),
-                   '退款额_发货后': ('退款额_发货后', np.max),
-                   '退货量_发货后': ('退货量_发货后', np.max),
+                   '退款额（发货后）': ('退款额（发货后）', np.max),
+                   '退货量（发货后）': ('退货量（发货后）', np.max),
                    }
             )
             df['件均价'] = df.apply(lambda x: x['销售额'] / x['销售量'] if x['销售量'] > 0 else 0, axis=1).round(
@@ -294,7 +294,7 @@ class GroupBy:
             )
             self.data_tgyj.update(
                 {
-                    tabel_name: df[['日期', '宝贝id', '销售额', '销售量', '退款额_发货后', '退货量_发货后']],
+                    tabel_name: df[['日期', '宝贝id', '销售额', '销售量', '退款额（发货后）', '退货量（发货后）']],
                 }
             )
             return df

{mdbq-1.0.1 → mdbq-1.0.2}/mdbq.egg-info/PKG-INFO

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

{mdbq-1.0.1 → mdbq-1.0.2}/setup.py

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