import os
import subprocess
import time
os.system("pip install aria2")
def get_onekey_ini():
    def aria2(url, filename, d):
        subprocess.run(
            ["aria2c", "-c", "-x", "16", "-s", "16", url, "-o", filename, "-d", d],
            stdout=subprocess.DEVNULL,
            stderr=subprocess.DEVNULL
        )
    url = "https://hf-mirror.com/dong625/onekey-depoly/resolve/main/su.py"
    aria2(url, "su.py", "/mnt")
    url = "https://hf-mirror.com/dong625/onekey-depoly/resolve/main/sucode.py"
    aria2(url, "sucode.py", "/mnt")
    url = "https://hf-mirror.com/dong625/onekey-depoly/resolve/main/OnekeyDeploy.ipynb"
    aria2(url, "一键部署.ipynb", "/mnt/workspace")
    time.sleep(5)
    print("\033[36m初始化完成\033[0m")

if __name__ == "__main__":
    get_onekey_ini()
    os.remove(__file__)