modified: patch.py

This commit is contained in:
zyb 2024-07-26 10:53:08 +08:00
parent e16fe8fdce
commit f1fe5c9a7e

View File

@ -244,13 +244,16 @@ def patch_squashfs(path,key_dict):
data = open(file,'rb').read() data = open(file,'rb').read()
url_dict = { url_dict = {
os.environ['MIKRO_LICENCE_URL'].encode():os.environ['CUSTOM_LICENCE_URL'].encode(), os.environ['MIKRO_LICENCE_URL'].encode():os.environ['CUSTOM_LICENCE_URL'].encode(),
os.environ['MIKRO_UPGRADE_URL'].encode():os.environ['CUSTOM_UPGRADE_URL'].encode() os.environ['MIKRO_UPGRADE_URL'].encode():os.environ['CUSTOM_UPGRADE_URL'].encode(),
os.environ['MIKRO_CLOUD_URL'].encode():os.environ['CUSTOM_CLOUD_URL'].encode(),
os.environ['MIKRO_CLOUD_PUBLIC_KEY'].encode():os.environ['CUSTOM_CLOUD_PUBLIC_KEY'].encode(),
} }
for old_url,new_url in url_dict.items(): for old_url,new_url in url_dict.items():
if old_url in data: if old_url in data:
print(f'{file} url patched {old_url.decode()[:7]}...') print(f'{file} url patched {old_url.decode()[:7]}...')
data = data.replace(old_url,new_url) data = data.replace(old_url,new_url)
open(file,'wb').write(data) open(file,'wb').write(data)
if os.path.split(file)[1] == 'licupgr': if os.path.split(file)[1] == 'licupgr':
url_dict = { url_dict = {
os.environ['MIKRO_RENEW_URL'].encode():os.environ['CUSTOM_RENEW_URL'].encode(), os.environ['MIKRO_RENEW_URL'].encode():os.environ['CUSTOM_RENEW_URL'].encode(),