Merge pull request #72 from ltonetwork/tx-use-none

Default properties as None instead of an empty string or zero.

Author: jasny

src/lto/crypto.py

@@ -82,14 +82,18 @@ def key_type_id(key_type):
         raise Exception('Key Type not supported')
 
 
-
 def merge_dicts(x, y):
     z = x.copy()
-    z.update(y)
+    z.update(y)    
     return z
 
 
+def clean_dict(x):
+    return {k:v for (k,v) in x.items() if v is not None}
+
+
 def compare_data_transaction(data, transaction):
     for key in data:
         key2 = inflection.underscore(key)
         assert data[key] == getattr(transaction, key2)
+

src/lto/transaction.py

@@ -7,17 +7,17 @@
 class Transaction(ABC):
 
     def __init__(self):
-        self.tx_fee = 0
-        self.timestamp = 0
-        self.proofs = []
-        self.sender = ''
-        self.sender_public_key = ''
-        self.chain_id = ''
-        self.sponsor = ''
-        self.sponsor_public_key = ''
-        self.sender_key_type = 'ed25519'
-        self.sponsor_key_type = 'ed25519'
         self.id = None
+        self.tx_fee = None
+        self.timestamp = None
+        self.sender = None
+        self.sender_key_type = None
+        self.sender_public_key = None
+        self.sponsor = None
+        self.sponsor_key_type = None
+        self.sponsor_public_key = None
+        self.chain_id = None
+        self.proofs = []
         self.height = None
 
     @abstractmethod
@@ -28,16 +28,17 @@ def is_signed(self):
         return len(self.proofs) != 0
 
     def sign_with(self, account):
-        if self.timestamp == 0:
+        if self.timestamp is None:
             self.timestamp = int(time() * 1000)
 
-        if self.sender == '':
+        if self.sender is None:
             self.sender = account.address
+            self.sender_key_type = account.key_type
             self.sender_public_key = account.get_public_key()
-
-        self.chain_id = account.get_network()
-        self.sender_key_type = account.key_type
-
+            
+        if self.chain_id is None:
+            self.chain_id = account.get_network()
+        
         self.proofs.append(account.sign(self.to_binary()))
 
     def sponsor_with(self, sponsor_account):
@@ -56,11 +57,3 @@ def broadcast_to(self, node):
     def to_json(self):
         pass
 
-    def _sponsor_json(self):
-        if self.sponsor:
-            return {"sponsor": self.sponsor,
-                    "sponsorPublicKey": self.sponsor_public_key,
-                    "sponsorKeyType": self.sponsor_key_type}
-        else:
-            return {}
-

src/lto/transactions/anchor.py

@@ -54,9 +54,8 @@ def to_binary(self):
             raise Exception('Incorrect Version')
 
     def to_json(self):
-        return (crypto.merge_dicts(
-            {
-                "id": self.id if self.id else "",
+        return crypto.clean_dict({
+                "id": self.id,
                 "type": self.TYPE,
                 "version": self.version,
                 "sender": self.sender,
@@ -65,10 +64,12 @@ def to_json(self):
                 "fee": self.tx_fee,
                 "timestamp": self.timestamp,
                 "anchors": list(map(lambda anchor: base58.b58encode(crypto.str2bytes(anchor)), self.anchors)),
-                "proofs": self.proofs,
-                "height": self.height if self.height else ""
-            },
-            self._sponsor_json()))
+                "sponsor": self.sponsor,
+                "sponsorKeyType": self.sponsor_key_type,
+                "sponsorPublicKey": self.sponsor_public_key,
+                "proofs": self.proofs or None,
+                "height": self.height
+            })
 
     @staticmethod
     def from_data(data):

src/lto/transactions/association.py

@@ -10,18 +10,15 @@ class Association(Transaction):
     TYPE = 16
     DEFAULT_VERSION = 3
 
