kuaishou_follow.py 45 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777
  1. # -*- coding: utf-8 -*-
  2. # @Author: wangkun
  3. # @Time: 2023/2/24
  4. import os
  5. import random
  6. import shutil
  7. import sys
  8. import time
  9. from hashlib import md5
  10. import requests
  11. import json
  12. import urllib3
  13. from requests.adapters import HTTPAdapter
  14. sys.path.append(os.getcwd())
  15. from common.common import Common
  16. from common.feishu import Feishu
  17. from common.getuser import getUser
  18. from common.db import MysqlHelper
  19. from common.publish import Publish
  20. class Follow:
  21. platform = "快手"
  22. tag = "快手爬虫,定向爬虫策略"
  23. @classmethod
  24. def get_rule(cls, log_type, crawler, index):
  25. try:
  26. while True:
  27. rule_sheet = Feishu.get_values_batch(log_type, crawler, "3iqG4z")
  28. if rule_sheet is None:
  29. Common.logger(log_type, crawler).warning("rule_sheet is None! 10秒后重新获取")
  30. time.sleep(10)
  31. continue
  32. if index == 1:
  33. rule_dict = {
  34. "play_cnt": f"{rule_sheet[1][1]}{rule_sheet[1][2]}",
  35. "video_width": f"{rule_sheet[2][1]}{rule_sheet[2][2]}",
  36. "video_height": f"{rule_sheet[3][1]}{rule_sheet[3][2]}",
  37. "like_cnt": f"{rule_sheet[4][1]}{rule_sheet[4][2]}",
  38. "duration": f"{rule_sheet[5][1]}{rule_sheet[5][2]}",
  39. "download_cnt": f"{rule_sheet[6][1]}{rule_sheet[6][2]}",
  40. "publish_time": f"{rule_sheet[7][1]}{rule_sheet[7][2]}",
  41. }
  42. # for k, v in rule_dict.items():
  43. # Common.logger(log_type, crawler).info(f"{k}:{v}")
  44. return rule_dict
  45. elif index == 2:
  46. rule_dict = {
  47. "play_cnt": f"{rule_sheet[9][1]}{rule_sheet[9][2]}",
  48. "video_width": f"{rule_sheet[10][1]}{rule_sheet[10][2]}",
  49. "video_height": f"{rule_sheet[11][1]}{rule_sheet[11][2]}",
  50. "like_cnt": f"{rule_sheet[12][1]}{rule_sheet[12][2]}",
  51. "duration": f"{rule_sheet[13][1]}{rule_sheet[13][2]}",
  52. "download_cnt": f"{rule_sheet[14][1]}{rule_sheet[14][2]}",
  53. "publish_time": f"{rule_sheet[15][1]}{rule_sheet[15][2]}",
  54. }
  55. # for k, v in rule_dict.items():
  56. # Common.logger(log_type, crawler).info(f"{k}:{v}")
  57. return rule_dict
  58. except Exception as e:
  59. Common.logger(log_type, crawler).error(f"get_rule:{e}\n")
  60. @classmethod
  61. def download_rule(cls, video_dict, rule_dict):
  62. if eval(f"{video_dict['play_cnt']}{rule_dict['play_cnt']}") is True \
  63. and eval(f"{video_dict['video_width']}{rule_dict['video_width']}") is True \
  64. and eval(f"{video_dict['video_height']}{rule_dict['video_height']}") is True \
  65. and eval(f"{video_dict['like_cnt']}{rule_dict['like_cnt']}") is True \
  66. and eval(f"{video_dict['duration']}{rule_dict['duration']}") is True \
  67. and eval(f"{video_dict['publish_time']}{rule_dict['publish_time']}") is True:
  68. return True
  69. else:
  70. return False
  71. # 过滤词库
  72. @classmethod
  73. def filter_words(cls, log_type, crawler):
  74. try:
  75. while True:
  76. filter_words_sheet = Feishu.get_values_batch(log_type, crawler, 'HIKVvs')
  77. if filter_words_sheet is None:
  78. Common.logger(log_type, crawler).warning(f"filter_words_sheet:{filter_words_sheet} 10秒钟后重试")
  79. continue
  80. filter_words_list = []
  81. for x in filter_words_sheet:
  82. for y in x:
  83. if y is None:
  84. pass
  85. else:
  86. filter_words_list.append(y)
  87. return filter_words_list
  88. except Exception as e:
  89. Common.logger(log_type, crawler).error(f'filter_words异常:{e}\n')
  90. # 万能标题
  91. @classmethod
  92. def random_title(cls, log_type, crawler):
  93. try:
  94. while True:
  95. random_title_sheet = Feishu.get_values_batch(log_type, crawler, '0DiyXe')
  96. if random_title_sheet is None:
  97. Common.logger(log_type, crawler).warning(f"filter_words_sheet:{random_title_sheet} 10秒钟后重试")
  98. continue
  99. random_title_list = []
  100. for x in random_title_sheet:
  101. for y in x:
  102. if y is None:
  103. pass
  104. else:
  105. random_title_list.append(y)
  106. return random.choice(random_title_list)
  107. except Exception as e:
  108. Common.logger(log_type, crawler).error(f'random_title:{e}\n')
  109. # 获取站外用户信息
  110. @classmethod
  111. def get_out_user_info(cls, log_type, crawler, out_uid):
  112. try:
  113. url = "https://www.kuaishou.com/graphql"
  114. payload = json.dumps({
  115. "operationName": "visionProfile",
  116. "variables": {
  117. "userId": str(out_uid)
  118. },
  119. "query": "query visionProfile($userId: String) {\n visionProfile(userId: $userId) {\n result\n hostName\n userProfile {\n ownerCount {\n fan\n photo\n follow\n photo_public\n __typename\n }\n profile {\n gender\n user_name\n user_id\n headurl\n user_text\n user_profile_bg_url\n __typename\n }\n isFollowing\n __typename\n }\n __typename\n }\n}\n"
  120. })
  121. headers = {
  122. # 'Cookie': f'kpf=PC_WEB; clientid=3; did=web_e2901e1c5a13c60af81ba88bc7a3ee24; userId={"".join(str(random.choice(range(1, 10))) for _ in range(10))}; kpn=KUAISHOU_VISION; kuaishou.server.web_st=ChZrdWFpc2hvdS5zZXJ2ZXIud2ViLnN0EqABE4wGjnJauApJelOpl9Xqo8TVDAyra7Pvo0rZtVgMSZxgVuw4Z6P2UtHv_CHOk2Ne2el1hdE_McCptWs8tRdtYlhXFlVOu8rQX7CwexzOBudJAfB3lDN8LPc4o4qHNwqFxy5J5j_WzdllbqMmaDUK9yUxX6XA-JFezzq9jvBwtGv7_hzB7pFrUcH39z0EYOQaZo5lDl-pE09Gw7wr8NvlZRoSdWlbobCW6oJxuQLJTUr9oj_uIiBhkeb1psaIIc3VwfYQ1UfvobrXAP_WpnRabE_3UZUBOygFMAE; kuaishou.server.web_ph=2b981e2051d7130c977fd31df97fe6f5ad54',
  123. 'Cookie': f'kpf=PC_WEB; clientid=3; did=web_e2901e1c5a13c60af81ba88bc7a3ee24; kuaishou.server.web_st=ChZrdWFpc2hvdS5zZXJ2ZXIud2ViLnN0EqABszwASP0eGT2ym0NUdGf1PTi8_gDYrDms9otf5FKMUrx0V7MOTT1hhvpVJijmgku8KYtu3a6g3X1hZODMpX0ebZUip44txWPoY3VRqbsnBKEOs9Qkmx3uLaX33nq8KevOKLoO0cIE8nfwMJISe_BtCHr22cbJkofI0xfJXBt_ZgPqJIqWsUwdgwKzqRYn47ROkIqzeDfEjbxaZxiXy22ZhhoSlCobbmtjkvjpY9x730BPP_C5IiAl9EBaCTyvMw5IIcnkmcjMeWC8w6LKzeFMCNhcqrfkgCgFMAE; kuaishou.server.web_ph=19ae269a54f619c49def39ef5b54ae6d4353; kpn=KUAISHOU_VISION',
  124. 'Referer': f'https://www.kuaishou.com/profile/{out_uid}',
  125. 'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/110.0.0.0 Safari/537.36 Edg/110.0.1587.41',
  126. 'content-type': 'application/json',
  127. 'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8,en-GB;q=0.7,en-US;q=0.6',
  128. 'Cache-Control': 'no-cache',
  129. 'Connection': 'keep-alive',
  130. 'Origin': 'https://www.kuaishou.com',
  131. 'Pragma': 'no-cache',
  132. 'Sec-Fetch-Dest': 'empty',
  133. 'Sec-Fetch-Mode': 'cors',
  134. 'Sec-Fetch-Site': 'same-origin',
  135. 'accept': '*/*',
  136. 'sec-ch-ua': '"Chromium";v="110", "Not A(Brand";v="24", "Microsoft Edge";v="110"',
  137. 'sec-ch-ua-mobile': '?0',
  138. 'sec-ch-ua-platform': '"macOS"'
  139. }
  140. urllib3.disable_warnings()
  141. s = requests.session()
  142. # max_retries=3 重试3次
  143. s.mount('http://', HTTPAdapter(max_retries=3))
  144. s.mount('https://', HTTPAdapter(max_retries=3))
  145. response = s.post(url=url, headers=headers, data=payload, proxies=Common.tunnel_proxies(), verify=False,
  146. timeout=5)
  147. response.close()
  148. # Common.logger(log_type, crawler).info(f"get_out_user_info_response:{response.text}")
  149. if response.status_code != 200:
  150. Common.logger(log_type, crawler).warning(f"get_out_user_info_response:{response.text}\n")
  151. return
  152. elif 'data' not in response.json():
  153. Common.logger(log_type, crawler).warning(f"get_out_user_info_response:{response.json()}\n")
  154. return
  155. elif 'visionProfile' not in response.json()['data']:
  156. Common.logger(log_type, crawler).warning(f"get_out_user_info_response:{response.json()['data']}\n")
  157. return
  158. elif 'userProfile' not in response.json()['data']['visionProfile']:
  159. Common.logger(log_type, crawler).warning(
  160. f"get_out_user_info_response:{response.json()['data']['visionProfile']['userProfile']}\n")
  161. return
  162. else:
  163. userProfile = response.json()['data']['visionProfile']['userProfile']
  164. # Common.logger(log_type, crawler).info(f"userProfile:{userProfile}")
  165. try:
  166. out_fans_str = str(userProfile['ownerCount']['fan'])
  167. except Exception:
  168. out_fans_str = "0"
  169. try:
  170. out_follow_str = str(userProfile['ownerCount']['follow'])
  171. except Exception:
  172. out_follow_str = "0"
  173. try:
  174. out_avatar_url = userProfile['profile']['headurl']
  175. except Exception:
  176. out_avatar_url = ""
  177. Common.logger(log_type, crawler).info(f"out_fans_str:{out_fans_str}")
  178. Common.logger(log_type, crawler).info(f"out_follow_str:{out_follow_str}")
  179. Common.logger(log_type, crawler).info(f"out_avatar_url:{out_avatar_url}")
  180. if "万" in out_fans_str:
  181. out_fans = int(float(out_fans_str.split("万")[0]) * 10000)
  182. else:
  183. out_fans = int(out_fans_str.replace(",", ""))
  184. if "万" in out_follow_str:
  185. out_follow = int(float(out_follow_str.split("万")[0]) * 10000)
  186. else:
  187. out_follow = int(out_follow_str.replace(",", ""))
  188. out_user_dict = {
  189. "out_fans": out_fans,
  190. "out_follow": out_follow,
  191. "out_avatar_url": out_avatar_url
  192. }
  193. Common.logger(log_type, crawler).info(f"out_user_dict:{out_user_dict}")
  194. return out_user_dict
  195. except Exception as e:
  196. Common.logger(log_type, crawler).error(f"get_out_user_info:{e}\n")
  197. # 获取用户信息列表
  198. @classmethod
  199. def get_user_list(cls, log_type, crawler, sheetid, env, machine):
  200. try:
  201. while True:
  202. user_sheet = Feishu.get_values_batch(log_type, crawler, sheetid)
  203. if user_sheet is None:
  204. Common.logger(log_type, crawler).warning(f"user_sheet:{user_sheet} 10秒钟后重试")
  205. continue
  206. our_user_list = []
  207. for i in range(1, len(user_sheet)):
  208. # for i in range(1, 2):
  209. out_uid = user_sheet[i][2]
  210. user_name = user_sheet[i][3]
  211. our_uid = user_sheet[i][6]
  212. our_user_link = user_sheet[i][7]
  213. if out_uid is None or user_name is None:
  214. Common.logger(log_type, crawler).info("空行\n")
  215. else:
  216. Common.logger(log_type, crawler).info(f"正在更新 {user_name} 用户信息\n")
  217. if our_uid is None:
  218. out_user_info = cls.get_out_user_info(log_type, crawler, out_uid)
  219. out_user_dict = {
  220. "out_uid": out_uid,
  221. "user_name": user_name,
  222. "out_avatar_url": out_user_info["out_avatar_url"],
  223. "out_create_time": '',
  224. "out_tag": '',
  225. "out_play_cnt": 0,
  226. "out_fans": out_user_info["out_fans"],
  227. "out_follow": out_user_info["out_follow"],
  228. "out_friend": 0,
  229. "out_like": 0,
  230. "platform": cls.platform,
  231. "tag": cls.tag,
  232. }
  233. our_user_dict = getUser.create_user(log_type=log_type, crawler=crawler,
  234. out_user_dict=out_user_dict, env=env, machine=machine)
  235. our_uid = our_user_dict['our_uid']
  236. our_user_link = our_user_dict['our_user_link']
  237. Feishu.update_values(log_type, crawler, sheetid, f'G{i + 1}:H{i + 1}',
  238. [[our_uid, our_user_link]])
  239. Common.logger(log_type, crawler).info(f'站内用户信息写入飞书成功!\n')
  240. our_user_list.append(our_user_dict)
  241. else:
  242. our_user_dict = {
  243. 'out_uid': out_uid,
  244. 'user_name': user_name,
  245. 'our_uid': our_uid,
  246. 'our_user_link': our_user_link,
  247. }
  248. our_user_list.append(our_user_dict)
  249. return our_user_list
  250. except Exception as e:
  251. Common.logger(log_type, crawler).error(f'get_user_list:{e}\n')
  252. # 处理视频标题
  253. @classmethod
  254. def video_title(cls, log_type, crawler, title):
  255. title_split1 = title.split(" #")
  256. if title_split1[0] != "":
  257. title1 = title_split1[0]
  258. else:
  259. title1 = title_split1[-1]
  260. title_split2 = title1.split(" #")
  261. if title_split2[0] != "":
  262. title2 = title_split2[0]
  263. else:
  264. title2 = title_split2[-1]
  265. title_split3 = title2.split("@")
  266. if title_split3[0] != "":
  267. title3 = title_split3[0]
  268. else:
  269. title3 = title_split3[-1]
  270. video_title = title3.strip().replace("\n", "") \
  271. .replace("/", "").replace("快手", "").replace(" ", "") \
  272. .replace(" ", "").replace("&NBSP", "").replace("\r", "") \
  273. .replace("#", "").replace(".", "。").replace("\\", "") \
  274. .replace(":", "").replace("*", "").replace("?", "") \
  275. .replace("?", "").replace('"', "").replace("<", "") \
  276. .replace(">", "").replace("|", "").replace("@", "")[:40]
  277. if video_title.replace(" ", "") == "" or video_title == "。。。" or video_title == "...":
  278. return cls.random_title(log_type, crawler)
  279. else:
  280. return video_title
  281. # @classmethod
  282. # def get_cookie(cls, log_type, crawler, out_uid, machine):
  283. # try:
  284. # # 打印请求配置
  285. # ca = DesiredCapabilities.CHROME
  286. # ca["goog:loggingPrefs"] = {"performance": "ALL"}
  287. #
  288. # # 不打开浏览器运行
  289. # chrome_options = webdriver.ChromeOptions()
  290. # chrome_options.add_argument("headless")
  291. # chrome_options.add_argument(
  292. # f'user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/79.0.3945.79 Safari/537.36')
  293. # chrome_options.add_argument("--no-sandbox")
  294. #
  295. # # driver初始化
  296. # if machine == "aliyun":
  297. # driver = webdriver.Chrome(desired_capabilities=ca, options=chrome_options)
  298. # elif machine == "macpro":
  299. # driver = webdriver.Chrome(desired_capabilities=ca, options=chrome_options,
  300. # service=Service('/Users/lieyunye/Downloads/chromedriver_v107/chromedriver'))
  301. # elif machine == "macair":
  302. # driver = webdriver.Chrome(desired_capabilities=ca, options=chrome_options,
  303. # service=Service('/Users/piaoquan/Downloads/chromedriver_v108/chromedriver'))
  304. # else:
  305. # driver = webdriver.Chrome(desired_capabilities=ca, options=chrome_options, service=Service(
  306. # '/Users/wangkun/Downloads/chromedriver/chromedriver_v109/chromedriver'))
  307. #
  308. # driver.implicitly_wait(10)
  309. # # print('打开个人主页')
  310. # driver.get(f'https://www.kuaishou.com/profile/{out_uid}')
  311. # time.sleep(1)
  312. #
  313. # # print('解析cookies')
  314. # logs = driver.get_log("performance")
  315. # # Common.logger(log_type, crawler).info('已获取logs:{}\n', logs)
  316. # # print('退出浏览器')
  317. # driver.quit()
  318. # for line in logs:
  319. # msg = json.loads(line['message'])
  320. # # Common.logger(log_type, crawler).info(f"{msg}\n\n")
  321. # if 'message' not in msg:
  322. # pass
  323. # elif 'params' not in msg['message']:
  324. # pass
  325. # elif 'headers' not in msg['message']['params']:
  326. # pass
  327. # elif 'Cookie' not in msg['message']['params']['headers']:
  328. # pass
  329. # elif msg['message']['params']['headers']['Host'] != 'www.kuaishou.com':
  330. # pass
  331. # else:
  332. # cookie = msg['message']['params']['headers']['Cookie']
  333. # # Common.logger(log_type, crawler).info(f"{cookie}")
  334. # return cookie
  335. # except Exception as e:
  336. # Common.logger(log_type, crawler).error(f"get_cookie:{e}\n")
  337. @classmethod
  338. def get_videoList(cls, log_type, crawler, strategy, our_uid, out_uid, oss_endpoint, env, machine):
  339. download_cnt_1, download_cnt_2 = 0, 0
  340. pcursor = ""
  341. while True:
  342. rule_dict_1 = cls.get_rule(log_type, crawler, 1)
  343. rule_dict_2 = cls.get_rule(log_type, crawler, 2)
  344. if rule_dict_1 is None or rule_dict_2 is None:
  345. Common.logger(log_type, crawler).warning(f"rule_dict is None, 10秒后重试")
  346. time.sleep(10)
  347. else:
  348. break
  349. try:
  350. if download_cnt_1 >= int(
  351. rule_dict_1['download_cnt'].replace("=", "")[-1].replace("<", "")[-1].replace(">", "")[
  352. -1]) and download_cnt_2 >= int(
  353. rule_dict_2['download_cnt'].replace("=", "")[-1].replace("<", "")[-1].replace(">", "")[-1]):
  354. Common.logger(log_type, crawler).info(
  355. f"规则1已下载{download_cnt_1}条视频,规则2已下载{download_cnt_2}条视频\n")
  356. return
  357. url = "https://www.kuaishou.com/graphql"
  358. payload = json.dumps({
  359. "operationName": "visionProfilePhotoList",
  360. "variables": {
  361. "userId": out_uid,
  362. "pcursor": pcursor,
  363. "page": "profile"
  364. },
  365. "query": "fragment photoContent on PhotoEntity {\n id\n duration\n caption\n originCaption\n likeCount\n viewCount\n realLikeCount\n coverUrl\n photoUrl\n photoH265Url\n manifest\n manifestH265\n videoResource\n coverUrls {\n url\n __typename\n }\n timestamp\n expTag\n animatedCoverUrl\n distance\n videoRatio\n liked\n stereoType\n profileUserTopPhoto\n musicBlocked\n __typename\n}\n\nfragment feedContent on Feed {\n type\n author {\n id\n name\n headerUrl\n following\n headerUrls {\n url\n __typename\n }\n __typename\n }\n photo {\n ...photoContent\n __typename\n }\n canAddComment\n llsid\n status\n currentPcursor\n tags {\n type\n name\n __typename\n }\n __typename\n}\n\nquery visionProfilePhotoList($pcursor: String, $userId: String, $page: String, $webPageArea: String) {\n visionProfilePhotoList(pcursor: $pcursor, userId: $userId, page: $page, webPageArea: $webPageArea) {\n result\n llsid\n webPageArea\n feeds {\n ...feedContent\n __typename\n }\n hostName\n pcursor\n __typename\n }\n}\n"
  366. })
  367. headers = {
  368. 'Cookie': 'kpf=PC_WEB; clientid=3; did=web_8832834f71f4c6ed405b5a35e95abc6d; kpn=KUAISHOU_VISION',
  369. 'Referer': f'https://www.kuaishou.com/profile/{out_uid}',
  370. 'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/110.0.0.0 Safari/537.36 Edg/110.0.1587.41',
  371. 'content-type': 'application/json',
  372. }
  373. urllib3.disable_warnings()
  374. s = requests.session()
  375. # max_retries=3 重试3次
  376. s.mount('http://', HTTPAdapter(max_retries=3))
  377. s.mount('https://', HTTPAdapter(max_retries=3))
  378. response = s.post(url=url, headers=headers, data=payload, proxies=Common.tunnel_proxies(), verify=False,
  379. timeout=10)
  380. response.close()
  381. # Common.logger(log_type, crawler).info(f"get_videoList:{response.text}\n")
  382. if response.status_code != 200:
  383. Common.logger(log_type, crawler).warning(f"get_videoList_response:{response.text}\n")
  384. return
  385. elif 'data' not in response.json():
  386. Common.logger(log_type, crawler).warning(f"get_videoList_response:{response.json()}\n")
  387. return
  388. elif 'visionProfilePhotoList' not in response.json()['data']:
  389. Common.logger(log_type, crawler).warning(f"get_videoList_response:{response.json()['data']}\n")
  390. return
  391. elif 'feeds' not in response.json()['data']['visionProfilePhotoList']:
  392. Common.logger(log_type, crawler).warning(
  393. f"get_videoList_response:{response.json()['data']['visionProfilePhotoList']}\n")
  394. return
  395. elif len(response.json()['data']['visionProfilePhotoList']['feeds']) == 0:
  396. Common.logger(log_type, crawler).info("没有更多视频啦 ~\n")
  397. return
  398. else:
  399. feeds = response.json()['data']['visionProfilePhotoList']['feeds']
  400. pcursor = response.json()['data']['visionProfilePhotoList']['pcursor']
  401. # Common.logger(log_type, crawler).info(f"feeds0: {feeds}\n")
  402. for i in range(len(feeds)):
  403. if 'photo' not in feeds[i]:
  404. Common.logger(log_type, crawler).warning(f"get_videoList:{feeds[i]}\n")
  405. break
  406. # video_title
  407. if 'caption' not in feeds[i]['photo']:
  408. video_title = cls.random_title(log_type, crawler)
  409. elif feeds[i]['photo']['caption'].strip() == "":
  410. video_title = cls.random_title(log_type, crawler)
  411. else:
  412. video_title = cls.video_title(log_type, crawler, feeds[i]['photo']['caption'])
  413. if 'videoResource' not in feeds[i]['photo'] \
  414. and 'manifest' not in feeds[i]['photo'] \
  415. and 'manifestH265' not in feeds[i]['photo']:
  416. Common.logger(log_type, crawler).warning(f"get_videoList:{feeds[i]['photo']}\n")
  417. break
  418. videoResource = feeds[i]['photo']['videoResource']
  419. if 'h264' not in videoResource and 'hevc' not in videoResource:
  420. Common.logger(log_type, crawler).warning(f"get_videoList:{videoResource}\n")
  421. break
  422. # video_id
  423. if 'h264' in videoResource and 'videoId' in videoResource['h264']:
  424. video_id = videoResource['h264']['videoId']
  425. elif 'hevc' in videoResource and 'videoId' in videoResource['hevc']:
  426. video_id = videoResource['hevc']['videoId']
  427. else:
  428. video_id = ""
  429. # play_cnt
  430. if 'viewCount' not in feeds[i]['photo']:
  431. play_cnt = 0
  432. else:
  433. play_cnt = int(feeds[i]['photo']['viewCount'])
  434. # like_cnt
  435. if 'realLikeCount' not in feeds[i]['photo']:
  436. like_cnt = 0
  437. else:
  438. like_cnt = feeds[i]['photo']['realLikeCount']
  439. # publish_time
  440. if 'timestamp' not in feeds[i]['photo']:
  441. publish_time_stamp = 0
  442. publish_time_str = ''
  443. publish_time = 0
  444. else:
  445. publish_time_stamp = int(int(feeds[i]['photo']['timestamp']) / 1000)
  446. publish_time_str = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(publish_time_stamp))
  447. publish_time = int((int(time.time()) - publish_time_stamp) / (3600 * 24))
  448. # duration
  449. if 'duration' not in feeds[i]['photo']:
  450. duration = 0
  451. else:
  452. duration = int(int(feeds[i]['photo']['duration']) / 1000)
  453. # video_width / video_height / video_url
  454. mapping = {}
  455. for item in ['width', 'height']:
  456. try:
  457. val = str(videoResource['h264']['adaptationSet'][0]['representation'][0][item])
  458. except Exception:
  459. val = str(videoResource['hevc']['adaptationSet'][0]['representation'][0][item])
  460. except:
  461. val = ''
  462. mapping[item] = val
  463. video_width = int(mapping['width']) if mapping['width'] != '' else 0
  464. video_height = int(mapping['height']) if mapping['height'] != '' else 0
  465. # cover_url
  466. if 'coverUrl' not in feeds[i]['photo']:
  467. cover_url = ""
  468. else:
  469. cover_url = feeds[i]['photo']['coverUrl']
  470. # user_name / avatar_url
  471. try:
  472. user_name = feeds[i]['author']['name']
  473. avatar_url = feeds[i]['author']['headerUrl']
  474. except Exception:
  475. user_name = ''
  476. avatar_url = ''
  477. video_url = feeds[i]['photo']['photoUrl']
  478. video_dict = {'video_title': video_title,
  479. 'video_id': video_id,
  480. 'play_cnt': play_cnt,
  481. 'comment_cnt': 0,
  482. 'like_cnt': like_cnt,
  483. 'share_cnt': 0,
  484. 'video_width': video_width,
  485. 'video_height': video_height,
  486. 'duration': duration,
  487. 'publish_time': publish_time,
  488. 'publish_time_stamp': publish_time_stamp,
  489. 'publish_time_str': publish_time_str,
  490. 'user_name': user_name,
  491. 'user_id': out_uid,
  492. 'avatar_url': avatar_url,
  493. 'cover_url': cover_url,
  494. 'video_url': video_url,
  495. 'session': f"kuaishou{int(time.time())}"}
  496. rule_1 = cls.download_rule(video_dict, rule_dict_1)
  497. Common.logger(log_type, crawler).info(f"video_title:{video_title}")
  498. Common.logger(log_type, crawler).info(f"video_id:{video_id}\n")
  499. Common.logger(log_type, crawler).info(
  500. f"play_cnt:{video_dict['play_cnt']}{rule_dict_1['play_cnt']}, {eval(str(video_dict['play_cnt']) + str(rule_dict_1['play_cnt']))}")
  501. Common.logger(log_type, crawler).info(
  502. f"like_cnt:{video_dict['like_cnt']}{rule_dict_1['like_cnt']}, {eval(str(video_dict['like_cnt']) + str(rule_dict_1['like_cnt']))}")
  503. Common.logger(log_type, crawler).info(
  504. f"video_width:{video_dict['video_width']}{rule_dict_1['video_width']}, {eval(str(video_dict['video_width']) + str(rule_dict_1['video_width']))}")
  505. Common.logger(log_type, crawler).info(
  506. f"video_height:{video_dict['video_height']}{rule_dict_1['video_height']}, {eval(str(video_dict['video_height']) + str(rule_dict_1['video_height']))}")
  507. Common.logger(log_type, crawler).info(
  508. f"duration:{video_dict['duration']}{rule_dict_1['duration']}, {eval(str(video_dict['duration']) + str(rule_dict_1['duration']))}")
  509. Common.logger(log_type, crawler).info(
  510. f"publish_time:{video_dict['publish_time']}{rule_dict_1['publish_time']}, {eval(str(video_dict['publish_time']) + str(rule_dict_1['publish_time']))}")
  511. Common.logger(log_type, crawler).info(f"rule_1:{rule_1}\n")
  512. rule_2 = cls.download_rule(video_dict, rule_dict_2)
  513. Common.logger(log_type, crawler).info(
  514. f"play_cnt:{video_dict['play_cnt']}{rule_dict_2['play_cnt']}, {eval(str(video_dict['play_cnt']) + str(rule_dict_2['play_cnt']))}")
  515. Common.logger(log_type, crawler).info(
  516. f"like_cnt:{video_dict['like_cnt']}{rule_dict_2['like_cnt']}, {eval(str(video_dict['like_cnt']) + str(rule_dict_2['like_cnt']))}")
  517. Common.logger(log_type, crawler).info(
  518. f"video_width:{video_dict['video_width']}{rule_dict_2['video_width']}, {eval(str(video_dict['video_width']) + str(rule_dict_2['video_width']))}")
  519. Common.logger(log_type, crawler).info(
  520. f"video_height:{video_dict['video_height']}{rule_dict_2['video_height']}, {eval(str(video_dict['video_height']) + str(rule_dict_2['video_height']))}")
  521. Common.logger(log_type, crawler).info(
  522. f"duration:{video_dict['duration']}{rule_dict_2['duration']}, {eval(str(video_dict['duration']) + str(rule_dict_2['duration']))}")
  523. Common.logger(log_type, crawler).info(
  524. f"publish_time:{video_dict['publish_time']}{rule_dict_2['publish_time']}, {eval(str(video_dict['publish_time']) + str(rule_dict_2['publish_time']))}")
  525. Common.logger(log_type, crawler).info(f"rule_2:{rule_2}\n")
  526. if video_title == "" or video_url == "":
  527. Common.logger(log_type, crawler).info("无效视频\n")
  528. break
  529. elif rule_1 is True:
  530. if download_cnt_1 < int(
  531. rule_dict_1['download_cnt'].replace("=", "")[-1].replace("<", "")[-1].replace(">",
  532. "")[
  533. -1]):
  534. download_finished = cls.download_publish(log_type=log_type,
  535. crawler=crawler,
  536. strategy=strategy,
  537. video_dict=video_dict,
  538. rule_dict=rule_dict_1,
  539. our_uid=our_uid,
  540. oss_endpoint=oss_endpoint,
  541. env=env,
  542. machine=machine)
  543. if download_finished is True:
  544. download_cnt_1 += 1
  545. elif rule_2 is True:
  546. if download_cnt_2 < int(
  547. rule_dict_2['download_cnt'].replace("=", "")[-1].replace("<", "")[-1].replace(">",
  548. "")[
  549. -1]):
  550. download_finished = cls.download_publish(log_type=log_type,
  551. crawler=crawler,
  552. strategy=strategy,
  553. video_dict=video_dict,
  554. rule_dict=rule_dict_2,
  555. our_uid=our_uid,
  556. oss_endpoint=oss_endpoint,
  557. env=env,
  558. machine=machine)
  559. if download_finished is True:
  560. download_cnt_2 += 1
  561. else:
  562. Common.logger(log_type, crawler).info("不满足下载规则\n")
  563. # Common.logger(log_type, crawler).info(f"feeds: {feeds}\n")
  564. # if pcursor == "no_more":
  565. # Common.logger(log_type, crawler).info("已经到底了,没有更多内容了\n")
  566. # return
  567. Common.logger(log_type, crawler).info(f"作者:{out_uid},第一页抓取完毕,退出\n")
  568. except Exception as e:
  569. Common.logger(log_type, crawler).error(f"get_videoList:{e}\n")
  570. @classmethod
  571. def repeat_video(cls, log_type, crawler, video_id, video_title, publish_time, env, machine):
  572. sql = f""" select * from crawler_video where platform="{cls.platform}" and out_video_id="{video_id}" or (platform="{cls.platform}" and video_title="{video_title}" and publish_time="{publish_time}") """
  573. repeat_video = MysqlHelper.get_values(log_type, crawler, sql, env, machine)
  574. return len(repeat_video)
  575. @classmethod
  576. def download_publish(cls, log_type, crawler, strategy, video_dict, rule_dict, our_uid, oss_endpoint, env, machine):
  577. try:
  578. download_finished = False
  579. if cls.repeat_video(log_type, crawler, video_dict['video_id'], video_dict['video_title'],
  580. video_dict['publish_time_str'], env, machine) != 0:
  581. Common.logger(log_type, crawler).info('视频已下载\n')
  582. # elif video_dict['video_id'] in [x for y in Feishu.get_values_batch(log_type, crawler, "3cd128") for x in y]:
  583. # Common.logger(log_type, crawler).info('视频已下载\n')
  584. elif any(word if word in video_dict['video_title'] else False for word in
  585. cls.filter_words(log_type, crawler)) is True:
  586. Common.logger(log_type, crawler).info('标题已中过滤词\n')
  587. else:
  588. # 下载视频
  589. Common.download_method(log_type=log_type, crawler=crawler, text='video',
  590. title=video_dict['video_title'], url=video_dict['video_url'])
  591. md_title = md5(video_dict['video_title'].encode('utf8')).hexdigest()
  592. if os.path.getsize(f"./{crawler}/videos/{md_title}/video.mp4") == 0:
  593. # 删除视频文件夹
  594. shutil.rmtree(f"./{crawler}/videos/{md_title}")
  595. Common.logger(log_type, crawler).info("视频size=0,删除成功\n")
  596. return
  597. # ffmpeg_dict = Common.ffmpeg(log_type, crawler,
  598. # f"./{crawler}/videos/{video_dict['video_title']}/video.mp4")
  599. # if ffmpeg_dict is None or ffmpeg_dict['size'] == 0:
  600. # Common.logger(log_type, crawler).warning(f"下载的视频无效,已删除\n")
  601. # # 删除视频文件夹
  602. # shutil.rmtree(f"./{crawler}/videos/{video_dict['video_title']}")
  603. # return download_finished
  604. # 下载封面
  605. Common.download_method(log_type=log_type, crawler=crawler, text='cover',
  606. title=video_dict['video_title'], url=video_dict['cover_url'])
  607. # 保存视频信息至txt
  608. Common.save_video_info(log_type=log_type, crawler=crawler, video_dict=video_dict)
  609. # 上传视频
  610. Common.logger(log_type, crawler).info("开始上传视频...")
  611. our_video_id = Publish.upload_and_publish(log_type=log_type,
  612. crawler=crawler,
  613. strategy=strategy,
  614. our_uid=our_uid,
  615. env=env,
  616. oss_endpoint=oss_endpoint)
  617. if env == 'dev':
  618. our_video_link = f"https://testadmin.piaoquantv.com/cms/post-detail/{our_video_id}/info"
  619. else:
  620. our_video_link = f"https://admin.piaoquantv.com/cms/post-detail/{our_video_id}/info"
  621. Common.logger(log_type, crawler).info("视频上传完成")
  622. if our_video_id is None:
  623. Common.logger(log_type, crawler).warning(f"our_video_id:{our_video_id} 删除该视频文件夹")
  624. # 删除视频文件夹
  625. shutil.rmtree(f"./{crawler}/videos/{video_dict['video_title']}")
  626. return download_finished
  627. # 视频信息保存数据库
  628. insert_sql = f""" insert into crawler_video(video_id,
  629. user_id,
  630. out_user_id,
  631. platform,
  632. strategy,
  633. out_video_id,
  634. video_title,
  635. cover_url,
  636. video_url,
  637. duration,
  638. publish_time,
  639. play_cnt,
  640. crawler_rule,
  641. width,
  642. height)
  643. values({our_video_id},
  644. {our_uid},
  645. "{video_dict['user_id']}",
  646. "{cls.platform}",
  647. "定向爬虫策略",
  648. "{video_dict['video_id']}",
  649. "{video_dict['video_title']}",
  650. "{video_dict['cover_url']}",
  651. "{video_dict['video_url']}",
  652. {int(video_dict['duration'])},
  653. "{video_dict['publish_time_str']}",
  654. {int(video_dict['play_cnt'])},
  655. '{json.dumps(rule_dict)}',
  656. {int(video_dict['video_width'])},
  657. {int(video_dict['video_height'])}) """
  658. Common.logger(log_type, crawler).info(f"insert_sql:{insert_sql}")
  659. MysqlHelper.update_values(log_type, crawler, insert_sql, env, machine)
  660. Common.logger(log_type, crawler).info('视频信息插入数据库成功!\n')
  661. # 视频写入飞书
  662. Feishu.insert_columns(log_type, 'kuaishou', "fYdA8F", "ROWS", 1, 2)
  663. upload_time = int(time.time())
  664. values = [[our_video_id,
  665. time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(upload_time)),
  666. "定向榜",
  667. str(video_dict['video_id']),
  668. video_dict['video_title'],
  669. our_video_link,
  670. video_dict['play_cnt'],
  671. video_dict['comment_cnt'],
  672. video_dict['like_cnt'],
  673. video_dict['share_cnt'],
  674. video_dict['duration'],
  675. f"{video_dict['video_width']}*{video_dict['video_height']}",
  676. video_dict['publish_time_str'],
  677. video_dict['user_name'],
  678. video_dict['user_id'],
  679. video_dict['avatar_url'],
  680. video_dict['cover_url'],
  681. video_dict['video_url']]]
  682. time.sleep(1)
  683. Feishu.update_values(log_type, 'kuaishou', "fYdA8F", "E2:Z2", values)
  684. Common.logger(log_type, crawler).info(f"视频已保存至云文档\n")
  685. download_finished = True
  686. return download_finished
  687. except Exception as e:
  688. Common.logger(log_type, crawler).error(f"download_publish:{e}\n")
  689. @classmethod
  690. def get_follow_videos(cls, log_type, crawler, strategy, oss_endpoint, env, machine):
  691. user_list = cls.get_user_list(log_type=log_type, crawler=crawler, sheetid="bTSzxW", env=env, machine=machine)
  692. for user in user_list:
  693. out_uid = user["out_uid"]
  694. user_name = user["user_name"]
  695. our_uid = user["our_uid"]
  696. Common.logger(log_type, crawler).info(f"开始抓取 {user_name} 用户主页视频\n")
  697. try:
  698. cls.get_videoList(log_type=log_type,
  699. crawler=crawler,
  700. strategy=strategy,
  701. our_uid=our_uid,
  702. out_uid=out_uid,
  703. oss_endpoint=oss_endpoint,
  704. env=env,
  705. machine=machine)
  706. except Exception as e:
  707. Common.logger(log_type, crawler).info(f"用户:{user_name}, 抓取异常:{e}\n")
  708. continue
  709. # sleep_time = 120
  710. # Common.logger(log_type, crawler).info(f"休眠{sleep_time}秒\n")
  711. # time.sleep(sleep_time)
  712. if __name__ == "__main__":
  713. pass
  714. # Follow.get_videoList(log_type="follow",
  715. # crawler="kuaishou",
  716. # strategy="定向爬虫策略",
  717. # our_uid="6282431",
  718. # out_uid="3x3wsqjsxu5j6de",
  719. # oss_endpoint="out",
  720. # env="dev",
  721. # machine="local")
  722. # print(Follow.get_out_user_info("follow", "kuaishou", "3xgh4ja9be3wcaw"))
  723. # print(Follow.get_out_user_info("follow", "kuaishou", "3x5wgjhfc7tx8ue"))