|
import os |
|
import shutil |
|
import time |
|
import argparse |
|
import requests |
|
|
|
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" |
|
|
|
def ClearLike(): |
|
daily_sentence = get_daily_sentence(['a','b','c','d','e','f','g','h','j','k','l',]) |
|
print(daily_sentence) |
|
path = "/mnt/workspace/Photo/Like" |
|
if os.path.exists(path): |
|
shutil.rmtree(path) |
|
print(f"{path} 已删除") |
|
else: |
|
print(f"{path} 不存在") |
|
|
|
def PackLike(): |
|
daily_sentence = get_daily_sentence(['a','b','c','d','e','f','g','h','j','k','l',]) |
|
print(daily_sentence) |
|
path = "/mnt/workspace/Photo/Like" |
|
zip_path = "/mnt/workspace/Like.zip" |
|
if os.path.exists(path): |
|
if os.path.exists(zip_path): |
|
os.system(f"7z a {zip_path} {path} -y") |
|
print("检测到旧的Like包存在,已经覆盖,已保存在根目录") |
|
else: |
|
os.system(f"7z a {zip_path} {path}") |
|
print("打包完成,已保存在根目录,文件名为Like.zip") |
|
else: |
|
print(f"{path} 文件夹不存在") |
|
|
|
def ClearUpscale(): |
|
daily_sentence = get_daily_sentence(['a','b','c','d','e','f','g','h','j','k','l',]) |
|
print(daily_sentence) |
|
path = "/mnt/workspace/Photo/Upscale" |
|
if os.path.exists(path): |
|
shutil.rmtree(path) |
|
print(f"{path} 已删除") |
|
else: |
|
print(f"{path} 不存在") |
|
|
|
def PackUpscale(): |
|
daily_sentence = get_daily_sentence(['a','b','c','d','e','f','g','h','j','k','l',]) |
|
print(daily_sentence) |
|
path = "/mnt/workspace/Photo/Upscale" |
|
zip_path = "/mnt/workspace/Upscale.zip" |
|
if os.path.exists(path): |
|
if os.path.exists(zip_path): |
|
os.system(f"7z a {zip_path} {path} -y") |
|
print("检测到旧的Upscale包存在,已经覆盖,已保存在根目录") |
|
else: |
|
os.system(f"7z a {zip_path} {path}") |
|
print("打包完成,已保存在根目录,文件名为Upscale.zip") |
|
else: |
|
print(f"{path} 文件夹不存在") |
|
|
|
def ClearAll(): |
|
daily_sentence = get_daily_sentence(['a','b','c','d','e','f','g','h','j','k','l']) |
|
print(daily_sentence) |
|
path = "/mnt/workspace/Photo/" |
|
if os.path.exists(path): |
|
for item in os.listdir(path): |
|
item_path = os.path.join(path, item) |
|
if os.path.isdir(item_path): |
|
shutil.rmtree(item_path) |
|
print(f"{path} 已清空") |
|
else: |
|
print(f"{path} 文件夹不存在") |
|
|
|
|
|
def PackAll(): |
|
daily_sentence = get_daily_sentence(['a','b','c','d','e','f','g','h','j','k','l',]) |
|
print(daily_sentence) |
|
path = "/mnt/workspace/Photo" |
|
zip_path = "/mnt/workspace/Photo.zip" |
|
if os.path.exists(path): |
|
if os.path.exists(zip_path): |
|
os.system(f"7z a {zip_path} {path} -y") |
|
print("检测到旧的Photo包存在,已经覆盖,已保存在根目录") |
|
else: |
|
os.system(f"7z a {zip_path} {path}") |
|
print("打包完成,已保存在根目录,文件名为Photo.zip") |
|
else: |
|
print(f"{path} 文件夹不存在") |
|
|
|
def ClearLikeUpscale(): |
|
daily_sentence = get_daily_sentence(['a','b','c','d','e','f','g','h','j','k','l',]) |
|
print(daily_sentence) |
|
path = "/mnt/workspace/Photo/Like_Upscale" |
|
if os.path.exists(path): |
|
shutil.rmtree(path) |
|
print(f"{path} 已删除") |
|
else: |
|
print(f"{path} 不存在") |
|
|
|
def PackLikeUpscale(): |
|
daily_sentence = get_daily_sentence(['a','b','c','d','e','f','g','h','j','k','l',]) |
|
print(daily_sentence) |
|
path = "/mnt/workspace/Photo/Like_Upscale" |
|
zip_path = "/mnt/workspace/Like_Upscale.zip" |
|
if os.path.exists(path): |
|
if os.path.exists(zip_path): |
|
os.system(f"7z a {zip_path} {path} -y") |
|
print("检测到旧的Like_Upscale包存在,已经覆盖,已保存在根目录") |
|
else: |
|
os.system(f"7z a {zip_path} {path}") |
|
print("打包完成,已保存在根目录,文件名为Like_Upscale.zip") |
|
else: |
|
print(f"{path} 文件夹不存在\n") |
|
print("请在SD界面-后期处理-批量处理文件夹中设置\n") |
|
print("输入目录") |
|
print("/mnt/workspace/Photo/Like\n") |
|
print("输出目录") |
|
print("/mnt/workspace/Photo/Like_Upscale\n") |
|
|
|
def Cleartodaytxt2img(): |
|
daily_sentence = get_daily_sentence(['a','b','c','d','e','f','g','h','j','k','l',]) |
|
print(daily_sentence) |
|
current_time = time.localtime() |
|
formatted_time = time.strftime("%Y-%m-%d", current_time) |
|
todaypath = f"/mnt/workspace/Photo/txt2img/{formatted_time}" |
|
if os.path.exists(todaypath): |
|
shutil.rmtree(todaypath) |
|
print(f"{todaypath} 已删除") |
|
else: |
|
print(f"{todaypath} 不存在") |
|
|
|
def Packtodaytxt2img(): |
|
daily_sentence = get_daily_sentence(['a','b','c','d','e','f','g','h','j','k','l',]) |
|
print(daily_sentence) |
|
current_time = time.localtime() |
|
formatted_time = time.strftime("%Y-%m-%d", current_time) |
|
todaypath = f"/mnt/workspace/Photo/txt2img/{formatted_time}" |
|
zip_path = f"/mnt/workspace/【文生图】{formatted_time}.zip" |
|
if os.path.exists(todaypath): |
|
if os.path.exists(zip_path): |
|
os.system(f"7z a {zip_path} {todaypath} -y") |
|
print(f"检测到旧的【文生图】{formatted_time}包存在,已经覆盖,已保存在根目录") |
|
else: |
|
os.system(f"7z a {zip_path} {todaypath}") |
|
print(f"打包完成,已保存在根目录,文件名为【文生图】{formatted_time}.zip") |
|
else: |
|
print(f"{todaypath} 文件夹不存在") |
|
|
|
def cleartodayimg2img(): |
|
daily_sentence = get_daily_sentence(['a','b','c','d','e','f','g','h','j','k','l',]) |
|
print(daily_sentence) |
|
current_time = time.localtime() |
|
formatted_time = time.strftime("%Y-%m-%d", current_time) |
|
todaypath = f"/mnt/workspace/Photo/txt2img/{formatted_time}" |
|
if os.path.exists(todaypath): |
|
shutil.rmtree(todaypath) |
|
print(f"{todaypath} 已删除") |
|
else: |
|
print(f"{todaypath} 不存在") |
|
|
|
def Packtodayimg2img(): |
|
daily_sentence = get_daily_sentence(['a','b','c','d','e','f','g','h','j','k','l',]) |
|
print(daily_sentence) |
|
current_time = time.localtime() |
|
formatted_time = time.strftime("%Y-%m-%d", current_time) |
|
todaypath = f"/mnt/workspace/Photo/img2img/{formatted_time}" |
|
zip_path = f"/mnt/workspace/【图生图】{formatted_time}.zip" |
|
if os.path.exists(todaypath): |
|
if os.path.exists(zip_path): |
|
os.system(f"7z a {zip_path} {todaypath} -y") |
|
print(f"检测到旧的【图生图】{formatted_time}包存在,已经覆盖,已保存在根目录") |
|
else: |
|
os.system(f"7z a {zip_path} {todaypath}") |
|
print(f"打包完成,已保存在根目录,文件名为【图生图】{formatted_time}.zip") |
|
else: |
|
print(f"{todaypath} 文件夹不存在") |
|
|
|
def Clearoriginalimg2img(): |
|
daily_sentence = get_daily_sentence(['a','b','c','d','e','f','g','h','j','k','l',]) |
|
print(daily_sentence) |
|
path = f"/mnt/workspace/Photo/img2img/original" |
|
if os.path.exists(path): |
|
shutil.rmtree(path) |
|
print(f"{path} 已删除") |
|
else: |
|
print(f"{path} 不存在") |
|
|
|
if __name__ == "__main__": |
|
parser = argparse.ArgumentParser(description="Process some integers.") |
|
parser.add_argument("-ClearLike", action="store_true", help="Clear Like folder") |
|
parser.add_argument("-PackLike", action="store_true", help="Pack Like folder") |
|
parser.add_argument("-ClearUpscale", action="store_true", help="Clear Upscale folder") |
|
parser.add_argument("-PackUpscale", action="store_true", help="Pack Upscale folder") |
|
parser.add_argument("-ClearAll", action="store_true", help="Clear All folder") |
|
parser.add_argument("-PackAll", action="store_true", help="Pack All folder") |
|
parser.add_argument("-ClearLikeUpscale", action="store_true", help="Clear Like_Upscale folder") |
|
parser.add_argument("-PackLikeUpscale", action="store_true", help="Pack Like_Upscale folder") |
|
parser.add_argument("-Cleartodaytxt2img", action="store_true", help="Clear todaytxt2img folder") |
|
parser.add_argument("-Packtodaytxt2img", action="store_true", help="Pack todaytxt2img folder") |
|
parser.add_argument("-Packtodayimg2img", action="store_true", help="Pack todayimg2img folder") |
|
parser.add_argument("-Cleartodayimg2img", action="store_true", help="Clear todayimg2img folder") |
|
parser.add_argument("-Clearoriginalimg2img", action="store_true", help="Pack Clearoriginalimg2img folder") |
|
|
|
args = parser.parse_args() |
|
|
|
if args.ClearLike: |
|
ClearLike() |
|
elif args.PackLike: |
|
PackLike() |
|
elif args.ClearUpscale: |
|
ClearUpscale() |
|
elif args.PackUpscale: |
|
PackUpscale() |
|
elif args.ClearAll: |
|
ClearAll() |
|
elif args.PackAll: |
|
PackAll() |
|
elif args.ClearLikeUpscale: |
|
ClearLikeUpscale() |
|
elif args.PackLikeUpscale: |
|
PackLikeUpscale() |
|
elif args.Cleartodaytxt2img: |
|
Cleartodaytxt2img() |
|
elif args.Packtodaytxt2img: |
|
Packtodaytxt2img() |
|
elif args.Packtodayimg2img: |
|
Packtodayimg2img() |
|
elif args.Cleartodayimg2img: |
|
Cleartodayimg2img() |
|
elif args.Clearoriginalimg2img: |
|
Clearoriginalimg2img() |
|
|