-    def __init__(self, recipient, association_type, anchor='', expires=0):
+    def __init__(self, recipient, association_type, anchor='', expires=None):
         super().__init__()
         self.recipient = recipient
         self.association_type = association_type
         self.anchor = anchor
         self.tx_fee = self.DEFAULT_FEE
         self.version = self.DEFAULT_VERSION
-
         self.expires = expires
-        current = int(time() * 1000)
-        if self.expires != 0 and self.expires <= current:
-            raise Exception('Wring exipration date')
+      
 
     def __to_binary_v1(self):
         if self.anchor:
@@ -58,7 +55,7 @@ def __to_binary_v3(self):
                 struct.pack(">Q", self.tx_fee) +
                 base58.b58decode(self.recipient) +
                 struct.pack(">i", self.association_type) +
-                struct.pack(">Q", self.expires) +
+                struct.pack(">Q", self.expires or 0) +
                 struct.pack(">H", len(crypto.str2bytes(self.anchor))) +
                 crypto.str2bytes(self.anchor))
 
@@ -73,8 +70,8 @@ def to_binary(self):
 
 
     def to_json(self):
-        tx = {
-                "id": self.id if self.id else "",
+        return crypto.clean_dict({
+                "id": self.id,
                 "type": self.TYPE,
                 "version": self.version,
                 "sender": self.sender,
@@ -86,12 +83,12 @@ def to_json(self):
                 "timestamp": self.timestamp,
                 "expires": self.expires if self.version != 1 else None,
                 "fee": self.tx_fee,
-                "proofs": self.proofs,
-                "height": self.height if self.height else ""
-            }
-        if self.version == 1:
-            tx.pop('expires')
-        return crypto.merge_dicts(tx, self._sponsor_json())
+                "sponsor": self.sponsor,
+                "sponsorKeyType": self.sponsor_key_type,
+                "sponsorPublicKey": self.sponsor_public_key,
+                "proofs": self.proofs or None,
+                "height": self.height
+            })
 
 
     @staticmethod

src/lto/transactions/burn.py

@@ -34,19 +34,22 @@ def to_binary(self):
             raise Exception('Incorrect Version')
 
     def to_json(self):
-        return (crypto.merge_dicts({
+        return crypto.clean_dict({
+            "id": self.id,
             "type": self.TYPE,
             "version": self.version,
-            "id": self.id if self.id else "",
             "sender": self.sender,
             "senderKeyType": self.sender_key_type,
             "senderPublicKey": self.sender_public_key,
             "fee": self.tx_fee,
             "timestamp": self.timestamp,
             "amount": self.amount,
-            "proofs": self.proofs,
-            "height": self.height if self.height else ""
-        }, self._sponsor_json()))
+            "sponsor": self.sponsor,
+            "sponsorKeyType": self.sponsor_key_type,
+            "sponsorPublicKey": self.sponsor_public_key,
+            "proofs": self.proofs or None,
+            "height": self.height
+        })
 
     @staticmethod
     def from_data(data):

src/lto/transactions/cancel_lease.py

@@ -44,20 +44,22 @@ def to_binary(self):
             raise Exception('Incorrect Version')
 
     def to_json(self):
-        return(crypto.merge_dicts({
-            "id": self.id if self.id else "",
+        return crypto.clean_dict({
+            "id": self.id,
             "type": self.TYPE,
             "version": self.version,
             "sender": self.sender,
             "senderKeyType": self.sender_key_type,
             "senderPublicKey": self.sender_public_key,
             "fee": self.tx_fee,
             "timestamp": self.timestamp,
-            "proofs": self.proofs,
+            "sponsor": self.sponsor,
+            "sponsorKeyType": self.sponsor_key_type,
+            "sponsorPublicKey": self.sponsor_public_key,
+            "proofs": self.proofs or None,
             "leaseId": self.lease_id,
-            "height": self.height if self.height else ""
-        },
-        self._sponsor_json()))
+            "height": self.height
+        })
 
     @staticmethod
     def from_data(data):

src/lto/transactions/cancel_sponsorship.py

