85 lines
2.7 KiB
Python
85 lines
2.7 KiB
Python
'''
|
||
Author: SpenserCai
|
||
Date: 2023-08-23 23:12:27
|
||
version:
|
||
LastEditors: SpenserCai
|
||
LastEditTime: 2023-08-24 11:36:53
|
||
Description: file content
|
||
'''
|
||
import os
|
||
import modules.scripts as scripts
|
||
import requests
|
||
import sys
|
||
import tarfile
|
||
import shutil
|
||
|
||
bin_path = None
|
||
api_url = "https://api.github.com/repos/SpenserCai/sd-webui-discord/releases/latest"
|
||
|
||
def init_base():
|
||
global bin_path
|
||
bin_path = os.path.join(get_my_dir(), "bin")
|
||
|
||
def get_bin_path():
|
||
return os.path.join(get_my_dir(), "bin")
|
||
|
||
def get_bin_process_path():
|
||
if sys.platform == "win32":
|
||
return os.path.join(bin_path, "sd-webui-discord.exe")
|
||
elif sys.platform == "linux":
|
||
return os.path.join(bin_path, "sd-webui-discord")
|
||
|
||
def get_my_dir():
|
||
if os.path.isdir("extensions/sd-webui-discord-ex"):
|
||
return "extensions/sd-webui-discord-ex"
|
||
return scripts.basedir()
|
||
|
||
def check_bin():
|
||
print(bin_path)
|
||
if os.path.isfile(get_bin_process_path()):
|
||
return True
|
||
return False
|
||
|
||
def need_update():
|
||
response = requests.get(api_url)
|
||
data = response.json()
|
||
tag_name = data["tag_name"]
|
||
with open(os.path.join(bin_path, ".version"), "r") as file:
|
||
version = file.read()
|
||
# version 去掉头尾空格换行符
|
||
version = version.strip()
|
||
if tag_name == version:
|
||
return False
|
||
return True
|
||
|
||
def download_bin():
|
||
response = requests.get(api_url)
|
||
data = response.json()
|
||
tag_name = data["tag_name"]
|
||
print(f"Downloading sd-webui-discord {tag_name}...")
|
||
download_url = f"https://github.com/SpenserCai/sd-webui-discord/releases/download/{tag_name}/sd-webui-discord-release-{tag_name}.tar.gz"
|
||
response = requests.get(download_url)
|
||
release_path = os.path.join(bin_path, f"sd-webui-discord-{tag_name}.tar.gz")
|
||
with open(release_path, "wb") as file:
|
||
file.write(response.content)
|
||
print("Extracting sd-webui-discord...")
|
||
with tarfile.open(release_path, "r:gz") as tar:
|
||
members = [m for m in tar.getmembers() if m.name.startswith("release")]
|
||
tar.extractall(path=bin_path, members=members)
|
||
# 判断release目录是否存在，如果存在把里面的文件移动到bin目录下，然后删除release目录
|
||
if os.path.isdir(os.path.join(bin_path, "release")):
|
||
for file in os.listdir(os.path.join(bin_path, "release")):
|
||
# 如果config.json存在，不移动
|
||
if file == "config.json":
|
||
continue
|
||
shutil.move(os.path.join(bin_path, "release", file), bin_path)
|
||
shutil.rmtree(os.path.join(bin_path, "release"))
|
||
|
||
os.remove(release_path)
|
||
|
||
with open(os.path.join(bin_path, ".version"), "w") as file:
|
||
file.write(tag_name)
|
||
print("Done!")
|
||
|
||
|
||
|