danvics/python-validity
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

validity-sensors-initializer: Set referrer and user agent

Browse source
This commit is contained in:
Marco Trevisan (Treviño) 2020-06-10 13:14:33 +02:00
parent 37bf786c01
commit c43b99a02e
1 changed files with 14 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
validity-sensors-initializer.py
View file

@ -42,8 +42,14 @@ class VFS(Enum):
DEV_97 = 0x0097 DEV_97 = 0x0097
DEFAULT_URIS = { DEFAULT_URIS = {
VFS.DEV_90: 'https://download.lenovo.com/pccbbs/mobiles/n1cgn08w.exe', VFS.DEV_90: {
VFS.DEV_97: 'https://download.lenovo.com/pccbbs/mobiles/n1mgf03w.exe', 'driver': 'https://download.lenovo.com/pccbbs/mobiles/n1cgn08w.exe',
'referral': 'https://support.lenovo.com/us/en/downloads/DS120491',
},
VFS.DEV_97: {
'driver': 'https://download.lenovo.com/pccbbs/mobiles/n1mgf03w.exe',
'referral': 'https://download.lenovo.com/pccbbs/mobiles/n1mgf03w.exe'
}
} }
DEFAULT_FW_NAMES = { DEFAULT_FW_NAMES = {
@ -88,13 +94,17 @@ class VFSInitializer():
self.open_device(init=True) self.open_device(init=True)
def download_and_extract_fw(self, fwdir, fwuri=None): def download_and_extract_fw(self, fwdir, fwuri=None):
fwuri = fwuri if fwuri else DEFAULT_URIS[self.dev_type] fwuri = fwuri if fwuri else DEFAULT_URIS[self.dev_type]['driver']
fwarchive = os.path.join(fwdir, 'fwinstaller.exe') fwarchive = os.path.join(fwdir, 'fwinstaller.exe')
fwname = DEFAULT_FW_NAMES[self.dev_type] fwname = DEFAULT_FW_NAMES[self.dev_type]
print('Downloading {} to extract {}'.format(fwuri, fwname)) print('Downloading {} to extract {}'.format(fwuri, fwname))
with urllib.request.urlopen(fwuri) as response: req = urllib.request.Request(fwuri)
req.add_header('Referer', DEFAULT_URIS[self.dev_type].get('referral', ''))
req.add_header('User-Agent', 'Mozilla/5.0 (X11; U; Linux)')
with urllib.request.urlopen(req) as response:
with open(fwarchive, 'wb') as out_file: with open(fwarchive, 'wb') as out_file:
out_file.write(response.read()) out_file.write(response.read())