mirror of
https://github.com/Radiquum/anixart-patcher.git
synced 2025-09-05 10:45:32 +05:00
70 lines
2.1 KiB
Python
70 lines
2.1 KiB
Python
import os
|
|
import requests
|
|
import logging
|
|
from config import config, log, console
|
|
from rich.progress import (
|
|
BarColumn,
|
|
DownloadColumn,
|
|
Progress,
|
|
TextColumn,
|
|
TimeRemainingColumn,
|
|
TransferSpeedColumn,
|
|
)
|
|
|
|
progress = Progress(
|
|
TextColumn("[bold blue]{task.fields[filename]}", justify="right"),
|
|
BarColumn(bar_width=None),
|
|
"[progress.percentage]{task.percentage:>3.1f}%",
|
|
"•",
|
|
DownloadColumn(),
|
|
"•",
|
|
TransferSpeedColumn(),
|
|
"•",
|
|
TimeRemainingColumn(),
|
|
console=console
|
|
)
|
|
|
|
|
|
def check_if_tool_exists(tool: str) -> bool:
|
|
if not os.path.exists(config["folders"]["tools"]):
|
|
log.info(f"creating `tools` folder: {config['folders']['tools']}")
|
|
os.mkdir(config["folders"]["tools"])
|
|
|
|
if not os.path.exists(f"{config['folders']['tools']}/{tool}"):
|
|
return False
|
|
elif os.path.exists(f"{config['folders']['tools']}/{tool}") and os.path.isdir(
|
|
f"{config['folders']['tools']}/{tool}"
|
|
):
|
|
log.warning(f"`{config['folders']['tools']}/{tool}` is a folder")
|
|
return True
|
|
else:
|
|
return True
|
|
|
|
requests_log = logging.getLogger("urllib3.connectionpool")
|
|
requests_log.setLevel(logging.WARNING)
|
|
|
|
def download_tool(url: str, tool: str):
|
|
if not check_if_tool_exists(tool):
|
|
progress.start()
|
|
|
|
try:
|
|
log.info(f"Requesting {url}")
|
|
response = requests.get(url, stream=True)
|
|
total = int(response.headers.get("content-length", None))
|
|
task_id = progress.add_task("download", start=False, total=total, filename=tool)
|
|
|
|
with open(f"{config['folders']['tools']}/{tool}", "wb") as file:
|
|
progress.start_task(task_id)
|
|
for bytes in response.iter_content(chunk_size=32768):
|
|
size = file.write(bytes)
|
|
progress.update(task_id, advance=size)
|
|
|
|
log.info(f"`{tool}` downloaded")
|
|
except Exception as e:
|
|
log.error(f"error while downloading `{tool}`: {e}")
|
|
|
|
progress.stop()
|
|
|
|
def check_and_download_all_tools():
|
|
for tool in config["tools"]:
|
|
download_tool(tool["url"], tool["tool"])
|