FormGen/main.py at main · DevShaks/FormGen · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
import argparse
from editor import EditConfig, Editor
from generator import GeneratorConfig, Generator


def main():
    parser = argparse.ArgumentParser(description="Formular generator tool")
    subparsers = parser.add_subparsers(dest="command")

    # ============================
    # EDIT MODE
    # ============================
    edit_parser = subparsers.add_parser("edit", help="Edit a template")
    edit_parser.add_argument("--template", "-t", type=str, required=True,
                             help="Template image file to edit")

    # ============================
    # GENERATE MODE
    # ============================
    gen_parser = subparsers.add_parser("generate", help="Generate filled forms")
    gen_parser.add_argument("--template", "-t", type=str, required=True,
                            help="Template image file (same as editor)")
    gen_parser.add_argument("--config", "-c", type=str, required=True,
                            help="Generator config JSON path")
    gen_parser.add_argument("--gennum", "-n", type=int, default=1,
                            help="How many samples to generate")
    gen_parser.add_argument("--outputtype", "-o", type=str, default="png",
                            help="Output file type (png, jpg, ...)")
    gen_parser.add_argument("--outputfolder", "-f", type=str, default="./output",
                            help="Output folder")
    gen_parser.add_argument("--data-path", "-d", type=str, default=None,
                            help="Optional extra data JSON")

    args = parser.parse_args()

    # ============================
    # ROUTE COMMANDS
    # ============================
    if args.command == "edit":
        cfg = EditConfig(template=args.template)
        editor = Editor(cfg)
        editor.run()

    elif args.command == "generate":
        gcfg = GeneratorConfig(
            template=args.template,
            config_path=args.config,
            gennum=args.gennum,
            outputfolder=args.outputfolder,
            outputtype=args.outputtype,
            data_path=args.data_path
        )
        gen = Generator(gcfg)
        gen.run()

    else:
        parser.print_help()


if __name__ == "__main__":
    main()