Fix formatting

* Revert autoformatted lines
* Rearrange imports correctly
This commit is contained in:
Alice Kits 2025-10-17 13:03:00 +05:00
parent fcf8a7d021
commit 175669d2b7
No known key found for this signature in database
GPG Key ID: 8D5EE63C035D3D3A
3 changed files with 17 additions and 6 deletions

View File

@ -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,

View File

@ -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('<i', len({sanitized_name}))")
@ -161,5 +165,7 @@ def generate_function(writer: SourceWriter, defn: Definition) -> 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)")

View File

@ -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