84 lines
2.6 KiB
Python
84 lines
2.6 KiB
Python
|
import os
|
||
|
|
import json
|
||
|
|
import urllib.parse
|
||
|
|
import requests
|
||
|
|
import hashlib
|
||
|
|
from os import path as op
|
||
|
|
from os import getcwd as pwd
|
||
|
|
from rich.console import Console
|
||
|
|
|
||
|
|
from .utils import get_progress
|
||
|
|
|
||
|
|
console = Console(log_path=False)
|
||
|
|
|
||
|
|
|
||
|
|
class File:
|
||
|
|
def __init__(self, url: str):
|
||
|
|
self.name = url.split("/")[-1]
|
||
|
|
self.url = url
|
||
|
|
self.filepath = op.join(pwd(), "files/", self.name)
|
||
|
|
if not op.exists(op.join(pwd(), "files/")):
|
||
|
|
os.mkdir(op.join(pwd(), "files/"))
|
||
|
|
elif not op.isdir(op.join(pwd(), "files/")):
|
||
|
|
os.remove(op.join(pwd(), "files/"))
|
||
|
|
os.mkdir(op.join(pwd(), "files/"))
|
||
|
|
|
||
|
|
def md5sum(self) -> str | None:
|
||
|
|
try:
|
||
|
|
return json.loads(requests.get(f"https://timoxa0.su/?info={urllib.parse.urlparse(self.url).path}")
|
||
|
|
.content.decode())["hashes"]["md5"]
|
||
|
|
except requests.ConnectionError:
|
||
|
|
pass
|
||
|
|
|
||
|
|
def get(self):
|
||
|
|
while True:
|
||
|
|
md5sum = self.md5sum()
|
||
|
|
if not md5sum:
|
||
|
|
console.log(f"Unable to verify {self.name} checksum")
|
||
|
|
|
||
|
|
if op.exists(self.filepath):
|
||
|
|
with open(self.filepath, "rb") as file:
|
||
|
|
data = file.read()
|
||
|
|
if hashlib.md5(data).hexdigest() == md5sum or not md5sum:
|
||
|
|
return data
|
||
|
|
|
||
|
|
download_image = b""
|
||
|
|
r = requests.get(self.url, stream=True)
|
||
|
|
if r.status_code != 200:
|
||
|
|
console.log(f"{self.name} not found on server. Please contact developer")
|
||
|
|
total_size = int(r.headers.get("content-length", 0))
|
||
|
|
block_size = 204800
|
||
|
|
with get_progress() as pbar:
|
||
|
|
task = pbar.add_task(f"[green]Downloading {self.name}", total=total_size)
|
||
|
|
for data in r.iter_content(block_size):
|
||
|
|
download_image += data
|
||
|
|
pbar.update(task, advance=len(data))
|
||
|
|
with open(self.filepath, "wb") as file:
|
||
|
|
file.write(download_image)
|
||
|
|
|
||
|
|
if hashlib.md5(download_image).hexdigest() == md5sum or not md5sum:
|
||
|
|
return download_image
|
||
|
|
else:
|
||
|
|
console.log("Downloaded file corrupted!")
|
||
|
|
|
||
|
|
|
||
|
|
OrangeFox = File(
|
||
|
|
url="https://timoxa0.su/share/nabu/deployer/orangefox.img"
|
||
|
|
)
|
||
|
|
|
||
|
|
UEFI_Payload = File(
|
||
|
|
url="https://timoxa0.su/share/nabu/deployer/uefi/nabu_UEFI.fd"
|
||
|
|
)
|
||
|
|
|
||
|
|
BootShim = File(
|
||
|
|
url="https://timoxa0.su/share/nabu/deployer/uefi/BootShim.Dualboot.bin"
|
||
|
|
)
|
||
|
|
|
||
|
|
GPT_Both0 = File(
|
||
|
|
url="https://timoxa0.su/share/nabu/deployer/gpt_both0.bin"
|
||
|
|
)
|
||
|
|
|
||
|
|
UserData_Empty = File(
|
||
|
|
url="https://timoxa0.su/share/nabu/deployer/userdata.img"
|
||
|
|
)
|