diff --git a/generator/src/telethon_generator/_impl/codegen/generator.py b/generator/src/telethon_generator/_impl/codegen/generator.py index ed8d49d3..84c2c51a 100644 --- a/generator/src/telethon_generator/_impl/codegen/generator.py +++ b/generator/src/telethon_generator/_impl/codegen/generator.py @@ -6,8 +6,9 @@ from .serde.common import ( inner_type_fmt, is_computed, param_type_fmt, + sanitize_name, to_class_name, - to_method_name, sanitize_name, + to_method_name, ) from .serde.deserialization import ( function_deserializer_fmt, diff --git a/generator/src/telethon_generator/_impl/codegen/serde/serialization.py b/generator/src/telethon_generator/_impl/codegen/serde/serialization.py index 361f96a3..34e4f60e 100644 --- a/generator/src/telethon_generator/_impl/codegen/serde/serialization.py +++ b/generator/src/telethon_generator/_impl/codegen/serde/serialization.py @@ -15,7 +15,9 @@ def param_value_expr(param: Parameter) -> str: return f"{pre}{mid}{suf}" -def generate_buffer_append(writer: SourceWriter, buffer: str, name: str, ty: Type) -> None: +def generate_buffer_append( + writer: SourceWriter, buffer: str, name: str, ty: Type +) -> None: sanitized_name = sanitize_name(name) if is_trivial(NormalParameter(ty=ty, flag=None)): @@ -60,7 +62,9 @@ def generate_normal_param_write( if param.ty.generic_arg: if param.ty.name not in ("Vector", "vector"): - raise ValueError("generic_arg deserialization for non-vectors is not supported") + raise ValueError( + "generic_arg deserialization for non-vectors is not supported" + ) if param.ty.bare: writer.write(f"{buffer} += struct.pack(' None: for param in iter: if not isinstance(param.ty, NormalParameter): raise RuntimeError("FlagsParameter should be considered trivial") - generate_normal_param_write(writer, tmp_names, "_buffer", param.name, param.ty) + generate_normal_param_write( + writer, tmp_names, "_buffer", param.name, param.ty + ) writer.write("return Request(b'' + _buffer)") diff --git a/generator/tests/generator_test.py b/generator/tests/generator_test.py index eef606c0..38267dee 100644 --- a/generator/tests/generator_test.py +++ b/generator/tests/generator_test.py @@ -14,7 +14,9 @@ def gen_py_code( functiondefs: Optional[list[Definition]] = None, ) -> str: fs = FakeFs() - generate(fs, ParsedTl(layer=0, typedefs=typedefs or [], functiondefs=functiondefs or [])) + generate( + fs, ParsedTl(layer=0, typedefs=typedefs or [], functiondefs=functiondefs or []) + ) generated = bytearray() for path, data in fs._files.items(): if path.stem not in ("__init__", "layer"): @@ -25,7 +27,9 @@ def gen_py_code( def test_generic_functions_use_bytes_parameters() -> None: - definitions = get_definitions("invokeWithLayer#da9b0d0d {X:Type} layer:int query:!X = X;") + definitions = get_definitions( + "invokeWithLayer#da9b0d0d {X:Type} layer:int query:!X = X;" + ) result = gen_py_code(functiondefs=definitions) assert "invoke_with_layer" in result assert "query: _bytes" in result