def get_extracted_zip(self):
if not self.download_url or not self.extract_to:
return
with busy_dialog():
response = self.open_url(self.download_url)
if not response:
xbmcgui.Dialog().ok(ADDON.getAddonInfo('name'),
ADDON.getLocalizedString(32058))
return
if not os.path.exists(self.extract_to):
os.makedirs(self.extract_to)
if xbmcgui.Dialog().yesno(ADDON.getAddonInfo('name'),
self.msg_cleardir):
with busy_dialog():
self.clear_dir(self.extract_to)
with busy_dialog():
num_files = 0
with zipfile.ZipFile(BytesIO(response.content)) as downloaded_zip:
for item in [
x for x in downloaded_zip.namelist()
if x.endswith('.json')
]:
filename = os.path.basename(item)
if not filename:
continue
_file = downloaded_zip.open(item)
with open(os.path.join(self.extract_to, filename),
'w') as target:
target.write(_file.read())
num_files += 1
try:
_tempzip = os.path.join(self.extract_to, 'temp.zip')
os.remove(_tempzip)
except Exception as e:
kodi_log(u'Could not delete package {0}: {1}'.format(
_tempzip, str(e)))
if num_files:
xbmcgui.Dialog().ok(
ADDON.getAddonInfo('name'),
u'{0}\n\n{1} {2}.'.format(ADDON.getLocalizedString(32059),
num_files,
ADDON.getLocalizedString(32060)))
def open_url(self, url, stream=False, check=False, cred=None, count=0):
if not url:
return False
valid = self.check_url(url, cred)
if not valid:
return False
if check:
return True
if valid == 'auth' and not cred:
cred = (xbmcgui.Dialog().input(heading=xbmc.getLocalizedString(1014)) or '', xbmcgui.Dialog().input(heading=xbmc.getLocalizedString(733), option=xbmcgui.ALPHANUM_HIDE_INPUT) or '')
response = requests.get(url, timeout=10.000, stream=stream, auth=cred)
if response.status_code == 401:
if count > 2 or not xbmcgui.Dialog().yesno(ADDON.getAddonInfo('name'), ADDON.getLocalizedString(32056), yeslabel=ADDON.getLocalizedString(32057), nolabel=xbmc.getLocalizedString(222)):
xbmcgui.Dialog().ok(ADDON.getAddonInfo('name'), ADDON.getLocalizedString(32055))
return False
count += 1
cred = (xbmcgui.Dialog().input(heading=xbmc.getLocalizedString(1014)) or '', xbmcgui.Dialog().input(heading=xbmc.getLocalizedString(733), option=xbmcgui.ALPHANUM_HIDE_INPUT) or '')
response = self.open_url(url, stream, check, cred, count)
return response
def get_gzip_text(self):
if not self.download_url:
return
with busy_dialog():
response = self.open_url(self.download_url)
if not response:
xbmcgui.Dialog().ok(ADDON.getAddonInfo('name'), ADDON.getLocalizedString(32058))
return
with gzip.GzipFile(fileobj=BytesIO(response.content)) as downloaded_gzip:
content = downloaded_gzip.read()
return content
