浏览代码

Merge branch 'python-verison-grammer-error' of Server/title_with_video into master

luojunhui 5 月之前
父节点
当前提交
8fc5a87bc5
共有 1 个文件被更改,包括 35 次插入32 次删除
  1. 35 32
      matchVideoFromHistoryArticleASC.py

+ 35 - 32
matchVideoFromHistoryArticleASC.py

@@ -23,38 +23,41 @@ class TaskMySQLClient(object):
         初始化连接
         :return:
         """
-        match self.db_name:
-            case "pq":
-                self.mysql_pool = await aiomysql.create_pool(
-                    host='rm-bp1159bu17li9hi94.mysql.rds.aliyuncs.com',
-                    port=3306,
-                    user='crawler',
-                    password='crawler123456@',
-                    db='piaoquan-crawler',
-                    charset='utf8mb4',
-                    connect_timeout=120,
-                )
-            case "denet":
-                self.mysql_pool = await aiomysql.create_pool(
-                    host='rm-t4na9qj85v7790tf84o.mysql.singapore.rds.aliyuncs.com',
-                    port=3306,
-                    user='crawler_admin',
-                    password='cyber#crawler_2023',
-                    db='aigc-admin-prod',
-                    charset='utf8mb4',
-                    connect_timeout=120,
-                )
-            case "long-article":
-                self.mysql_pool = await aiomysql.create_pool(
-                    host='rm-bp14529nwwcw75yr1ko.mysql.rds.aliyuncs.com',
-                    port=3306,
-                    user='changwen_admin',
-                    password='changwen@123456',
-                    db='long_articles',
-                    charset='utf8mb4',
-                    connect_timeout=120,
-                )
-        print("mysql init successfully")
+        if self.db_name == 'pq':
+            self.mysql_pool = await aiomysql.create_pool(
+                host='rm-bp1159bu17li9hi94.mysql.rds.aliyuncs.com',
+                port=3306,
+                user='crawler',
+                password='crawler123456@',
+                db='piaoquan-crawler',
+                charset='utf8mb4',
+                connect_timeout=120,
+            )
+            print("mysql init successfully")
+        elif self.db_name == 'denet':
+            self.mysql_pool = await aiomysql.create_pool(
+                host='rm-t4na9qj85v7790tf84o.mysql.singapore.rds.aliyuncs.com',
+                port=3306,
+                user='crawler_admin',
+                password='cyber#crawler_2023',
+                db='aigc-admin-prod',
+                charset='utf8mb4',
+                connect_timeout=120,
+            )
+            print("mysql init successfully")
+        elif self.db_name == 'long-article':
+            self.mysql_pool = await aiomysql.create_pool(
+                host='rm-bp14529nwwcw75yr1ko.mysql.rds.aliyuncs.com',
+                port=3306,
+                user='changwen_admin',
+                password='changwen@123456',
+                db='long_articles',
+                charset='utf8mb4',
+                connect_timeout=120,
+            )
+            print("mysql init successfully")
+        else:
+            print("mysql init fail")
 
     async def close_pool(self):
         """