Use requests.Session.

This commit is contained in:
voussoir 2020-11-10 20:14:33 -08:00
parent 26a5b69b4f
commit 47418c464c

View file

@ -16,6 +16,8 @@ FILENAME_BADCHARS = '\\/:*?<>|"'
WEBSTORE_URL = 'https://chrome.google.com/webstore/detail/x/{extension_id}' WEBSTORE_URL = 'https://chrome.google.com/webstore/detail/x/{extension_id}'
CRX_URL = 'https://clients2.google.com/service/update2/crx?response=redirect&prodversion=83.0.4103.116&acceptformat=crx2,crx3&x=id%3D{extension_id}%26uc' CRX_URL = 'https://clients2.google.com/service/update2/crx?response=redirect&prodversion=83.0.4103.116&acceptformat=crx2,crx3&x=id%3D{extension_id}%26uc'
session = requests.Session()
def sanitize_filename(name): def sanitize_filename(name):
for c in FILENAME_BADCHARS: for c in FILENAME_BADCHARS:
name = name.replace(c, '-') name = name.replace(c, '-')
@ -23,7 +25,7 @@ def sanitize_filename(name):
def get_webstore_name_version(extension_id): def get_webstore_name_version(extension_id):
url = WEBSTORE_URL.format(extension_id=extension_id) url = WEBSTORE_URL.format(extension_id=extension_id)
response = requests.get(url) response = session.get(url)
try: try:
name = response.text name = response.text
name = name.split('meta property="og:title" content="')[1] name = name.split('meta property="og:title" content="')[1]
@ -50,7 +52,7 @@ def get_crx_name_version(crx_bytes):
def getcrx(extension_id, auto_overwrite=None): def getcrx(extension_id, auto_overwrite=None):
url = CRX_URL.format(extension_id=extension_id) url = CRX_URL.format(extension_id=extension_id)
response = requests.get(url) response = session.get(url)
response.raise_for_status() response.raise_for_status()
(name, version) = get_webstore_name_version(extension_id) (name, version) = get_webstore_name_version(extension_id)