jamcodec 0.1.1__tar.gz → 0.1.3__tar.gz

{jamcodec-0.1.1 → jamcodec-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: jamcodec
-Version: 0.1.1
+Version: 0.1.3
 Summary: Python JAM Codec Library
 Author-email: JAMdot Technologies <dev@jamdot.tech>
 Project-URL: Homepage, https://github.com/JAMdotTech/py-jam-codec
@@ -21,6 +21,7 @@ Requires-Dist: mkdocs-material; extra == "dev"
 Requires-Dist: mkdocs-autorefs; extra == "dev"
 Requires-Dist: mkdocstrings; extra == "dev"
 Requires-Dist: mkdocstrings[python]; extra == "dev"
+Dynamic: license-file
 
 # Python JAM Codec
 
@@ -42,7 +43,7 @@ pip install jamcodec
 ## Code examples
 
 ```python
-from jamcodec.types import JamBytes, Bool, String, U32, U8, U16, Struct, Vec, Compact, Tuple, Enum
+from jamcodec.types import JamBytes, Bool, String, U32, U8, U16, Struct, Vec, VarInt64, Tuple, Enum
 
 # encode a Vec<u16>
 obj = Vec(U16).new()
@@ -57,7 +58,7 @@ value = scale_obj.decode(JamBytes("0x020105"))
 scale_obj = Enum(
     Bool=Bool(),
     Number=U32,
-    Complex=Struct(data=String(), version=Compact(U8)),
+    Complex=Struct(data=String(), version=VarInt64),
     None_=None
 ).new()
 value = {'Bool': True}

{jamcodec-0.1.1 → jamcodec-0.1.3}/README.md

@@ -18,7 +18,7 @@ pip install jamcodec
 ## Code examples
 
 ```python
-from jamcodec.types import JamBytes, Bool, String, U32, U8, U16, Struct, Vec, Compact, Tuple, Enum
+from jamcodec.types import JamBytes, Bool, String, U32, U8, U16, Struct, Vec, VarInt64, Tuple, Enum
 
 # encode a Vec<u16>
 obj = Vec(U16).new()
@@ -33,7 +33,7 @@ value = scale_obj.decode(JamBytes("0x020105"))
 scale_obj = Enum(
     Bool=Bool(),
     Number=U32,
-    Complex=Struct(data=String(), version=Compact(U8)),
+    Complex=Struct(data=String(), version=VarInt64),
     None_=None
 ).new()
 value = {'Bool': True}

{jamcodec-0.1.1 → jamcodec-0.1.3}/jamcodec/types.py

@@ -371,8 +371,11 @@ class Enum(JamCodecTypeDef):
 
         for idx, (variant_name, variant_obj) in enumerate(self.variants.items()):
 
-            if enum_key == variant_name:
+            if type(variant_obj) is dict:
+                idx = variant_obj.get('id', idx)
+                variant_obj = variant_obj.get('type', variant_obj)
 
+            if enum_key == variant_name:
                 data = JamBytes(bytearray([idx]))
 
                 if variant_obj is not None:
@@ -387,9 +390,19 @@ class Enum(JamCodecTypeDef):
 
         index = int.from_bytes(data.get_next_bytes(1), byteorder='little')
 
-        try:
-            enum_key, enum_variant = list(self.variants.items())[index]
-        except IndexError:
+        enum_key = None
+        enum_variant = None
+
+        for idx, (variant_key, variant_obj) in enumerate(self.variants.items()):
+            if type(variant_obj) is dict:
+                idx = variant_obj.get('id', idx)
+                variant_obj = variant_obj.get('type', variant_obj)
+            if index == idx:
+                enum_key = variant_key
+                enum_variant = variant_obj
+                break
+
+        if enum_key is None:
             raise ScaleDecodeException(f"Index '{index}' not present in Enum type mapping")
 
         if enum_variant is None:
@@ -434,6 +447,9 @@ class Enum(JamCodecTypeDef):
 
         for idx, (variant_name, variant_obj) in enumerate(self.variants.items()):
 
+            if type(variant_obj) is dict:
+                variant_obj = variant_obj.get('type', variant_obj)
+
             if enum_key == variant_name:
 
                 if variant_obj is not None:
@@ -508,7 +524,7 @@ class Option(JamCodecTypeDef):
         return None, self.some.example_value()
 
 
-class Compact(JamCodecTypeDef):
+class CompactDef(JamCodecTypeDef):
     def __init__(self, type_: JamCodecTypeDef = None):
         self.type = type_
         self.compact_length = 0
@@ -595,7 +611,6 @@ class Vec(JamCodecTypeDef):
         self.type_def = type_def
 
     def encode(self, value: list) -> JamBytes:
-
         if self.type_def is U8:
             return Bytes.encode(value)
 
@@ -959,8 +974,11 @@ class Map(JamCodecTypeDef):
     def encode(self, value: Union[typing.Mapping, list]) -> JamBytes:
 
         if isinstance(value, Mapping):
-            # Sort Map by keys
-            value = sorted(value.items(), key=lambda x: x[0])
+            # Convert to list
+            value = value.items()
+
+        # Sort Map by keys
+        value = sorted(value, key=lambda x: x[0])
 
         # Encode length of Vec
         data = VarInt64.encode(len(value))
@@ -1137,9 +1155,6 @@ class HashDef(JamCodecTypeDef):
 
 class VarInt64Def(JamCodecTypeDef):
 
-    def encode(self, value: any, external_call=True) -> JamBytes:
-        return self.encode(value)
-
     def encode(self, value: int) -> JamBytes:
         if value < 0:
             raise ScaleEncodeException("Cannot encode negative value")
@@ -1235,3 +1250,4 @@ Hash = H256
 HashMap = Map
 BTreeMap = Map
 VarInt64 = VarInt64Def()
+Compact = CompactDef()

{jamcodec-0.1.1 → jamcodec-0.1.3}/jamcodec.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: jamcodec
-Version: 0.1.1
+Version: 0.1.3
 Summary: Python JAM Codec Library
 Author-email: JAMdot Technologies <dev@jamdot.tech>
 Project-URL: Homepage, https://github.com/JAMdotTech/py-jam-codec
@@ -21,6 +21,7 @@ Requires-Dist: mkdocs-material; extra == "dev"
 Requires-Dist: mkdocs-autorefs; extra == "dev"
 Requires-Dist: mkdocstrings; extra == "dev"
 Requires-Dist: mkdocstrings[python]; extra == "dev"
+Dynamic: license-file
 
 # Python JAM Codec
 
@@ -42,7 +43,7 @@ pip install jamcodec
 ## Code examples
 
 ```python
-from jamcodec.types import JamBytes, Bool, String, U32, U8, U16, Struct, Vec, Compact, Tuple, Enum
+from jamcodec.types import JamBytes, Bool, String, U32, U8, U16, Struct, Vec, VarInt64, Tuple, Enum
 
 # encode a Vec<u16>
 obj = Vec(U16).new()
@@ -57,7 +58,7 @@ value = scale_obj.decode(JamBytes("0x020105"))
 scale_obj = Enum(
     Bool=Bool(),
     Number=U32,
-    Complex=Struct(data=String(), version=Compact(U8)),
+    Complex=Struct(data=String(), version=VarInt64),
     None_=None
 ).new()
 value = {'Bool': True}

{jamcodec-0.1.1 → jamcodec-0.1.3}/test/test_compact.py

@@ -19,52 +19,52 @@ import unittest
 
 from jamcodec.base import JamBytes
 from jamcodec.exceptions import ScaleDecodeException, RemainingScaleBytesNotEmptyException
-from jamcodec.types import Compact, U32, U128
+from jamcodec.types import CompactDef, U32, U128
 
 
 class TestCompact(unittest.TestCase):
 
     def test_compact_u32(self):
-        obj = Compact(U32).new()
+        obj = CompactDef(U32).new()
         obj.decode(JamBytes("0x02093d00"))
         self.assertEqual(obj.value, 1000000)
 
     def test_compact_u32_1byte(self):
-        obj = Compact(U32).new()
+        obj = CompactDef(U32).new()
         obj.decode(JamBytes("0x18"))
         self.assertEqual(obj.value, 6)
 
     def test_compact_u32_remaining_bytes(self):
-        obj = Compact(U32).new()
+        obj = CompactDef(U32).new()
         with self.assertRaises(ScaleDecodeException):
             obj.decode(JamBytes("0x02093d0001"), check_remaining=True)
 
     def test_compact_u32_invalid(self):
-        obj = Compact(U32).new()
+        obj = CompactDef(U32).new()
         self.assertRaises(RemainingScaleBytesNotEmptyException, obj.decode, JamBytes("0x"))
 
     def test_compact_u32_1byte_encode(self):
-        obj = Compact(U32).new()
+        obj = CompactDef(U32).new()
         obj.decode(JamBytes("0x18"))
 
-        obj = Compact(U32).new()
+        obj = CompactDef(U32).new()
         obj.encode(6)
         self.assertEqual(str(obj.data), "0x18")
 
     def test_compact_u32_2bytes_encode(self):
-        obj = Compact(U32).new()
+        obj = CompactDef(U32).new()
         obj.encode(6000)
         self.assertEqual(str(obj.data), "0xc15d")
 
     def test_compact_u32_4bytes_encode(self):
 
-        obj = Compact(U32).new()
+        obj = CompactDef(U32).new()
         obj.encode(1000000)
         self.assertEqual(str(obj.data), "0x02093d00")
 
     def test_compact_u32_larger_than_4bytes_encode(self):
 
-        obj = Compact(U32).new()
+        obj = CompactDef(U32).new()
         obj.encode(150000000000000)
         self.assertEqual(str(obj.data), "0x0b0060b7986c88")
 
@@ -72,10 +72,10 @@ class TestCompact(unittest.TestCase):
 
         value = 2000001
 
-        obj = Compact(U32).new()
+        obj = CompactDef(U32).new()
         data = obj.encode(value)
 
-        obj = Compact(U32).new()
+        obj = CompactDef(U32).new()
 
         self.assertEqual(obj.decode(data), value)
 
@@ -83,10 +83,10 @@ class TestCompact(unittest.TestCase):
 
         value = 2**30
 
-        obj = Compact(U32).new()
+        obj = CompactDef(U32).new()
         data = obj.encode(value)
 
-        obj = Compact(U32).new()
+        obj = CompactDef(U32).new()
 
         self.assertEqual(obj.decode(data), value)
 
@@ -96,7 +96,7 @@ class TestCompact(unittest.TestCase):
 
         Balance = U128
 
-        obj = Compact(Balance).new()
+        obj = CompactDef(Balance).new()
         data = obj.encode(value)
 
         self.assertEqual(str(scale_data), str(data))
@@ -105,13 +105,13 @@ class TestCompact(unittest.TestCase):
 
     def test_balance(self):
         Balance = U128
-        obj = Compact(Balance).new()
+        obj = CompactDef(Balance).new()
         obj.decode(JamBytes("0x130080cd103d71bc22"))
         self.assertEqual(obj.value, 2503000000000000000)
 
     def test_compact_no_type(self):
         data = JamBytes("0x02093d00")
-        compact = Compact().new()
+        compact = CompactDef().new()
         compact.decode(data)
         self.assertEqual(compact.value, 1000000)
 

{jamcodec-0.1.1 → jamcodec-0.1.3}/test/test_enum.py

@@ -17,7 +17,7 @@
 
 import unittest
 
-from jamcodec.types import Enum, Bool, U32
+from jamcodec.types import Enum, Bool, U32, String
 
 
 class TestEnum(unittest.TestCase):
@@ -60,6 +60,14 @@ class TestEnum(unittest.TestCase):
         scale_obj.deserialize('None')
         self.assertEqual(('None', None), scale_obj.value_object)
 
+    def test_enum_explicit_id(self):
+        scale_obj = Enum(Bool=Bool(), Number=U32, Error={'id': 255, 'type': String}).new()
+        jam_bytes = scale_obj.encode({'Error': 'test'})
+        self.assertEqual('0xff0474657374', jam_bytes.to_hex())
+
+        value = scale_obj.decode(jam_bytes)
+        self.assertEqual({'Error': 'test'}, value)
+        
 
 if __name__ == '__main__':
     unittest.main()

{jamcodec-0.1.1 → jamcodec-0.1.3}/test/test_map.py

@@ -24,14 +24,13 @@ from jamcodec.types import U32, Map, H256, Bytes
 
 class TestMap(unittest.TestCase):
 
-    def test_map_encode(self):
+    def test_map_dict_encode(self):
 
         obj = Map(H256, Bytes).new()
 
         value = {
             "0xbb30a42c1e62f0afda5f0a4e8a562f7a13a24cea00ee81917b86b89e801314aa": b'\x01\x02',
-            "0x03170a2e7597b7b7e3d84c05391d139a62b157e78786d8c082f29dcf4c111314": b'test',
-
+            "0x03170a2e7597b7b7e3d84c05391d139a62b157e78786d8c082f29dcf4c111314": b'test'
         }
         # Should be encoded sorted by key
         data = obj.encode(value)
@@ -40,6 +39,23 @@ class TestMap(unittest.TestCase):
             data
         )
 
+    def test_map_encode_list(self):
+        obj = Map(H256, Bytes).new()
+
+        value = [
+            ("0xbb30a42c1e62f0afda5f0a4e8a562f7a13a24cea00ee81917b86b89e801314aa",  b'\x01\x02'),
+            ("0x03170a2e7597b7b7e3d84c05391d139a62b157e78786d8c082f29dcf4c111314", b'test')
+        ]
+
+        # Should be encoded sorted by key
+        data = obj.encode(value)
+        self.assertEqual(
+            JamBytes(
+                "0x0203170a2e7597b7b7e3d84c05391d139a62b157e78786d8c082f29dcf4c1113140474657374bb30a42c1e62f0afda5f0a4e8a562f7a13a24cea00ee81917b86b89e801314aa020102"
+                ),
+            data
+        )
+
     def test_map_decode(self):
         obj = Map(U32, Bytes).new()
 
@@ -49,6 +65,8 @@ class TestMap(unittest.TestCase):
         self.assertEqual({2: b'test'}, obj.to_serializable_obj())
 
 
+
+
     def test_mapping_type(self):
 
         class StorageMap(Mapping):

{jamcodec-0.1.1 → jamcodec-0.1.3}/test/test_scale_types.py

@@ -17,7 +17,7 @@
 import unittest
 
 from jamcodec.base import JamBytes
-from jamcodec.types import Compact, U32, U16, Tuple
+from jamcodec.types import CompactDef, U32, U16, Tuple
 
 
 class TestScaleTypes(unittest.TestCase):
@@ -29,19 +29,19 @@ class TestScaleTypes(unittest.TestCase):
         self.assertEqual(obj.value, 64302)
 
     def test_value_equals_value_serialized_and_value_object(self):
-        obj = Tuple(Compact(U32), Compact(U32)).new()
+        obj = Tuple(CompactDef(U32), CompactDef(U32)).new()
         obj.decode(JamBytes("0x0c00"))
         self.assertEqual(obj.value, obj.value_serialized)
         self.assertEqual(obj.value, obj.value_object)
 
     def test_value_object(self):
-        obj = Tuple(Compact(U32), Compact(U32)).new()
+        obj = Tuple(CompactDef(U32), CompactDef(U32)).new()
         obj.decode(JamBytes("0x0c00"))
         self.assertEqual(obj.value_object[0].value_object, 3)
         self.assertEqual(obj.value_object[1].value_object, 0)
 
     def test_value_object_shorthand(self):
-        obj = Tuple(Compact(U32), Compact(U32)).new()
+        obj = Tuple(CompactDef(U32), CompactDef(U32)).new()
         obj.decode(JamBytes("0x0c00"))
         self.assertEqual(obj[0], 3)
         self.assertEqual(obj[1], 0)

{jamcodec-0.1.1 → jamcodec-0.1.3}/test/test_scalebytes.py

@@ -21,7 +21,7 @@ import unittest
 
 from jamcodec.base import JamBytes
 from jamcodec.exceptions import RemainingScaleBytesNotEmptyException
-from jamcodec.types import Compact, U32, Array, U8, String
+from jamcodec.types import CompactDef, U32, Array, U8, String
 
 
 class TestScaleBytes(unittest.TestCase):
@@ -31,7 +31,7 @@ class TestScaleBytes(unittest.TestCase):
         self.assertRaises(ValueError, JamBytes, 'test')
 
     def test_bytes_data_format(self):
-        obj = Compact(U32).new()
+        obj = CompactDef(U32).new()
         obj.decode(JamBytes(b"\x02\x09\x3d\x00"))
         self.assertEqual(obj.value, 1000000)
 

{jamcodec-0.1.1 → jamcodec-0.1.3}/test/test_tuple.py

@@ -18,13 +18,13 @@
 import unittest
 
 from jamcodec.base import JamBytes
-from jamcodec.types import Tuple, Compact, U32, U8
+from jamcodec.types import Tuple, CompactDef, U32, U8
 
 
 class TestTuple(unittest.TestCase):
 
     def test_tuple_decode(self):
-        obj = Tuple(Compact(U32), Compact(U32)).new()
+        obj = Tuple(CompactDef(U32), CompactDef(U32)).new()
         obj.decode(JamBytes("0x0c00"))
         self.assertEqual(obj.value, (3, 0))
 
@@ -38,7 +38,7 @@ class TestTuple(unittest.TestCase):
         self.assertEqual(value, scale_obj.value)
 
     def test_tuple_deserialize(self):
-        obj = Tuple(Compact(U32), Compact(U32)).new()
+        obj = Tuple(CompactDef(U32), CompactDef(U32)).new()
         obj.deserialize((3, 2))
         self.assertEqual(obj.value, (3, 2))