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