mdbq 0.1.5__py3-none-any.whl → 0.1.6__py3-none-any.whl

mdbq/mysql/s_query.py

@@ -86,6 +86,33 @@ class QueryDatas:
             print(f'database: {db_name}, table: {tabel_name} 查询的数据为空')
         return df
 
+    def columns_to_list(self, db_name, tabel_name,  columns_name) -> list:
+        """
+        获取数据表的指定列, 返回列表
+        [{'视频bv号': 'BV1Dm4y1S7BU', '下载进度': 1}, {'视频bv号': 'BV1ov411c7US', '下载进度': 1}]
+        """
+        if self.check_infos(db_name, tabel_name) == False:  # 检查传入的数据库和数据表是否存在
+            return []
+
+        self.config.update({'database': db_name})
+        connection = pymysql.connect(**self.config)  # 重新连接数据库
+        with connection.cursor() as cursor:
+            # 3. 获取数据表的所有列信息
+            sql = 'SELECT COLUMN_NAME FROM information_schema.columns WHERE table_schema = %s AND table_name = %s'
+            cursor.execute(sql, (db_name, {tabel_name}))
+            columns = cursor.fetchall()
+            cols_exist = [col['COLUMN_NAME'] for col in columns]  # 数据表的所有列, 返回 list
+            columns_name = [item for item in columns_name if item in cols_exist]
+            if len(columns_name) == 0:
+                return []
+            columns_in = ', '.join(columns_name)
+            sql = (f"SELECT {columns_in} FROM {db_name}.{tabel_name} ")
+            cursor.execute(sql)
+            column_values = cursor.fetchall()  # 返回指定列，结果是[dict, dict, dict, ...]
+            # column_values = [item[column_name] for item in column_values]  # 提取字典的值, 组成列表
+        connection.close()
+        return column_values
+
     def check_infos(self, db_name, tabel_name) -> bool:
         """ 检查数据库、数据表是否存在 """
         connection = pymysql.connect(**self.config)  # 连接数据库
@@ -119,5 +146,9 @@ class QueryDatas:
 
 
 if __name__ == '__main__':
-    username, password, host, port = get_myconf.select_config_values(target_service='company', database='mysql')
+    username, password, host, port = get_myconf.select_config_values(target_service='home_lx', database='mysql')
     print(username, password, host, port)
+
+    q = QueryDatas(username, password, host, port)
+    res = q.columns_to_list(db_name='视频数据', tabel_name='bilibili视频', columns_name=['视频bv号', '下载进度'])
+    print(res)

{mdbq-0.1.5.dist-info → mdbq-0.1.6.dist-info}/METADATA

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

{mdbq-0.1.5.dist-info → mdbq-0.1.6.dist-info}/RECORD

@@ -20,7 +20,7 @@ mdbq/mongo/__init__.py,sha256=SILt7xMtQIQl_m-ik9WLtJSXIVf424iYgCfE_tnQFbw,13
 mdbq/mongo/mongo.py,sha256=q0B4wXDSTtXg_vMN7MPh6zdxl6tT68tM74LmdVNQQek,31892
 mdbq/mysql/__init__.py,sha256=A_DPJyAoEvTSFojiI2e94zP0FKtCkkwKP1kYUCSyQzo,11
 mdbq/mysql/mysql.py,sha256=w_FDE4ulOnHdfIjjXB1Ff_cnXyx2PuBAQWGjTYyVm3U,30629
-mdbq/mysql/s_query.py,sha256=pzxRGBRP3Ku4oVLqfpMd1VWWct3YlxxnVavtKa9kgSM,5302
+mdbq/mysql/s_query.py,sha256=mNrdyMeiQ5wQVD_9IDTrk6jrMtszQftYQg3AwNWhROc,6997
 mdbq/mysql/year_month_day.py,sha256=VgewoE2pJxK7ErjfviL_SMTN77ki8GVbTUcao3vFUCE,1523
 mdbq/other/__init__.py,sha256=jso1oHcy6cJEfa7udS_9uO5X6kZLoPBF8l3wCYmr5dM,18
 mdbq/other/porxy.py,sha256=UHfgEyXugogvXgsG68a7QouUCKaohTKKkI4RN-kYSdQ,4961
@@ -30,7 +30,7 @@ mdbq/pbix/__init__.py,sha256=Trtfaynu9RjoTyLLYBN2xdRxTvm_zhCniUkVTAYwcjo,24
 mdbq/pbix/pbix_refresh.py,sha256=JUjKW3bNEyoMVfVfo77UhguvS5AWkixvVhDbw4_MHco,2396
 mdbq/pbix/refresh_all.py,sha256=wulHs4rivf4Mi0Pii2QR5Nk9-TBcvSwnCB_WH9QULKE,5939
 mdbq/spider/__init__.py,sha256=RBMFXGy_jd1HXZhngB2T2XTvJqki8P_Fr-pBcwijnew,18
-mdbq-0.1.5.dist-info/METADATA,sha256=X-iiiuZTrT9STDW8bThvKswQ_cLjqv_rqdIVZ4Of4oI,245
-mdbq-0.1.5.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
-mdbq-0.1.5.dist-info/top_level.txt,sha256=2FQ-uLnCSB-OwFiWntzmwosW3X2Xqsg0ewh1axsaylA,5
-mdbq-0.1.5.dist-info/RECORD,,
+mdbq-0.1.6.dist-info/METADATA,sha256=a5jG4lDS-9TA8xU5IInf--XK_X0P84L9FOZQY-ysGHU,245
+mdbq-0.1.6.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
+mdbq-0.1.6.dist-info/top_level.txt,sha256=2FQ-uLnCSB-OwFiWntzmwosW3X2Xqsg0ewh1axsaylA,5
+mdbq-0.1.6.dist-info/RECORD,,