Prechádzať zdrojové kódy

上传读取标题功能

罗俊辉 1 rok pred
rodič
commit
1fcf600f21
2 zmenil súbory, kde vykonal 14 pridanie a 6 odobranie
  1. 11 4
      functions/mysql.py
  2. 3 2
      process_data.py

+ 11 - 4
functions/mysql.py

@@ -11,11 +11,12 @@ class MysqlClient(object):
 
     def __init__(self):
         mysql_config = {
-            "host": "rm-bp1159bu17li9hi94.mysql.rds.aliyuncs.com",
+            # "host": "rm-bp1159bu17li9hi94.mysql.rds.aliyuncs.com",
+            "host": "rr-bp1l12ea7e9wgu947.mysql.rds.aliyuncs.com",
             "port": 3306,  # 端口号
-            "user": "crawler",  # mysql用户名
-            "passwd": "crawler123456@",  # mysql用户登录密码
-            "db": "piaoquan-crawler",  # 数据库名
+            "user": "wx2016_longvideo",  # mysql用户名
+            "passwd": " wx2016_longvideoP@assword1234",  # mysql用户登录密码
+            "db": "longvideo",  # 数据库名
             "charset": "utf8mb4"  # 如果数据库里面的文本是utf8编码的,charset指定是utf8
         }
         self.connection = pymysql.connect(
@@ -57,3 +58,9 @@ class MysqlClient(object):
         关闭连接
         """
         self.connection.close()
+
+
+M = MysqlClient()
+sql = "SELECT title from crawler_video where id = '19591529';"
+w = M.select(sql)
+print(w)

+ 3 - 2
process_data.py

@@ -68,7 +68,7 @@ class DataProcessor(object):
         :param video_id: the video id
         :return: tag_list [tag, tag, tag, tag......]
         """
-        sql = f"""SELECT video_title from crawler_video where video_id = {video_id};"""
+        sql = f"""SELECT title from crawler_video where id = {video_id};"""
         try:
             title = self.client.select(sql)
             keywords_textrank = jieba.analyse.textrank(title, topK=3)
@@ -112,4 +112,5 @@ class DataProcessor(object):
 
 if __name__ == "__main__":
     D = DataProcessor(flag="train")
-    D.producer()
+    a, b = D.title_processor("19591529")
+    print(a, b)