api.py

import requests
from remotezip import RemoteZip

def get_keys(identifier, board, buildid):
	try:
		f = requests.get(f"https://api.m1sta.xyz/wikiproxy/{identifier}/{board}/{buildid}").json()
	except Exception:
		if input(f"[?] Keys not found for this IPSW ({buildid}) for the board {board}. Do you want to enter keys manually? (y/n) ") == "y":
			iBSS_iv = input("  - Enter the iBSS IV: ")
			iBSS_key = input("  - Enter the iBSS Key: ")
			iBEC_iv = input("  - Enter the iBEC IV: ")
			iBEC_key = input("  - Enter the iBEC Key: ")
			return iBSS_iv, iBSS_key, iBEC_iv, iBEC_key
		else:
			exit()
		print("Requesting keys...")
	for dev in f['keys']:
		if dev['image'] == "iBSS":
			iBSS_iv = dev['iv']
			iBSS_key = dev['key']
		if dev['image'] == "iBEC":
			iBEC_iv = dev['iv']
			iBEC_key = dev['key']
	try:
		return iBSS_iv, iBSS_key, iBEC_iv, iBEC_key
	except UnboundLocalError:
		print("[WARNING] Unable to get firmware keys, either the bootchain is not encrypted or the wikiproxy does not have it.")
		input("Continue or not? (Press ENTER to continue, Ctrl-C to quit)")

def partialzip_download(url, file, dest):
	with RemoteZip(url) as zip:
		data = zip.read(file)
	with open(dest, 'wb') as f:
		f.write(data)