def get_ia_addon(required=False, install=True):
try:
if install:
xbmc.executebuiltin('InstallAddon({})'.format(IA_ADDON_ID), True)
xbmc.executeJSONRPC(
'{{"jsonrpc":"2.0","id":1,"method":"Addons.SetAddonEnabled","params":{{"addonid":"{}","enabled":true}}}}'
.format(IA_ADDON_ID))
return xbmcaddon.Addon(IA_ADDON_ID)
except:
pass
if required:
raise InputStreamError(_.IA_NOT_FOUND)
return None
def _download(url, dst, dst_path, arch, md5=None):
filename = url.split('/')[-1]
tmp = ADDON_PROFILE + "tmp" + os.sep + "widevine"
downloaded = 0
if os.path.exists(dst_path):
if md5 and md5sum(dst_path) == md5:
#log.debug('MD5 of local file {} same. Skipping download'.format(filename))
return True
elif not gui.yes_no(_.NEW_IA_VERSION):
return False
else:
if os.path.exists(dst_path):
os.remove(dst_path)
with gui.progress(_(_.IA_DOWNLOADING_FILE, url=filename),
heading=_.IA_WIDEVINE_DRM) as progress:
resp = Session().get(url, stream=True)
if resp.status_code != 200:
raise InputStreamError(
_(_.ERROR_DOWNLOADING_FILE, filename=filename))
total_length = float(resp.headers.get('content-length', 1))
with open(tmp, 'wb') as f:
for chunk in resp.iter_content(chunk_size=SESSION_CHUNKSIZE):
f.write(chunk)
downloaded += len(chunk)
downloadperc = int(downloaded) * 100
downloadperc2 = int(downloadperc) // int(total_length)
percent = int(downloadperc2)
if progress.iscanceled():
progress.close()
resp.close()
progress.update(percent)
if os.path.isfile(tmp):
if 'arm' in arch:
with open(tmp, "rb") as encoded_file:
import base64
decoded_string = base64.b64decode(encoded_file.read())
with open(dst_path, "wb") as decoded_file:
decoded_file.write(decoded_string)
else:
from zipfile import ZipFile
with ZipFile(tmp, 'r') as zipObj:
zipObj.extractall(ADDON_PROFILE + "tmp" + os.sep)
if os.path.isfile(ADDON_PROFILE + "tmp" + os.sep + dst):
shutil.copy(ADDON_PROFILE + "tmp" + os.sep + dst, dst_path)
for file in glob.glob(ADDON_PROFILE + "tmp" + os.sep + "*"):
os.remove(file)
if progress.iscanceled():
return False
checksum = md5sum(dst_path)
if checksum != md5:
if os.path.exists(dst_path):
os.remove(dst_path)
raise InputStreamError(
_(_.MD5_MISMATCH,
filename=filename,
local_md5=checksum,
remote_md5=md5))
return True
def install_widevine(reinstall=False):
ia_addon = get_ia_addon(required=True)
system, arch = _get_system_arch()
kodi_version = get_kodi_version()
if kodi_version < 18:
raise InputStreamError(_(_.IA_KODI18_REQUIRED, system=system))
elif system == 'Android':
return True
elif system == 'UWP':
raise InputStreamError(_.IA_UWP_ERROR)
elif 'aarch64' in arch or 'arm64' in arch:
raise InputStreamError(_.IA_AARCH64_ERROR)
last_check = int(ia_addon.getSetting('_last_check') or 0)
ver_slug = system + arch + str(kodi_version) + ia_addon.getAddonInfo(
'version')
if ver_slug != ia_addon.getSetting(IA_VERSION_KEY):
reinstall = True
if not reinstall and time.time() - last_check < 86400:
return True
ia_addon.setSetting(IA_VERSION_KEY, '')
ia_addon.setSetting('_last_check', str(int(time.time())))
r = api_get_widevine()
widevine = r['widevine']
wv_platform = widevine['platforms'].get(system + arch, None)
if not wv_platform:
raise InputStreamError(
_(_.IA_NOT_SUPPORTED,
system=system,
arch=arch,
kodi_version=kodi_version))
decryptpath = xbmc.translatePath(ia_addon.getSetting('DECRYPTERPATH'))
if sys.version_info < (3, 0):
decryptpath = decryptpath.decode("utf-8")
if 'arm' in arch:
url = wv_platform['zip']
else:
url = widevine['base_url'] + wv_platform['zip']
wv_path = os.path.join(decryptpath, wv_platform['dst'])
if not os.path.isdir(decryptpath):
os.makedirs(decryptpath)
if not os.path.isdir(ADDON_PROFILE + "tmp"):
os.makedirs(ADDON_PROFILE + "tmp")
if not _download(url, wv_platform['dst'], wv_path, arch,
wv_platform['md5']):
return False
ia_addon.setSetting(IA_VERSION_KEY, ver_slug)
if reinstall:
gui.ok(_.IA_WV_INSTALL_OK)
return True