INNER CODE UNIT · Python

main

zhulinyv/Semi-Auto-NovelAI-to-Pixiv · src/batch_mosaic.py:102

def main(file_path, input_img, open_button, mode):
    if open_button:
        file_list = file_namel2pathl(file_path2list(file_path), file_path)
    else:
        input_img.save("./output/_temp.png")
        file_list = ["./output/_temp.png"]
    for file in file_list:
        logger.info(f"正在处理 {file_path2name(file)}...")
        if os.path.exists("./output/temp.png"):
            os.remove("./output/temp.png")
        shutil.copyfile(file, "./output/temp.png")
        if mode == "pixel":
            mosaic_pixel("./output/temp.png")
        elif mode == "blurry":
            mosaic_blurry("./output/temp.png")
        elif mode == "lines":
            mosaic_lines("./output/temp.png")
        logger.success("处理完成!")

View source record →