add 获取小年糕用户id

xiaoniangaoplus/xiaoniangaoplus/xiaoniangao_plus_get_userid.py

@@ -189,8 +189,12 @@ class XiaoNianGaoPlusRecommend:
         video_mid_elements = self.search_elements("//wx-view[@class='bar--navBar-content-capsule']")
         mid = int(video_mid_elements[0].get_attribute("data-mid"))
         repeat_video_id= self.repeat_video_id(mid)
+        data_list = []
         if repeat_video_id != 0:
             Common.logger(self.log_type, self.crawler).info(f"该用户已经存在")
+            status = 1
+            self.insert_user(mid, user_name, data_list, status)
+
             self.driver.press_keycode(AndroidKey.BACK)
             return
 
@@ -200,13 +204,14 @@ class XiaoNianGaoPlusRecommend:
             self.driver.press_keycode(AndroidKey.BACK)
             return
         else:
-            self.insert_user(mid, user_name, data_list)
+            status  = 1
+            self.insert_user(mid, user_name, data_list, status)
             Common.logger(self.log_type, self.crawler).info(f"{mid}:{user_name}入库")
             self.driver.press_keycode(AndroidKey.BACK)
             time.sleep(2)
 
-    def insert_user(self, mid, user_name, data_list):
-        insert_sql = f"""insert into crawler_xng_userid( user_id , user_name , user_title_text) values ({mid},"{user_name}", "{data_list}")"""
+    def insert_user(self, mid, user_name, data_list, status):
+        insert_sql = f"""insert into crawler_xng_userid( user_id , user_name , user_title_text , status) values ({mid},"{user_name}", "{data_list}",{status})"""
         MysqlHelper.update_values(self.log_type, self.crawler, insert_sql, self.env, action='')
 
     def get_user_list(self, mid):