Add assert_operation

Author: xeroc

bitsharesbase/objects.py

@@ -22,6 +22,7 @@
     Varint32,
     Void,
     VoteId,
+    Ripemd160,
 )
 
 from .account import PublicKey
@@ -31,6 +32,8 @@
 
 default_prefix = "BTS"
 
+BlockId = Ripemd160
+
 
 class Operation(GrapheneOperation):
     """ Need to overwrite a few attributes to load proper operations from
@@ -319,7 +322,7 @@ def __init__(self, *args, **kwargs):
         elif id == 2:
             data = Burn_worker_initializer(o[1])
         else:
-            raise Exception("Unknown Worker_initializer")
+            raise ValueError("Unknown {}".format(self.__class__.name))
         super().__init__(data, id)
 
 
@@ -415,3 +418,46 @@ def targetCollateralRatio(value):
             return None
 
     sorted_options = [("target_collateral_ratio", targetCollateralRatio)]
+
+
+class AssertPredicate(Static_variant):
+    def __init__(self, o):
+        class Account_name_eq_lit_predicate(GrapheneObject):
+            def __init__(self, *args, **kwargs):
+                kwargs.update(args[0])
+                super().__init__(
+                    OrderedDict(
+                        [
+                            ("account_id", ObjectId(kwargs["account_id"], "account")),
+                            ("name", String(kwargs["name"])),
+                        ]
+                    )
+                )
+
+        class Asset_symbol_eq_lit_predicate(GrapheneObject):
+            def __init__(self, *args, **kwargs):
+                kwargs.update(args[0])
+                super().__init__(
+                    OrderedDict(
+                        [
+                            ("asset_id", ObjectId(kwargs["asset_id"], "asset")),
+                            ("symbol", String(kwargs["symbol"])),
+                        ]
+                    )
+                )
+
+        class Block_id_predicate(GrapheneObject):
+            def __init__(self, *args, **kwargs):
+                kwargs.update(args[0])
+                super().__init__(OrderedDict([("id", BlockId(kwargs["id"]))]))
+
+        id = o[0]
+        if id == 0:
+            data = Account_name_eq_lit_predicate(o[1])
+        elif id == 1:
+            data = Asset_symbol_eq_lit_predicate(o[1])
+        elif id == 2:
+            data = Block_id_predicate(o[1])
+        else:
+            raise ValueError("Unknown {}".format(self.__class__.name))
+        super().__init__(data, id)

bitsharesbase/operations.py

@@ -1,8 +1,5 @@
 # -*- coding: utf-8 -*-
-import json
-
 from collections import OrderedDict
-from binascii import hexlify, unhexlify
 
 from graphenebase.types import (
     Array,
@@ -16,7 +13,6 @@
     Optional,
     PointInTime,
     Set,
-    Signature,
     Static_variant,
     String,
     Uint8,
@@ -25,7 +21,6 @@
     Uint64,
     Varint32,
     Void,
-    VoteId,
     Ripemd160,
     Sha1,
     Sha256,
@@ -46,9 +41,9 @@
     Permission,
     Price,
     PriceFeed,
-    SpecialAuthority,
     Worker_initializer,
     isArgsThisClass,
+    AssertPredicate,
 )
 from .operationids import operations
 
@@ -918,9 +913,15 @@ def detail(self, *args, **kwargs):
         return OrderedDict(
             [
                 ("fee", Asset(kwargs["fee"])),
-                ("deposit_to_account", ObjectId(kwargs["deposit_to_account"], "account")),
+                (
+                    "deposit_to_account",
+                    ObjectId(kwargs["deposit_to_account"], "account"),
+                ),
                 ("balance_to_claim", ObjectId(kwargs["balance_to_claim"], "balance")),
-                ("balance_owner_key", PublicKey(kwargs["balance_owner_key"], prefix=prefix)),
+                (
+                    "balance_owner_key",
+                    PublicKey(kwargs["balance_owner_key"], prefix=prefix),
+                ),
                 ("total_claimed", Asset(kwargs["total_claimed"])),
             ]
         )
@@ -1013,4 +1014,36 @@ def __init__(self, *args, **kwargs):
             )
 
 
+class Assert(GrapheneObject):
+    def __init__(self, *args, **kwargs):
+        if isArgsThisClass(self, args):
+            self.data = args[0].data
+        else:
+            super().__init__(
+                OrderedDict(
+                    [
+                        ("fee", Asset(kwargs["fee"])),
+                        (
+                            "fee_paying_account",
+                            ObjectId(kwargs["fee_paying_account"], "account"),
+                        ),
+                        (
+                            "predicates",
+                            Array([AssertPredicate(o) for o in kwargs["predicates"]]),
+                        ),
+                        (
+                            "required_auths",
+                            Array(
+                                [
+                                    ObjectId(o, "account")
+                                    for o in kwargs["required_auths"]
+                                ]
+                            ),
+                        ),
+                        ("extensions", Set([])),
+                    ]
+                )
+            )
+
+
 fill_classmaps()

tests/test_transactions.py

@@ -926,6 +926,48 @@ def test_htlc_extend(self):
         )
         self.doit(0)
 
+    def test_assert_a(self):
+        self.op = operations.Assert(
+            **{
+                "fee": {"amount": 0, "asset_id": "1.3.0"},
+                "fee_paying_account": "1.2.0",
+                "predicates": [],
+                "required_auths": [],
+                "extensions": [],
+            }
+        )
+        self.cm = (
+            "f68585abf4dce7c80457012400000000000000000000000000"
+            "00011f2ee18d66da4314d7fa4d4cf5ad327e7c0504ace99e5d"
+            "757b214107a955e363fe7a371aaa16101741163ce64d0c0ee0"
+            "5c799b94467dbf15322d0f6bd7c5ed29a4"
+        )
+        self.doit(0)
+
+    def test_assert_b(self):
+        self.op = operations.Assert(
+            **{
+                "fee": {"amount": 0, "asset_id": "1.3.0"},
+                "fee_paying_account": "1.2.4",
+                "predicates": [
+                    [0, {"account_id": "1.2.2414", "name": "foobar"}],
+                    [1, {"asset_id": "1.3.2424", "symbol": "USD"}],
+                    [2, {"id": "0260c042666dd23b6c380f55be2eaae0086f643f"}],
+                ],
+                "required_auths": ["1.2.124124"],
+                "extensions": [],
+            }
+        )
+        self.cm = (
+            "f68585abf4dce7c804570124000000000000000000040300ee"
+            "1206666f6f62617201f81203555344020260c042666dd23b6c"
+            "380f55be2eaae0086f643f01dcc9070000011f3820288751dd"
+            "38b422c3930cb5b703e8f3af06aa50fde004eb46a480846dd4"
+            "4e01b0d4893934225befa42e1077ab2d71fc13fff5c6e0697c"
+            "f103aec7dc8e7496"
+        )
+        self.doit(0)
+
     def compareConstructedTX(self):
         self.maxDiff = None
         self.op = operations.Call_order_update(