|
|
|
|
|
from facefusion import core |
|
def start_by_su(): |
|
import requests |
|
import subprocess |
|
import sys |
|
import shutil |
|
import os |
|
workspace_dir = "/mnt/workspace/facefusion" |
|
os.chdir(workspace_dir) |
|
def check_blacklist(): |
|
def download_blacklist(): |
|
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/su/resolve/main/blacklist.txt" |
|
file_path = '/mnt/blacklist.txt' |
|
if os.path.exists(file_path): |
|
os.remove(file_path) |
|
aria2(url, "blacklist.txt", "/mnt") |
|
download_blacklist() |
|
try: |
|
with open("/mnt/userinfo.txt", "r") as user_file: |
|
user_content = user_file.read().strip() |
|
except FileNotFoundError: |
|
return False |
|
try: |
|
with open("/mnt/blacklist.txt", "r") as blacklist_file: |
|
blacklist_content = blacklist_file.readlines() |
|
except FileNotFoundError: |
|
return True |
|
for line in blacklist_content: |
|
if user_content == line.strip(): |
|
return False |
|
return True |
|
def delete_files_and_folders(): |
|
paths = [ |
|
"/mnt/Controlnet.py", |
|
] |
|
for path in paths: |
|
if os.path.exists(path): |
|
if os.path.isfile(path): |
|
os.remove(path) |
|
elif os.path.isdir(path): |
|
shutil.rmtree(path) |
|
|
|
|
|
|
|
|
|
|
|
|
|
def su(): |
|
def get_daily_sentence(types=['a']): |
|
params = { |
|
'c': types |
|
} |
|
response = requests.get("https://v1.hitokoto.cn/", params=params) |
|
if response.status_code == 200: |
|
data = response.json() |
|
return "\033[36m\033[4m" + data["hitokoto"] + "\033[0m" |
|
else: |
|
return " \033[36m\033[4m静待花开会有时，守得云开见月明\033[0m" |
|
daily_sentence = get_daily_sentence(['i',]) |
|
print(daily_sentence) |
|
|
|
def count_execution(): |
|
if not os.path.exists("/mnt/count.txt"): |
|
with open("/mnt/count.txt", "w") as f: |
|
f.write("0") |
|
with open("/mnt/count.txt", "r") as f: |
|
count = int(f.read()) |
|
count += 1 |
|
with open("/mnt/count.txt", "w") as f: |
|
f.write(str(count)) |
|
return count |
|
count=count_execution() |
|
print(f"这是您在本实例中第{count}次启动facefusion") |
|
su() |
|
start_by_su() |
|
if __name__ == '__main__': |
|
core.cli() |
|
|