@@ -44,8 +44,8 @@ def to_binary(self):
             raise Exception('Incorrect Version')
 
     def to_json(self):
-        return (crypto.merge_dicts({
-            "id": self.id if self.id else "",
+        return crypto.clean_dict({
+            "id": self.id,
             "type": self.TYPE,
             "version": self.version,
             "senderKeyType": self.sender_key_type,
@@ -54,9 +54,12 @@ def to_json(self):
             "senderPublicKey": self.sender_public_key,
             "timestamp": self.timestamp,
             "fee": self.tx_fee,
-            "proofs": self.proofs,
-            "height": self.height if self.height else ""
-        }, self._sponsor_json()))
+            "sponsor": self.sponsor,
+            "sponsorKeyType": self.sponsor_key_type,
+            "sponsorPublicKey": self.sponsor_public_key,
+            "proofs": self.proofs or None,
+            "height": self.height
+        })
 
     @staticmethod
     def from_data(data):

src/lto/transactions/data.py

@@ -52,8 +52,8 @@ def to_binary(self):
             raise Exception('Incorrect Version')
 
     def to_json(self):
-        return (crypto.merge_dicts({
-                "id": self.id if self.id else "",
+        return crypto.clean_dict({
+                "id": self.id,
                 "type": self.TYPE,
                 "version": self.version,
                 "sender": self.sender,
@@ -62,10 +62,12 @@ def to_json(self):
                 "fee": self.tx_fee,
                 "timestamp": self.timestamp,
                 "data": list(map(lambda entry: entry.to_json(), self.data)),
-                "proofs": self.proofs,
-                "height": self.height if self.height else ""
-            },
-            self._sponsor_json()))
+                "sponsor": self.sponsor,
+                "sponsorKeyType": self.sponsor_key_type,
+                "sponsorPublicKey": self.sponsor_public_key,
+                "proofs": self.proofs or None,
+                "height": self.height
+            })
 
     def data_as_dict(self):
         dictionary = {}

src/lto/transactions/lease.py

@@ -48,8 +48,8 @@ def to_binary(self):
             raise Exception('Incorrect Version')
 
     def to_json(self):
-        return (crypto.merge_dicts({
-            "id": self.id if self.id else "",
+        return crypto.clean_dict({
+            "id": self.id,
             "type": self.TYPE,
             "version": self.version,
             "sender": self.sender,
@@ -59,10 +59,12 @@ def to_json(self):
             "amount": self.amount,
             "fee": self.tx_fee,
             "timestamp": self.timestamp,
-            "proofs": self.proofs,
-            "height": self.height if self.height else ""
-        },
-            self._sponsor_json()))
+            "sponsor": self.sponsor,
+            "sponsorKeyType": self.sponsor_key_type,
+            "sponsorPublicKey": self.sponsor_public_key,
+            "proofs": self.proofs or None,
+            "height": self.height
+        })
 
     @staticmethod
     def from_data(data):

src/lto/transactions/mass_transfer.py

@@ -64,20 +64,23 @@ def to_binary(self):
 
 
     def to_json(self):
-        return (crypto.merge_dicts({
-            "id": self.id if self.id else "",
+        return crypto.clean_dict({
+            "id": self.id,
             "type": self.TYPE,
             "version": self.version,
             "sender": self.sender,
             "senderKeyType": self.sender_key_type,
             "senderPublicKey": self.sender_public_key,
             "fee": self.tx_fee,
             "timestamp": self.timestamp,
-            "proofs": self.proofs,
             "attachment": base58.b58encode(crypto.str2bytes(self.attachment)),
             "transfers": self.transfers,
-            "height": self.height if self.height else ""
-        }, self._sponsor_json()))
+            "sponsor": self.sponsor,
+            "sponsorKeyType": self.sponsor_key_type,
+            "sponsorPublicKey": self.sponsor_public_key,
+            "proofs": self.proofs or None,
+            "height": self.height
+        })
 
     @staticmethod
     def from_data(data):