mirror of https://github.com/morpheus65535/bazarr
39 lines
1.5 KiB
Python
39 lines
1.5 KiB
Python
|
import os
|
||
|
|
import json
|
||
|
|
import random
|
||
|
|
import logging
|
||
|
|
|
||
|
|
from collections import OrderedDict
|
||
|
|
|
||
|
|
##########################################################################################################################################################
|
||
|
|
|
||
|
|
|
||
|
|
class User_Agent():
|
||
|
|
|
||
|
|
##########################################################################################################################################################
|
||
|
|
|
||
|
|
def __init__(self, *args, **kwargs):
|
||
|
|
self.headers = None
|
||
|
|
self.loadUserAgent(*args, **kwargs)
|
||
|
|
|
||
|
|
##########################################################################################################################################################
|
||
|
|
|
||
|
|
def loadUserAgent(self, *args, **kwargs):
|
||
|
|
browser = kwargs.pop('browser', 'chrome')
|
||
|
|
|
||
|
|
with open(os.path.join(os.path.dirname(__file__), 'browsers.json'), 'r') as file:
|
||
|
|
user_agents = json.load(file, object_pairs_hook=OrderedDict)
|
||
|
|
|
||
|
|
if not user_agents.get(browser):
|
||
|
|
logging.error('Sorry "{}" browser User-Agent was not found.'.format(browser))
|
||
|
|
raise
|
||
|
|
|
||
|
|
user_agent = random.choice(user_agents.get(browser))
|
||
|
|
|
||
|
|
self.headers = user_agent.get('headers')
|
||
|
|
self.headers['User-Agent'] = random.choice(user_agent.get('User-Agent'))
|
||
|
|
|
||
|
|
if not kwargs.get('allow_brotli', False):
|
||
|
|
if 'br' in self.headers['Accept-Encoding']:
|
||
|
|
self.headers['Accept-Encoding'] = ','.join([encoding for encoding in self.headers['Accept-Encoding'].split(',') if encoding.strip() != 'br']).strip()
|