wangkun 2 vuotta sitten
vanhempi
commit
79d9f3dfdf
2 muutettua tiedostoa jossa 15 lisäystä ja 2 poistoa
  1. 14 1
      main/demo.py
  2. 1 1
      main/zmyx_recommend.py

+ 14 - 1
main/demo.py

@@ -1,6 +1,8 @@
 # -*- coding: utf-8 -*-
 # @Author: wangkun
 # @Time: 2022/11/3
+import difflib
+
 from main.feishu_lib import Feishu
 
 
@@ -10,7 +12,18 @@ class Demo:
         sheet = Feishu.get_values_batch(log_type, crawler, sheetid)
         print(sheet)
 
+    @classmethod
+    def title_like(cls, log_type, title):
+        sheet = Feishu.get_values_batch(log_type, 'zmyx', '19c772')
+        for i in range(1, len(sheet)):
+            video_title = sheet[i][7]
+            if difflib.SequenceMatcher(None, title, video_title).quick_ratio() >= 0.8:
+                return True
+            else:
+                pass
+
 
 if __name__ == '__main__':
-    Demo.get_sheet('recommend', 'zmyx', '19c772')
+    # Demo.get_sheet('recommend', 'zmyx', '19c772')
+    print(Demo.title_like('recommend', '国人幡然泪下!!!'))
     pass

+ 1 - 1
main/zmyx_recommend.py

@@ -198,7 +198,7 @@ class ZMYXRecommend:
             if difflib.SequenceMatcher(None, title, video_title).quick_ratio() >= 0.8:
                 return True
             else:
-                return False
+                pass
 
     @classmethod
     def get_zmyx_recommend(cls, log_type, driver: WebDriver, env):