| 
					
				 | 
			
			
				@@ -0,0 +1,197 @@ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+import os 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+import random 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+import sys 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+import time 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+import uuid 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+import json 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+from datetime import datetime 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+import cv2 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+import requests 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+from application.common.feishu import FsData 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+from application.common.feishu.feishu_utils import FeishuUtils 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+from application.common.gpt import GPT4oMini 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+from application.common.mysql.sql import Sql 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+from application.common.redis.xng_redis import xng_in_video_data 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+sys.path.append(os.getcwd()) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+from application.items import VideoItem 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+from application.pipeline import PiaoQuanPipeline 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+from application.common.messageQueue import MQ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+from application.common.log import AliyunLogger 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+from application.common.mysql import MysqlHelper 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+class HYSPRecommend(object): 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+    """ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+    好友视频 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+    """ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+    def __init__(self, platform, mode, rule_dict, user_list, env="prod"): 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        self.limit_flag = False 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        self.platform = platform 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        self.mode = mode 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        self.rule_dict = rule_dict 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        self.user_list = user_list 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        self.env = env 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        self.download_cnt = 0 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        self.mq = MQ(topic_name="topic_crawler_etl_" + self.env) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        self.expire_flag = False 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        self.aliyun_log = AliyunLogger(mode=self.mode, platform=self.platform) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        self.mysql = MysqlHelper(mode=self.mode, platform=self) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+    def get_recommend_list(self): 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        print("好友视频") 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        """ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        获取推荐页视频 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        """ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        headers = { 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            'Content-Type': 'application/json' 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        } 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        cursor  = "" 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        url = "http://8.217.192.46:8889/crawler/hao_you_shi_pin/recommend" 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        data_rule = FsData() 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        title_rule = data_rule.get_title_rule() 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        while True: 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            payload = json.dumps({ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                "cursor": cursor 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            }) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            response = requests.request("POST", url, headers=headers, data=payload) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            response = response.json() 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            if response['code'] != 0: 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                self.aliyun_log.logging( 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                    code="3000", 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                    message="抓取单条视频失败,请求失败" 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                ), 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                return 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            cursor = response['data']["next_cursor"] 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            data = response['data']['data'] 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            if len(data) == 0: 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                return 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            for index, video_obj in enumerate(data, 1): 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                try: 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                    self.aliyun_log.logging( 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                        code="1001", message="扫描到一条视频", data=video_obj 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                    ) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                    self.process_video_obj(video_obj,title_rule) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                except Exception as e: 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                    self.aliyun_log.logging( 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                        code="3000", 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                        message="抓取单条视频失败, 该视频位于第{}页第{}条报错原因是{}".format( 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                            1, index, e 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                        ), 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                    ) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                if self.limit_flag: 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                    return 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                time.sleep(random.randint(1, 5)) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+    def process_video_obj(self, video_obj,title_rule): 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        """ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        处理视频 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        :param video_obj: 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        """ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        time.sleep(random.randint(3, 8)) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        trace_id = self.platform + str(uuid.uuid1()) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        our_user = random.choice(self.user_list) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        video_url = self.get_video_url(video_obj["vid"]) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        if video_url: 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            item = VideoItem() 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            item.add_video_info("video_id", video_obj["id"]) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            item.add_video_info("video_title", video_obj["vtitle"]) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            item.add_video_info("play_cnt", 0) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            item.add_video_info("publish_time_stamp", int(time.time())) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            item.add_video_info("out_user_id", video_obj["id"]) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            item.add_video_info("cover_url", "https://qiniu.818ao.com/"+video_obj["poster"]) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            item.add_video_info("like_cnt", 0) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            item.add_video_info("share_cnt", 0) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            item.add_video_info("comment_cnt", 0) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            item.add_video_info("video_url", video_url) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            item.add_video_info("out_video_id", video_obj["id"]) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            item.add_video_info("platform", self.platform) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            item.add_video_info("strategy", self.mode) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            item.add_video_info("session", "{}-{}".format(self.platform, int(time.time()))) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            item.add_video_info("user_id", our_user["uid"]) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            item.add_video_info("user_name", our_user["nick_name"]) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            mq_obj = item.produce_item() 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            pipeline = PiaoQuanPipeline( 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                platform=self.platform, 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                mode=self.mode, 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                rule_dict=self.rule_dict, 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                env=self.env, 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                item=mq_obj, 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                trace_id=trace_id, 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            ) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            if pipeline.process_item(): 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                title_list = title_rule.split(",") 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                title = video_obj["title"] 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                contains_keyword = any(keyword in title for keyword in title_list) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                if contains_keyword: 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                    new_title = GPT4oMini.get_ai_mini_title(title) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                    if new_title: 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                        item.add_video_info("video_title", new_title) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                        current_time = datetime.now() 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                        formatted_time = current_time.strftime("%Y-%m-%d %H:%M:%S") 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                        values = [ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                            [ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                                video_url, 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                                video_obj["cover"], 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                                title, 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                                new_title, 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                                formatted_time, 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                            ] 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                        ] 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                        FeishuUtils.insert_columns("U5dXsSlPOhiNNCtEfgqcm1iYnpf", "8c7191", "ROWS", 1, 2) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                        time.sleep(0.5) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                        FeishuUtils.update_values("U5dXsSlPOhiNNCtEfgqcm1iYnpf", "8c7191", "A2:Z2", values) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                self.download_cnt += 1 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                self.mq.send_msg(mq_obj) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                self.aliyun_log.logging(code="1002", message="成功发送至 ETL", data=mq_obj) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                if self.download_cnt >= int( 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                        self.rule_dict.get("videos_cnt", {}).get("min", 200) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                ): 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                    self.limit_flag = True 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+    """获取视频链接""" 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+    def get_video_url(self, vid): 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        url = "http://8.217.192.46:8889/crawler/hao_you_shi_pin/detail" 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        payload = json.dumps({ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            "content_id": f"{vid}" 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        }) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        headers = { 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            'Content-Type': 'application/json' 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        } 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        try: 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            response = requests.request("POST", url, headers=headers, data=payload) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            response = response.json() 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            if response['code'] != 0: 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                self.aliyun_log.logging( 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                    code="3000", 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                    message="获取视频链接失败" 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                ), 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                return None 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            video_url = response['data']['data']['video_url_list'][0]['video_url'] 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            return video_url 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        except Exception as e: 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            return None 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+    def run(self): 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        self.get_recommend_list() 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+if __name__ == '__main__': 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+    J = HYSPRecommend( 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        platform="haoyoushipin", 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        mode="recommend", 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        rule_dict={}, 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        user_list=[{'uid': "123456", 'nick_name': "xiaoxiao"}], 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+    ) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+    J.get_recommend_list() 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+    # J.logic() 
			 |