From 30aab9491ee1c58a02b6025da83e382c12a356eb Mon Sep 17 00:00:00 2001 From: Tim Rid <6593626+timrid@users.noreply.github.com> Date: Tue, 15 Dec 2020 17:06:41 +0100 Subject: [PATCH] fixed stubs for RawCopy --- construct-stubs/core.pyi | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/construct-stubs/core.pyi b/construct-stubs/core.pyi index 29864ac..22940ec 100644 --- a/construct-stubs/core.pyi +++ b/construct-stubs/core.pyi @@ -509,7 +509,19 @@ Terminated: Construct[None, None] #=============================================================================== # tunneling and byte/bit swapping #=============================================================================== -class RawCopy(Subconstruct[SubconParsedType, SubconBuildTypes, ParsedType, BuildTypes]): ... +@t.type_check_only +class RawCopyObj(t.Generic[ParsedType], Container[t.Any]): + data: bytes + value: ParsedType + offset1: int + offset2: int + length: int + +class RawCopy(Subconstruct[SubconParsedType, SubconBuildTypes, ParsedType, BuildTypes]): + def __new__( + cls, + subcon: Construct[SubconParsedType, SubconBuildTypes] + ) -> RawCopy[SubconParsedType, SubconBuildTypes, RawCopyObj[SubconParsedType], t.Dict[str, t.Union[SubconBuildTypes, BufferType]]]: ... def ByteSwapped(subcon: Construct[SubconParsedType, SubconBuildTypes]) -> Transformed[SubconParsedType, SubconBuildTypes]: ... def BitsSwapped(subcon: Construct[SubconParsedType, SubconBuildTypes]) -> t.Union[Transformed[SubconParsedType, SubconBuildTypes], Restreamed[SubconParsedType, SubconBuildTypes]]: ...