Merge pull request #12 from dannyhann/feature/issue-11

add to_dict method

Author: dannyhann

protobuf_decoder/protobuf_decoder.py

@@ -73,6 +73,19 @@ def __str__(self):
     def __repr__(self):
         return self.__str__()
 
+    def to_dict(self):
+        dict_result = dict(
+            value=self.value,
+            signed_int=self.signed_int,
+            unsigned_int=self.unsigned_int,
+            value_type=self._value_type,
+        )
+
+        if not self._is_unsigned:
+            dict_result.pop("unsigned_int")
+
+        return dict_result
+
 
 @dataclass(init=False)
 class ParsedResult:
@@ -86,8 +99,10 @@ def __init__(self, field: int, wire_type: str, data: ParsedDataType):
         self.data = data
 
     def to_dict(self):
-        if isinstance(self.data, list):
-            data = [sub_result.to_dict() for sub_result in self.data]
+        if isinstance(self.data, ParsedResults):
+            data = self.data.to_dict()
+        elif isinstance(self.data, FixedBitsValue):
+            data = self.data.to_dict()
         else:
             data = self.data
 
@@ -109,6 +124,12 @@ def has_results(self):
     def __getitem__(self, item):
         return self.results[item]
 
+    def to_dict(self):
+        results = [result.to_dict() for result in self.results]
+        return dict(
+            results=results,
+        )
+
 
 class State(Enum):
     FIND_FIELD = 1