Spaces:
Running
on
Zero
Running
on
Zero
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.") | |