mirror of
https://github.com/RGBCube/GitHubWrapper
synced 2025-05-20 16:05:08 +00:00
64 lines
1.6 KiB
Python
64 lines
1.6 KiB
Python
import json
|
|
import os
|
|
import time
|
|
from argparse import ArgumentParser
|
|
|
|
from .parser import generate
|
|
|
|
parser = ArgumentParser(description="Generate TypedDicts from a json schema.")
|
|
parser.add_argument("-f", "--from", required=True, help="The json schema to generate from.")
|
|
parser.add_argument("-t", "--to", help="The file to write the TypedDicts to.")
|
|
parser.add_argument(
|
|
"-on",
|
|
"--object-name",
|
|
default="GeneratedObjectResult",
|
|
help="The name of the object to generate.",
|
|
)
|
|
parser.add_argument(
|
|
"-nc", "--no-comments", action="store_true", help="If given, comments wont be added."
|
|
)
|
|
parser.add_argument(
|
|
"-p",
|
|
"--print",
|
|
action="store_true",
|
|
help="If given, the result will be printed.",
|
|
)
|
|
|
|
args = parser.parse_args()
|
|
|
|
with open(args.__getattribute__("from")) as f:
|
|
generated = generate(json.load(f), no_comments=args.no_comments)
|
|
|
|
start = time.perf_counter()
|
|
|
|
text = f"""from __future__ import annotations
|
|
|
|
from typing import Any, List, Optional, TypedDict, Union, TYPE_CHECKING
|
|
|
|
if TYPE_CHECKING:
|
|
from typing_extensions import NotRequired
|
|
|
|
{generated[0]}
|
|
|
|
{args.object_name} = {generated[1]}
|
|
"""
|
|
|
|
if args.print:
|
|
print(text)
|
|
|
|
end = time.perf_counter() - start
|
|
|
|
else:
|
|
if not (to := args.to):
|
|
raise ValueError("-t/--to is required when writing to a file.")
|
|
|
|
with open(to, "w") as f:
|
|
f.write(text)
|
|
|
|
end = time.perf_counter() - start
|
|
|
|
os.system(
|
|
f"unimport {to} --gitignore -r --ignore-init; isort {to}; black {to}; flynt {to} -tc",
|
|
)
|
|
|
|
print(f"\n\nSuccess! Finished in {end*1000:.3} milliseconds")
|