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

add generate-units and generate-ingredients subcommands

Browse source
This commit is contained in:
Emi Vasilek 2023-11-20 10:27:21 +01:00
parent e1f25dbb14
commit 3ae3dfe922
2 changed files with 99 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

25
comfyrecipes/cli.py
View file

@ -4,7 +4,7 @@ import socketserver
import sys
import os
import comfyrecipes.builder as builder
import comfyrecipes.builder
def main() -> None:
@ -19,11 +19,20 @@ def main() -> None:
parser_serve.add_argument("--port", type=int, default=8000)
parser_serve.add_argument("--address", type=str, default="127.0.0.1")
parser_generate_units = subparsers.add_parser("generate-units")
parser_generate_units.add_argument("directory", type=str)
parser_generate_units.add_argument("--force", dest="force", action="store_true")
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")
args = parser.parse_args()
ret = 0
builder = comfyrecipes.builder.Builder()
if args.subcommand == "build":
ret = builder.Builder().build(args.directory)
ret = builder.build(args.directory)
elif args.subcommand == "serve":
os.chdir(f"{args.directory}/out/html")
httpd = socketserver.TCPServer(
@ -34,6 +43,18 @@ def main() -> None:
httpd.serve_forever()
except KeyboardInterrupt:
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)
ret = 1
else:
builder.generate_units(args.directory)
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)
ret = 1
else:
builder.generate_ingredients(args.directory)
else:
# unhandled, but valid subcommand
assert False