kuaishou_follow.py 36 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672
  1. # -*- coding: utf-8 -*-
  2. # @Author: wangkun
  3. # @Time: 2023/2/24
  4. import os
  5. import random
  6. import shutil
  7. import string
  8. import sys
  9. import time
  10. from hashlib import md5
  11. import requests
  12. import json
  13. import urllib3
  14. from requests.adapters import HTTPAdapter
  15. sys.path.append(os.getcwd())
  16. from common.common import Common
  17. from common.feishu import Feishu
  18. from common.getuser import getUser
  19. from common.db import MysqlHelper
  20. from common.publish import Publish
  21. from common.public import random_title, get_config_from_mysql
  22. from common.public import get_user_from_mysql
  23. from common.userAgent import get_random_user_agent
  24. class KuaiShouFollow:
  25. platform = "快手"
  26. tag = "快手爬虫,定向爬虫策略"
  27. @classmethod
  28. def get_rule(cls, log_type, crawler, index):
  29. try:
  30. rule_sheet = Feishu.get_values_batch(log_type, crawler, "3iqG4z")
  31. if index == 1:
  32. rule_dict = {
  33. "play_cnt": f"{rule_sheet[1][1]}{rule_sheet[1][2]}",
  34. "video_width": f"{rule_sheet[2][1]}{rule_sheet[2][2]}",
  35. "video_height": f"{rule_sheet[3][1]}{rule_sheet[3][2]}",
  36. "like_cnt": f"{rule_sheet[4][1]}{rule_sheet[4][2]}",
  37. "duration": f"{rule_sheet[5][1]}{rule_sheet[5][2]}",
  38. "download_cnt": f"{rule_sheet[6][1]}{rule_sheet[6][2]}",
  39. "publish_time": f"{rule_sheet[7][1]}{rule_sheet[7][2]}",
  40. }
  41. # for k, v in rule_dict.items():
  42. # Common.logger(log_type, crawler).info(f"{k}:{v}")
  43. return rule_dict
  44. elif index == 2:
  45. rule_dict = {
  46. "play_cnt": f"{rule_sheet[9][1]}{rule_sheet[9][2]}",
  47. "video_width": f"{rule_sheet[10][1]}{rule_sheet[10][2]}",
  48. "video_height": f"{rule_sheet[11][1]}{rule_sheet[11][2]}",
  49. "like_cnt": f"{rule_sheet[12][1]}{rule_sheet[12][2]}",
  50. "duration": f"{rule_sheet[13][1]}{rule_sheet[13][2]}",
  51. "download_cnt": f"{rule_sheet[14][1]}{rule_sheet[14][2]}",
  52. "publish_time": f"{rule_sheet[15][1]}{rule_sheet[15][2]}",
  53. }
  54. # for k, v in rule_dict.items():
  55. # Common.logger(log_type, crawler).info(f"{k}:{v}")
  56. return rule_dict
  57. except Exception as e:
  58. Common.logger(log_type, crawler).error(f"get_rule:{e}\n")
  59. @classmethod
  60. def download_rule(cls, video_dict, rule_dict):
  61. if eval(f"{video_dict['play_cnt']}{rule_dict['play_cnt']}") is True \
  62. and eval(f"{video_dict['video_width']}{rule_dict['video_width']}") is True \
  63. and eval(f"{video_dict['video_height']}{rule_dict['video_height']}") is True \
  64. and eval(f"{video_dict['like_cnt']}{rule_dict['like_cnt']}") is True \
  65. and eval(f"{video_dict['duration']}{rule_dict['duration']}") is True \
  66. and eval(f"{video_dict['publish_time']}{rule_dict['publish_time']}") is True:
  67. return True
  68. else:
  69. return False
  70. # 过滤词库
  71. @classmethod
  72. def filter_words(cls, log_type, crawler):
  73. try:
  74. while True:
  75. filter_words_sheet = Feishu.get_values_batch(log_type, crawler, 'HIKVvs')
  76. if filter_words_sheet is None:
  77. Common.logger(log_type, crawler).warning(f"filter_words_sheet:{filter_words_sheet} 10秒钟后重试")
  78. continue
  79. filter_words_list = []
  80. for x in filter_words_sheet:
  81. for y in x:
  82. if y is None:
  83. pass
  84. else:
  85. filter_words_list.append(y)
  86. return filter_words_list
  87. except Exception as e:
  88. Common.logger(log_type, crawler).error(f'filter_words异常:{e}\n')
  89. # 获取站外用户信息
  90. @classmethod
  91. def get_out_user_info(cls, log_type, crawler, out_uid):
  92. try:
  93. url = "https://www.kuaishou.com/graphql"
  94. payload = json.dumps({
  95. "operationName": "visionProfile",
  96. "variables": {
  97. "userId": out_uid
  98. },
  99. "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"
  100. })
  101. s = string.ascii_lowercase
  102. r = random.choice(s)
  103. headers = {
  104. 'Accept': '*/*',
  105. 'Content-Type': 'application/json',
  106. 'Origin': 'https://www.kuaishou.com',
  107. <<<<<<< HEAD
  108. 'Cookie': 'kpf=PC_WEB; clientid=3; did=web_29de4c3ccc5977aced30922284f71d39; kpn=KUAISHOU_VISION',
  109. =======
  110. 'Cookie': 'kpf=PC_WEB; clientid=3; did=web_7cdc486ebd1aba220455a7781d6ae5b5{r}7; kpn=KUAISHOU_VISION;'.format(
  111. r=r),
  112. >>>>>>> wangkun
  113. 'Content-Length': '552',
  114. 'Accept-Language': 'zh-CN,zh-Hans;q=0.9',
  115. 'Host': 'www.kuaishou.com',
  116. 'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/15.6.1 Safari/605.1.15',
  117. 'Referer': 'https://www.kuaishou.com/profile/{}'.format(out_uid),
  118. 'Accept-Encoding': 'gzip, deflate, br',
  119. 'Connection': 'keep-alive'
  120. }
  121. urllib3.disable_warnings()
  122. s = requests.session()
  123. # max_retries=3 重试3次
  124. s.mount('http://', HTTPAdapter(max_retries=3))
  125. s.mount('https://', HTTPAdapter(max_retries=3))
  126. response = s.post(url=url, headers=headers, data=payload, proxies=Common.tunnel_proxies(), verify=False,
  127. timeout=5)
  128. response.close()
  129. # Common.logger(log_type, crawler).info(f"get_out_user_info_response:{response.text}")
  130. if response.status_code != 200:
  131. Common.logger(log_type, crawler).warning(f"get_out_user_info_response:{response.text}\n")
  132. return
  133. elif 'data' not in response.json():
  134. Common.logger(log_type, crawler).warning(f"get_out_user_info_response:{response.json()}\n")
  135. return
  136. elif 'visionProfile' not in response.json()['data']:
  137. Common.logger(log_type, crawler).warning(f"get_out_user_info_response:{response.json()['data']}\n")
  138. return
  139. elif 'userProfile' not in response.json()['data']['visionProfile']:
  140. Common.logger(log_type, crawler).warning(
  141. f"get_out_user_info_response:{response.json()['data']['visionProfile']['userProfile']}\n")
  142. return
  143. else:
  144. userProfile = response.json()['data']['visionProfile']['userProfile']
  145. # Common.logger(log_type, crawler).info(f"userProfile:{userProfile}")
  146. try:
  147. out_fans_str = str(userProfile['ownerCount']['fan'])
  148. except Exception:
  149. out_fans_str = "0"
  150. try:
  151. out_follow_str = str(userProfile['ownerCount']['follow'])
  152. except Exception:
  153. out_follow_str = "0"
  154. try:
  155. out_avatar_url = userProfile['profile']['headurl']
  156. except Exception:
  157. out_avatar_url = ""
  158. Common.logger(log_type, crawler).info(f"out_fans_str:{out_fans_str}")
  159. Common.logger(log_type, crawler).info(f"out_follow_str:{out_follow_str}")
  160. Common.logger(log_type, crawler).info(f"out_avatar_url:{out_avatar_url}")
  161. if "万" in out_fans_str:
  162. out_fans = int(float(out_fans_str.split("万")[0]) * 10000)
  163. else:
  164. out_fans = int(out_fans_str.replace(",", ""))
  165. if "万" in out_follow_str:
  166. out_follow = int(float(out_follow_str.split("万")[0]) * 10000)
  167. else:
  168. out_follow = int(out_follow_str.replace(",", ""))
  169. out_user_dict = {
  170. "out_fans": out_fans,
  171. "out_follow": out_follow,
  172. "out_avatar_url": out_avatar_url
  173. }
  174. Common.logger(log_type, crawler).info(f"out_user_dict:{out_user_dict}")
  175. return out_user_dict
  176. except Exception as e:
  177. Common.logger(log_type, crawler).error(f"get_out_user_info:{e}\n")
  178. # 获取用户信息列表
  179. @classmethod
  180. def get_user_list(cls, log_type, crawler, sheetid, env, machine):
  181. try:
  182. while True:
  183. user_sheet = Feishu.get_values_batch(log_type, crawler, sheetid)
  184. if user_sheet is None:
  185. Common.logger(log_type, crawler).warning(f"user_sheet:{user_sheet} 10秒钟后重试")
  186. continue
  187. our_user_list = []
  188. for i in range(1, len(user_sheet)):
  189. # for i in range(1, 2):
  190. out_uid = user_sheet[i][2]
  191. user_name = user_sheet[i][3]
  192. our_uid = user_sheet[i][6]
  193. our_user_link = user_sheet[i][7]
  194. if out_uid is None or user_name is None:
  195. Common.logger(log_type, crawler).info("空行\n")
  196. else:
  197. Common.logger(log_type, crawler).info(f"正在更新 {user_name} 用户信息\n")
  198. if our_uid is None:
  199. out_user_info = cls.get_out_user_info(log_type, crawler, out_uid)
  200. out_user_dict = {
  201. "out_uid": out_uid,
  202. "user_name": user_name,
  203. "out_avatar_url": out_user_info["out_avatar_url"],
  204. "out_create_time": '',
  205. "out_tag": '',
  206. "out_play_cnt": 0,
  207. "out_fans": out_user_info["out_fans"],
  208. "out_follow": out_user_info["out_follow"],
  209. "out_friend": 0,
  210. "out_like": 0,
  211. "platform": cls.platform,
  212. "tag": cls.tag,
  213. }
  214. our_user_dict = getUser.create_user(log_type=log_type, crawler=crawler,
  215. out_user_dict=out_user_dict, env=env, machine=machine)
  216. our_uid = our_user_dict['our_uid']
  217. our_user_link = our_user_dict['our_user_link']
  218. Feishu.update_values(log_type, crawler, sheetid, f'G{i + 1}:H{i + 1}',
  219. [[our_uid, our_user_link]])
  220. Common.logger(log_type, crawler).info(f'站内用户信息写入飞书成功!\n')
  221. our_user_list.append(our_user_dict)
  222. else:
  223. our_user_dict = {
  224. 'out_uid': out_uid,
  225. 'user_name': user_name,
  226. 'our_uid': our_uid,
  227. 'our_user_link': our_user_link,
  228. }
  229. our_user_list.append(our_user_dict)
  230. return our_user_list
  231. except Exception as e:
  232. Common.logger(log_type, crawler).error(f'get_user_list:{e}\n')
  233. # 处理视频标题
  234. @classmethod
  235. def video_title(cls, log_type, crawler, env, title):
  236. title_split1 = title.split(" #")
  237. if title_split1[0] != "":
  238. title1 = title_split1[0]
  239. else:
  240. title1 = title_split1[-1]
  241. title_split2 = title1.split(" #")
  242. if title_split2[0] != "":
  243. title2 = title_split2[0]
  244. else:
  245. title2 = title_split2[-1]
  246. title_split3 = title2.split("@")
  247. if title_split3[0] != "":
  248. title3 = title_split3[0]
  249. else:
  250. title3 = title_split3[-1]
  251. video_title = title3.strip().replace("\n", "") \
  252. .replace("/", "").replace("快手", "").replace(" ", "") \
  253. .replace(" ", "").replace("&NBSP", "").replace("\r", "") \
  254. .replace("#", "").replace(".", "。").replace("\\", "") \
  255. .replace(":", "").replace("*", "").replace("?", "") \
  256. .replace("?", "").replace('"', "").replace("<", "") \
  257. .replace(">", "").replace("|", "").replace("@", "").replace('"', '').replace("'", '')[:40]
  258. if video_title.replace(" ", "") == "" or video_title == "。。。" or video_title == "...":
  259. return random_title(log_type, crawler, env, text='title')
  260. else:
  261. return video_title
  262. @classmethod
  263. def get_videoList(cls, log_type, crawler, strategy, our_uid, out_uid, oss_endpoint, env, machine, pcursor=""):
  264. download_cnt_1, download_cnt_2 = 0, 0
  265. rule_dict_1 = cls.get_rule(log_type, crawler, 1)
  266. rule_dict_2 = cls.get_rule(log_type, crawler, 2)
  267. if rule_dict_1 is None or rule_dict_2 is None:
  268. Common.logger(log_type, crawler).warning(f"rule_dict is None")
  269. return
  270. url = "https://www.kuaishou.com/graphql"
  271. payload = json.dumps({
  272. "operationName": "visionProfilePhotoList",
  273. "variables": {
  274. "userId": out_uid,
  275. "pcursor": "",
  276. "page": "profile"
  277. },
  278. "query": "fragment photoContent on PhotoEntity {\n id\n duration\n caption\n originCaption\n likeCount\n viewCount\n commentCount\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"
  279. })
  280. headers = {
  281. 'Accept': '*/*',
  282. 'Content-Type': 'application/json',
  283. 'Origin': 'https://www.kuaishou.com',
  284. <<<<<<< HEAD
  285. 'Cookie': 'kpf=PC_WEB; clientid=3; did=web_29de4c3ccc5977aced30922284f71d39; kpn=KUAISHOU_VISION',
  286. =======
  287. 'Cookie': 'kpf=PC_WEB; clientid=3; did=web_a1aaee86bc76103992906d83d6f100c2; kpn=KUAISHOU_VISION',
  288. >>>>>>> wangkun
  289. 'Content-Length': '1260',
  290. 'Accept-Language': 'zh-CN,zh-Hans;q=0.9',
  291. 'Host': 'www.kuaishou.com',
  292. 'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/15.6.1 Safari/605.1.15',
  293. 'Referer': 'https://www.kuaishou.com/profile/{}'.format(out_uid),
  294. 'Accept-Encoding': 'gzip, deflate, br',
  295. 'Connection': 'keep-alive'
  296. }
  297. try:
  298. response = requests.post(url=url, headers=headers, data=payload, proxies=Common.tunnel_proxies(),
  299. verify=False, timeout=10)
  300. feeds = response.json()['data']['visionProfilePhotoList']['feeds']
  301. except Exception as e:
  302. Common.logger(log_type, crawler).error(f"get_videoList:{e},response:{response.text}")
  303. return
  304. if not feeds:
  305. Common.logger(log_type, crawler).info("没有更多视频啦 ~\n")
  306. return
  307. pcursor = response.json()['data']['visionProfilePhotoList']['pcursor']
  308. # Common.logger(log_type, crawler).info(f"feeds0: {feeds}\n")
  309. for i in range(len(feeds)):
  310. # if 'photo' not in feeds[i]:
  311. # Common.logger(log_type, crawler).warning(f"get_videoList:{feeds[i]}\n")
  312. # break
  313. # video_title
  314. if 'caption' not in feeds[i]['photo']:
  315. video_title = random_title(log_type, crawler, env, text='title')
  316. elif feeds[i]['photo']['caption'].strip() == "":
  317. video_title = random_title(log_type, crawler, env, text='title')
  318. else:
  319. video_title = cls.video_title(log_type, crawler, env, feeds[i]['photo']['caption'])
  320. if 'videoResource' not in feeds[i]['photo'] \
  321. and 'manifest' not in feeds[i]['photo'] \
  322. and 'manifestH265' not in feeds[i]['photo']:
  323. Common.logger(log_type, crawler).warning(f"get_videoList:{feeds[i]['photo']}\n")
  324. break
  325. videoResource = feeds[i]['photo']['videoResource']
  326. if 'h264' not in videoResource and 'hevc' not in videoResource:
  327. Common.logger(log_type, crawler).warning(f"get_videoList:{videoResource}\n")
  328. break
  329. # video_id
  330. if 'h264' in videoResource and 'videoId' in videoResource['h264']:
  331. video_id = videoResource['h264']['videoId']
  332. elif 'hevc' in videoResource and 'videoId' in videoResource['hevc']:
  333. video_id = videoResource['hevc']['videoId']
  334. else:
  335. video_id = ""
  336. # play_cnt
  337. if 'viewCount' not in feeds[i]['photo']:
  338. play_cnt = 0
  339. else:
  340. play_cnt = int(feeds[i]['photo']['viewCount'])
  341. # like_cnt
  342. if 'realLikeCount' not in feeds[i]['photo']:
  343. like_cnt = 0
  344. else:
  345. like_cnt = feeds[i]['photo']['realLikeCount']
  346. # publish_time
  347. if 'timestamp' not in feeds[i]['photo']:
  348. publish_time_stamp = 0
  349. publish_time_str = ''
  350. publish_time = 0
  351. else:
  352. publish_time_stamp = int(int(feeds[i]['photo']['timestamp']) / 1000)
  353. publish_time_str = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(publish_time_stamp))
  354. publish_time = int((int(time.time()) - publish_time_stamp) / (3600 * 24))
  355. # duration
  356. if 'duration' not in feeds[i]['photo']:
  357. duration = 0
  358. else:
  359. duration = int(int(feeds[i]['photo']['duration']) / 1000)
  360. # video_width / video_height / video_url
  361. mapping = {}
  362. for item in ['width', 'height']:
  363. try:
  364. val = str(videoResource['h264']['adaptationSet'][0]['representation'][0][item])
  365. except:
  366. val = str(videoResource['hevc']['adaptationSet'][0]['representation'][0][item])
  367. mapping[item] = val
  368. video_width = int(mapping['width']) if mapping['width'] else 0
  369. video_height = int(mapping['height']) if mapping['height'] else 0
  370. # cover_url
  371. if 'coverUrl' not in feeds[i]['photo']:
  372. cover_url = ""
  373. else:
  374. cover_url = feeds[i]['photo']['coverUrl']
  375. # user_name / avatar_url
  376. user_name = feeds[i]['author']['name']
  377. avatar_url = feeds[i]['author']['headerUrl']
  378. video_url = feeds[i]['photo']['photoUrl']
  379. video_dict = {'video_title': video_title,
  380. 'video_id': video_id,
  381. 'play_cnt': play_cnt,
  382. 'comment_cnt': 0,
  383. 'like_cnt': like_cnt,
  384. 'share_cnt': 0,
  385. 'video_width': video_width,
  386. 'video_height': video_height,
  387. 'duration': duration,
  388. 'publish_time': publish_time,
  389. 'publish_time_stamp': publish_time_stamp,
  390. 'publish_time_str': publish_time_str,
  391. 'user_name': user_name,
  392. 'user_id': out_uid,
  393. 'avatar_url': avatar_url,
  394. 'cover_url': cover_url,
  395. 'video_url': video_url,
  396. 'session': f"kuaishou{int(time.time())}"}
  397. rule_1 = cls.download_rule(video_dict, rule_dict_1)
  398. Common.logger(log_type, crawler).info(f"video_title:{video_title}")
  399. Common.logger(log_type, crawler).info(f"video_id:{video_id}\n")
  400. Common.logger(log_type, crawler).info(
  401. f"play_cnt:{video_dict['play_cnt']}{rule_dict_1['play_cnt']}, {eval(str(video_dict['play_cnt']) + str(rule_dict_1['play_cnt']))}")
  402. Common.logger(log_type, crawler).info(
  403. f"like_cnt:{video_dict['like_cnt']}{rule_dict_1['like_cnt']}, {eval(str(video_dict['like_cnt']) + str(rule_dict_1['like_cnt']))}")
  404. Common.logger(log_type, crawler).info(
  405. f"video_width:{video_dict['video_width']}{rule_dict_1['video_width']}, {eval(str(video_dict['video_width']) + str(rule_dict_1['video_width']))}")
  406. Common.logger(log_type, crawler).info(
  407. f"video_height:{video_dict['video_height']}{rule_dict_1['video_height']}, {eval(str(video_dict['video_height']) + str(rule_dict_1['video_height']))}")
  408. Common.logger(log_type, crawler).info(
  409. f"duration:{video_dict['duration']}{rule_dict_1['duration']}, {eval(str(video_dict['duration']) + str(rule_dict_1['duration']))}")
  410. Common.logger(log_type, crawler).info(
  411. f"publish_time:{video_dict['publish_time']}{rule_dict_1['publish_time']}, {eval(str(video_dict['publish_time']) + str(rule_dict_1['publish_time']))}")
  412. Common.logger(log_type, crawler).info(f"rule_1:{rule_1}\n")
  413. rule_2 = cls.download_rule(video_dict, rule_dict_2)
  414. Common.logger(log_type, crawler).info(
  415. f"play_cnt:{video_dict['play_cnt']}{rule_dict_2['play_cnt']}, {eval(str(video_dict['play_cnt']) + str(rule_dict_2['play_cnt']))}")
  416. Common.logger(log_type, crawler).info(
  417. f"like_cnt:{video_dict['like_cnt']}{rule_dict_2['like_cnt']}, {eval(str(video_dict['like_cnt']) + str(rule_dict_2['like_cnt']))}")
  418. Common.logger(log_type, crawler).info(
  419. f"video_width:{video_dict['video_width']}{rule_dict_2['video_width']}, {eval(str(video_dict['video_width']) + str(rule_dict_2['video_width']))}")
  420. Common.logger(log_type, crawler).info(
  421. f"video_height:{video_dict['video_height']}{rule_dict_2['video_height']}, {eval(str(video_dict['video_height']) + str(rule_dict_2['video_height']))}")
  422. Common.logger(log_type, crawler).info(
  423. f"duration:{video_dict['duration']}{rule_dict_2['duration']}, {eval(str(video_dict['duration']) + str(rule_dict_2['duration']))}")
  424. Common.logger(log_type, crawler).info(
  425. f"publish_time:{video_dict['publish_time']}{rule_dict_2['publish_time']}, {eval(str(video_dict['publish_time']) + str(rule_dict_2['publish_time']))}")
  426. Common.logger(log_type, crawler).info(f"rule_2:{rule_2}\n")
  427. if video_title == "" or video_url == "":
  428. Common.logger(log_type, crawler).info("无效视频\n")
  429. continue
  430. elif rule_1 is True:
  431. if download_cnt_1 < int(
  432. rule_dict_1['download_cnt'].replace("=", "")[-1].replace("<", "")[-1].replace(">",
  433. "")[
  434. -1]):
  435. cls.download_publish(log_type=log_type,
  436. crawler=crawler,
  437. strategy=strategy,
  438. video_dict=video_dict,
  439. rule_dict=rule_dict_1,
  440. our_uid=our_uid,
  441. oss_endpoint=oss_endpoint,
  442. env=env,
  443. machine=machine)
  444. # if download_finished is True:
  445. # download_cnt_1 += 1
  446. elif rule_2 is True:
  447. if download_cnt_2 < int(
  448. rule_dict_2['download_cnt'].replace("=", "")[-1].replace("<", "")[-1].replace(">",
  449. "")[
  450. -1]):
  451. cls.download_publish(log_type=log_type,
  452. crawler=crawler,
  453. strategy=strategy,
  454. video_dict=video_dict,
  455. rule_dict=rule_dict_2,
  456. our_uid=our_uid,
  457. oss_endpoint=oss_endpoint,
  458. env=env,
  459. machine=machine)
  460. # if download_finished is True:
  461. # download_cnt_2 += 1
  462. else:
  463. Common.logger(log_type, crawler).info("不满足下载规则\n")
  464. # Common.logger(log_type, crawler).info(f"feeds: {feeds}\n")
  465. # if pcursor == "no_more":
  466. # Common.logger(log_type, crawler).info(f"作者,{out_uid},已经到底了,没有更多内容了\n")
  467. # return
  468. # cls.get_videoList(log_type, crawler, strategy, our_uid, out_uid, oss_endpoint, env, machine,
  469. # pcursor=pcursor)
  470. # time.sleep(random.randint(1, 3))
  471. @classmethod
  472. def repeat_video(cls, log_type, crawler, video_id, video_title, publish_time, env, machine):
  473. 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}") """
  474. repeat_video = MysqlHelper.get_values(log_type, crawler, sql, env, machine)
  475. return len(repeat_video)
  476. @classmethod
  477. def download_publish(cls, log_type, crawler, strategy, video_dict, rule_dict, our_uid, oss_endpoint, env, machine):
  478. try:
  479. filter_words = get_config_from_mysql(log_type, crawler, env, text='filter')
  480. for filter_word in filter_words:
  481. if filter_word in video_dict['video_title']:
  482. Common.logger(log_type, crawler).info('标题已中过滤词:{}\n', video_dict['video_title'])
  483. return
  484. download_finished = False
  485. if cls.repeat_video(log_type, crawler, video_dict['video_id'], video_dict['video_title'],
  486. video_dict['publish_time_str'], env, machine) != 0:
  487. Common.logger(log_type, crawler).info('视频已下载\n')
  488. else:
  489. # 下载视频
  490. Common.download_method(log_type=log_type, crawler=crawler, text='video',
  491. title=video_dict['video_title'], url=video_dict['video_url'])
  492. md_title = md5(video_dict['video_title'].encode('utf8')).hexdigest()
  493. if os.path.getsize(f"./{crawler}/videos/{md_title}/video.mp4") == 0:
  494. # 删除视频文件夹
  495. shutil.rmtree(f"./{crawler}/videos/{md_title}")
  496. Common.logger(log_type, crawler).info("视频size=0,删除成功\n")
  497. return
  498. # ffmpeg_dict = Common.ffmpeg(log_type, crawler,
  499. # f"./{crawler}/videos/{video_dict['video_title']}/video.mp4")
  500. # if ffmpeg_dict is None or ffmpeg_dict['size'] == 0:
  501. # Common.logger(log_type, crawler).warning(f"下载的视频无效,已删除\n")
  502. # # 删除视频文件夹
  503. # shutil.rmtree(f"./{crawler}/videos/{video_dict['video_title']}")
  504. # return download_finished
  505. # 下载封面
  506. Common.download_method(log_type=log_type, crawler=crawler, text='cover',
  507. title=video_dict['video_title'], url=video_dict['cover_url'])
  508. # 保存视频信息至txt
  509. Common.save_video_info(log_type=log_type, crawler=crawler, video_dict=video_dict)
  510. # 上传视频
  511. Common.logger(log_type, crawler).info("开始上传视频...")
  512. our_video_id = Publish.upload_and_publish(log_type=log_type,
  513. crawler=crawler,
  514. strategy=strategy,
  515. our_uid=our_uid,
  516. env=env,
  517. oss_endpoint=oss_endpoint)
  518. if env == 'dev':
  519. our_video_link = f"https://testadmin.piaoquantv.com/cms/post-detail/{our_video_id}/info"
  520. else:
  521. our_video_link = f"https://admin.piaoquantv.com/cms/post-detail/{our_video_id}/info"
  522. Common.logger(log_type, crawler).info("视频上传完成")
  523. if our_video_id is None:
  524. Common.logger(log_type, crawler).warning(f"our_video_id:{our_video_id} 删除该视频文件夹")
  525. # 删除视频文件夹
  526. shutil.rmtree(f"./{crawler}/videos/{video_dict['video_title']}")
  527. return download_finished
  528. # 视频信息保存数据库
  529. insert_sql = f""" insert into crawler_video(video_id,
  530. user_id,
  531. out_user_id,
  532. platform,
  533. strategy,
  534. out_video_id,
  535. video_title,
  536. cover_url,
  537. video_url,
  538. duration,
  539. publish_time,
  540. play_cnt,
  541. crawler_rule,
  542. width,
  543. height)
  544. values({our_video_id},
  545. {our_uid},
  546. "{video_dict['user_id']}",
  547. "{cls.platform}",
  548. "定向爬虫策略",
  549. "{video_dict['video_id']}",
  550. "{video_dict['video_title']}",
  551. "{video_dict['cover_url']}",
  552. "{video_dict['video_url']}",
  553. {int(video_dict['duration'])},
  554. "{video_dict['publish_time_str']}",
  555. {int(video_dict['play_cnt'])},
  556. '{json.dumps(rule_dict)}',
  557. {int(video_dict['video_width'])},
  558. {int(video_dict['video_height'])}) """
  559. Common.logger(log_type, crawler).info(f"insert_sql:{insert_sql}")
  560. MysqlHelper.update_values(log_type, crawler, insert_sql, env, machine)
  561. Common.logger(log_type, crawler).info('视频信息插入数据库成功!\n')
  562. # 视频写入飞书
  563. Feishu.insert_columns(log_type, 'kuaishou', "fYdA8F", "ROWS", 1, 2)
  564. upload_time = int(time.time())
  565. values = [[our_video_id,
  566. time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(upload_time)),
  567. "定向榜",
  568. str(video_dict['video_id']),
  569. video_dict['video_title'],
  570. our_video_link,
  571. video_dict['play_cnt'],
  572. video_dict['comment_cnt'],
  573. video_dict['like_cnt'],
  574. video_dict['share_cnt'],
  575. video_dict['duration'],
  576. f"{video_dict['video_width']}*{video_dict['video_height']}",
  577. video_dict['publish_time_str'],
  578. video_dict['user_name'],
  579. video_dict['user_id'],
  580. video_dict['avatar_url'],
  581. video_dict['cover_url'],
  582. video_dict['video_url']]]
  583. time.sleep(1)
  584. Feishu.update_values(log_type, 'kuaishou', "fYdA8F", "E2:Z2", values)
  585. Common.logger(log_type, crawler).info(f"视频已保存至云文档\n")
  586. download_finished = True
  587. return download_finished
  588. except Exception as e:
  589. Common.logger(log_type, crawler).error(f"download_publish:{e}\n")
  590. @classmethod
  591. def get_follow_videos(cls, log_type, crawler, strategy, oss_endpoint, env, machine):
  592. # user_list = cls.get_user_list(log_type=log_type, crawler=crawler, sheetid="bTSzxW", env=env, machine=machine)
  593. user_list = get_user_from_mysql(log_type, crawler, crawler, env)
  594. for user in user_list:
  595. try:
  596. spider_link = user["link"]
  597. out_uid = spider_link.split('/')[-1]
  598. user_name = user["nick_name"]
  599. our_uid = user["uid"]
  600. Common.logger(log_type, crawler).info(f"开始抓取 {user_name} 用户主页视频\n")
  601. cls.get_videoList(log_type=log_type,
  602. crawler=crawler,
  603. strategy=strategy,
  604. our_uid=our_uid,
  605. out_uid=out_uid,
  606. oss_endpoint=oss_endpoint,
  607. env=env,
  608. machine=machine)
  609. except Exception as e:
  610. continue
  611. if __name__ == "__main__":
  612. KuaiShouFollow.get_videoList(log_type="follow",
  613. crawler="kuaishou",
  614. strategy="定向爬虫策略",
  615. our_uid="54719554",
  616. out_uid="3xnk3wbm3vfiha6",
  617. oss_endpoint="out",
  618. env="dev",
  619. machine="local")
  620. # print(KuaiShouFollow.get_out_user_info("follow", "kuaishou", "3xnk3wbm3vfiha6"))
  621. # print(Follow.get_out_user_info("follow", "kuaishou", "3x5wgjhfc7tx8ue"))