Merge pull request #291 from drov0/feature/update-proposal-end-date

support for the end_date extension for the update_proposal op

beem/version.py

-"""THIS FILE IS GENERATED FROM beem SETUP.PY."""
-version = '0.24.22'
+"""THIS FILE IS GENERATED FROM beem SETUP.PY."""
+version = '0.24.22'

beemapi/version.py

-"""THIS FILE IS GENERATED FROM beem SETUP.PY."""
-version = '0.24.22'
+"""THIS FILE IS GENERATED FROM beem SETUP.PY."""
+version = '0.24.22'

beembase/objects.py

@@ -324,3 +324,48 @@ class CommentOptionExtensions(Static_variant):
         else:
             raise Exception("Unknown CommentOptionExtension")
         super(CommentOptionExtensions, self).__init__(data, type_id)
+
+
+class UpdateProposalEndDate(GrapheneObject):
+    def __init__(self, *args, **kwargs):
+        if isArgsThisClass(self, args):
+            self.data = args[0].data
+        else:
+            if len(args) == 1 and len(kwargs) == 0:
+                kwargs = args[0]
+
+            super(UpdateProposalEndDate, self).__init__(
+                OrderedDict([
+                    ('end_date', PointInTime(kwargs['update_proposal_end_date'])),
+                ]))
+
+
+class UpdateProposalExtensions(Static_variant):
+    """ Serialize Update proposal extensions.
+
+        :param end_date: A static_variant containing the new end_date.
+
+        Example::
+
+            [1,
+                {'end_date': '2021-03-28T04:00:00'}
+            ]
+
+    """
+    def __init__(self, o):
+        if type(o) == dict and 'type' in o and 'value' in o:
+            if o['type'] == "end_date":
+                type_id = 1
+            else:
+                type_id = ~0
+            data = o['value']
+        else:
+            type_id, data = o
+
+        if type_id == 1:
+            data = (UpdateProposalEndDate(data))
+        else:
+            raise Exception("Unknown UpdateProposalExtension")
+        super(UpdateProposalExtensions, self).__init__(data, type_id)
+
+

beembase/operations.py

@@ -25,7 +25,8 @@ from .objects import (
     ExchangeRate,
     Beneficiaries,
     Beneficiary,
-    CommentOptionExtensions
+    CommentOptionExtensions,
+    UpdateProposalExtensions
 )
 
 default_prefix = "STM"
@@ -401,6 +402,9 @@ class Update_proposal(GrapheneObject):
 
         prefix = kwargs.get("prefix", default_prefix)
         extensions = Array([])
+        if "extensions" in kwargs and kwargs["extensions"]:
+            extensions = Array([UpdateProposalExtensions(o) for o in kwargs["extensions"]])
+
 
         super(Update_proposal, self).__init__(
             OrderedDict([

beembase/version.py

-"""THIS FILE IS GENERATED FROM beem SETUP.PY."""
-version = '0.24.22'
+"""THIS FILE IS GENERATED FROM beem SETUP.PY."""
+version = '0.24.22'

beemgraphenebase/version.py

-"""THIS FILE IS GENERATED FROM beem SETUP.PY."""
-version = '0.24.22'
+"""THIS FILE IS GENERATED FROM beem SETUP.PY."""
+version = '0.24.22'