flatcherlee's picture
Upload 2334 files
3d5837a verified
raw
history blame
1.93 kB
import os
from json import load, dumps
from datetime import datetime
_mapping = {
"Float": "DF_Float",
"Integer": "DF_Integer",
"Text": "DF_Text",
"Text box": "DF_Text_Box",
"String Concatenate": "DF_String_Concatenate",
"String Replace": "DF_String_Replace",
"To text (Debug)": "DF_To_text_(Debug)",
"Random": "DF_Random",
"Int to float": "DF_Int_to_Float",
"Ceil": "DF_Ceil",
"Floor": "DF_Floor",
"Absolute value": "DF_Absolute_value",
"Get latent size": "DF_Get_latent_size",
"Get image size": "DF_Get_image_size",
"Sum": "DF_Sum",
"Subtract": "DF_Subtract",
"Multiply": "DF_Multiply",
"Divide": "DF_Divide",
"Power": "DF_Power",
"Square root": "DF_Square_root",
"Sinus": "DF_Sinus",
"Cosines": "DF_Cosines",
"Tangent": "DF_Tangent",
"Logic node": "DF_Logic_node",
"Latent Scale by ratio": "DF_Latent_Scale_by_ratio",
"Latent Scale to side": "DF_Latent_Scale_to_side",
"Image scale by ratio": "DF_Image_scale_by_ratio",
"Image scale to side": "DF_Image_scale_to_side",
"Conditioning area scale by ratio": "DF_Conditioning_area_scale_by_ratio",
}
if __name__ == "__main__":
json_file = input("workflow.json where replace mappings: ")
with open(json_file, "r") as j_file:
workflow = load(j_file)
for node in workflow["nodes"]:
print(node["type"], end=" -> ")
if node["type"] in _mapping.keys():
node["type"] = _mapping[node["type"]]
print(node["type"])
replaced_folder = "replaced"
if replaced_folder not in os.listdir(os.curdir):
os.mkdir(replaced_folder)
path = f"{replaced_folder}/{datetime.now().strftime('%d-%m-%Y %H.%M.%S')}.json"
with open(path, "w") as j_file:
j_file.write(dumps(workflow))
print("Done.")
input("Press any key to close console.")