Emi/comfy-recipes
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity

black format

Browse source
This commit is contained in:
Emi Vasilek 2023-11-21 03:13:36 +01:00
parent 247fd37560
commit 238319b9e1
5 changed files with 23 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

14
comfyrecipes/cli.py
View file

@ -27,7 +27,9 @@ def main() -> None:
parser_generate_ingredients = subparsers.add_parser("generate-ingredients")
parser_generate_ingredients.add_argument("directory", type=str)
parser_generate_ingredients.add_argument("--force", dest="force", action="store_true")
parser_generate_ingredients.add_argument(
"--force", dest="force", action="store_true"
)
args = parser.parse_args()
@ -48,13 +50,19 @@ def main() -> None:
pass
elif args.subcommand == "generate-units":
if not args.force and os.path.isfile(args.directory + "/units.yaml"):
print("units.yaml already exists, pass --force if you want to overwrite it", file=sys.stderr)
print(
"units.yaml already exists, pass --force if you want to overwrite it",
file=sys.stderr,
)
ret = 1
else:
builder.generate_units()
elif args.subcommand == "generate-ingredients":
if not args.force and os.path.isfile(args.directory + "/ingredients.yaml"):
print("ingredients.yaml already exists, pass --force if you want to overwrite it", file=sys.stderr)
print(
"ingredients.yaml already exists, pass --force if you want to overwrite it",
file=sys.stderr,
)
ret = 1
else:
builder.generate_ingredients()