Sfoglia il codice sorgente

update 获取随机useragent方法修改

lierqiang 2 anni fa
parent
commit
b082fe9243

+ 1 - 1
common/userAgent.py

@@ -1968,7 +1968,7 @@ user_agent_dict = {
 }
 
 
-def random_user_agent(source):
+def get_random_header(source):
     headers = {'User-Agent': random.choice(user_agent_dict[source])}
     return headers
 

+ 2 - 3
xigua/xigua_search/xigua_search.py

@@ -5,7 +5,6 @@ import base64
 import json
 import os
 import random
-import re
 import shutil
 import string
 import sys
@@ -21,7 +20,7 @@ from common.users import Users
 from common.common import Common
 from common.feishu import Feishu
 from common.publish import Publish
-from common.userAgent import get_random_user_agent, random_user_agent
+from common.userAgent import get_random_user_agent, get_random_header
 
 
 class XiguaSearch:
@@ -591,7 +590,7 @@ class XiguaSearch:
                 "&device_brand=smartisan&os_api=19&os_version=4.4.2&uuid=864593021012562&openudid=e23a5ff037ef2d1a" \
                 "&manifest_version_code=532&resolution=1080*1920&dpi=480&update_version_code=5320".format(
             video_id=item_id)
-        res = requests.get(url=d_url, headers=random_user_agent('pc'), proxies=Common.tunnel_proxies())
+        res = requests.get(url=d_url, headers=get_random_header('pc'), proxies=Common.tunnel_proxies())
         data = json.loads(res.text)['data']
         item_counter = data['h5_extra']['itemCell']['itemCounter']
         user_info = data['user_info']

+ 3 - 3
youtube/youtube_follow/youtube_follow_scheduling.py

@@ -17,7 +17,7 @@ from common.feishu import Feishu
 from common.users import Users
 from common.publish import Publish
 from common.translate import Translate
-from common.userAgent import random_user_agent, get_random_user_agent
+from common.userAgent import get_random_header, get_random_user_agent
 
 
 def format_nums(data):
@@ -57,7 +57,7 @@ class YoutubeFollowSchedule:
         """
         try:
             url = f'https://www.youtube.com/{out_user_id}/about'
-            res = requests.get(url=url, headers=random_user_agent('pc'))
+            res = requests.get(url=url, headers=get_random_header('pc'))
             info = re.findall(r'var ytInitialData = (.*?);</script>', res.text, re.S)[0]
             data = json.loads(info)
             header = data['header']['c4TabbedHeaderRenderer']
@@ -407,7 +407,7 @@ class YoutubeFollowSchedule:
     @classmethod
     def get_first_page(cls, user_url):
         try:
-            res = requests.get(url=user_url, headers=random_user_agent('pc'))
+            res = requests.get(url=user_url, headers=get_random_header('pc'))
             info = re.findall(r'var ytInitialData = (.*?);', res.text, re.S)[0]
             ytInitialData = json.loads(info)
             video_list = \