|
14 | 14 | """
|
15 | 15 |
|
16 | 16 | from __future__ import annotations
|
17 |
| -from typing import Any |
| 17 | +from typing import Any, Set |
18 | 18 | import argparse
|
19 | 19 | import sys
|
20 | 20 | import os.path
|
21 | 21 | import importlib
|
22 | 22 | import logging
|
23 | 23 | import keyword
|
24 |
| -from dataclasses import dataclass, field |
| 24 | +import dataclasses |
25 | 25 |
|
26 | 26 | try:
|
27 | 27 | import black
|
@@ -57,21 +57,21 @@ def y(*args, **kwargs):
|
57 | 57 | }
|
58 | 58 |
|
59 | 59 |
|
60 |
| -@dataclass |
| 60 | +@dataclasses.dataclass |
61 | 61 | class Scope:
|
62 | 62 | name: str
|
63 | 63 | id: int
|
64 | 64 | parent: Scope | None
|
65 | 65 | return_scope: Scope | None
|
66 |
| - lines: list[str] = field(default_factory=list) |
| 66 | + lines: list[str] = dataclasses.field(default_factory=list) |
67 | 67 |
|
68 | 68 |
|
69 |
| -@dataclass |
| 69 | +@dataclasses.dataclass |
70 | 70 | class Type:
|
71 | 71 | schema: Any
|
72 | 72 | name: str
|
73 | 73 | scope: Scope
|
74 |
| - generic_params: list[str] = field(default_factory=list) |
| 74 | + generic_params: list[str] = dataclasses.field(default_factory=list) |
75 | 75 |
|
76 | 76 | @property
|
77 | 77 | def scope_path(self) -> list[Scope]:
|
@@ -319,14 +319,14 @@ def gen_struct(schema, writer, name: str = ""):
|
319 | 319 | writer.lookup_type(
|
320 | 320 | field.slot.type.list.elementType.struct.typeId
|
321 | 321 | )
|
322 |
| - except KeyError as ex: |
| 322 | + except KeyError: |
323 | 323 | gen_nested(raw_field.schema.elementType, writer)
|
324 | 324 | elif field.slot.type.list.elementType.which() == "enum":
|
325 | 325 | try:
|
326 | 326 | writer.lookup_type(
|
327 | 327 | field.slot.type.list.elementType.enum.typeId
|
328 | 328 | )
|
329 |
| - except KeyError as ex: |
| 329 | + except KeyError: |
330 | 330 | gen_nested(raw_field.schema.elementType, writer)
|
331 | 331 | type_name = writer.type_ref(field.slot.type.list.elementType)
|
332 | 332 | field_py_code = f"{field.name}: List[{type_name}]"
|
@@ -357,7 +357,7 @@ def gen_struct(schema, writer, name: str = ""):
|
357 | 357 | elem_type = raw_field.schema
|
358 | 358 | try:
|
359 | 359 | writer.lookup_type(elem_type.node.id)
|
360 |
| - except KeyError as ex: |
| 360 | + except KeyError: |
361 | 361 | gen_struct(elem_type, writer)
|
362 | 362 | type_name = writer.type_ref(field.slot.type)
|
363 | 363 | field_py_code = f"{field.name}: {type_name}"
|
@@ -397,9 +397,9 @@ def gen_struct(schema, writer, name: str = ""):
|
397 | 397 | scoped_name = ".".join([scope.name for scope in scope_path] + [name])
|
398 | 398 | else:
|
399 | 399 | scoped_name = name
|
400 |
| - writer.writeln(f"@staticmethod") |
| 400 | + writer.writeln("@staticmethod") |
401 | 401 | writer.writeln(f"def from_bytes(data: bytes) -> {scoped_name}: ...")
|
402 |
| - writer.writeln(f"def to_bytes(self) -> bytes: ...") |
| 402 | + writer.writeln("def to_bytes(self) -> bytes: ...") |
403 | 403 | have_body = True
|
404 | 404 |
|
405 | 405 | if schema.node.struct.discriminantCount:
|
|
0 commit comments