RGBCube/GitHubWrapper
1
Fork 0
mirror of https://github.com/RGBCube/GitHubWrapper synced 2025-05-18 15:05:08 +00:00
Code Issues Activity Actions

Rename it to generate

Browse source
This commit is contained in:
RGBCube 2022-06-26 17:28:01 +03:00
parent 132f42cf28
commit 193b8ad93d
3 changed files with 9 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

20
tools/unschema/parser.py
View file

@ -21,7 +21,7 @@ types = {
# super().append(obj)
def generate_typed_dicts_from_json_schema(
def generate(
obj: dict, /, *, title: str = "GeneratedObject", no_comments: bool = False
) -> Tuple[str, str]: # sourcery skip: low-code-quality
"""Makes TypedDict from a JSON Schema object.
@ -53,9 +53,7 @@ def generate_typed_dicts_from_json_schema(
if obj_schema["type"] == "null":
optional = True
else:
extras, target = generate_typed_dicts_from_json_schema(
obj_schema, title=title, no_comments=no_comments
)
extras, target = generate(obj_schema, title=title, no_comments=no_comments)
result.append(extras)
union_items.append(target)
@ -122,9 +120,7 @@ def generate_typed_dicts_from_json_schema(
typed_dict = [f"class {obj_title}(TypedDict{total}):"]
for key, value in obj_properties.items():
extras, param_annotation = generate_typed_dicts_from_json_schema(
value, title=key.capitalize(), no_comments=no_comments
)
extras, param_annotation = generate(value, title=key.capitalize(), no_comments=no_comments)
result.append(extras)
if key not in obj.get("required", []):
@ -151,9 +147,7 @@ def generate_typed_dicts_from_json_schema(
# maxContains, minLength, maxLength, uniqueItems
if obj_items := obj.get("items"):
extras, target = generate_typed_dicts_from_json_schema(
obj_items, no_comments=no_comments
)
extras, target = generate(obj_items, no_comments=no_comments)
result.append(extras)
annotation = f"List[{target}]"
@ -163,15 +157,13 @@ def generate_typed_dicts_from_json_schema(
(
extras,
target,
) = generate_typed_dicts_from_json_schema(item, no_comments=no_comments)
) = generate(item, no_comments=no_comments)
result.append(extras)
tuple_annotation.append(target)
if extra_item_type := obj.get("items"):
if extra_item_type is not True:
extras, extra_type = generate_typed_dicts_from_json_schema(
extra_item_type, no_comments=no_comments
)
extras, extra_type = generate(extra_item_type, no_comments=no_comments)
result.append(extras)
if not no_comments:
result.append(