fix: basic allowance

terra-money · Feb 13, 2022 · 55d308b · 55d308b
1 parent 69a9673
commit 55d308b
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/terra_sdk/core/feegrant/data.py b/terra_sdk/core/feegrant/data.py
@@ -39,15 +39,15 @@ def to_amino(self) -> dict:
         return {
             "type": self.type_amino,
             "value": {
-                "spend_limit": self.spend_limit.to_amino(),
-                "expiration": to_isoformat(self.expiration)
+                "spend_limit": self.spend_limit.to_amino() if self.spend_limit else None,
+                "expiration": to_isoformat(self.expiration) if self.expiration else None
             }
         }
 
     def to_data(self) -> dict:
         return {
-            "spend_limit": self.spend_limit.to_data(),
-            "expiration": to_isoformat(self.expiration)
+            "spend_limit": self.spend_limit.to_data() if self.spend_limit else None,
+            "expiration": to_isoformat(self.expiration) if self.expiration else None
         }
 
     @classmethod
@@ -56,12 +56,13 @@ def from_data(cls, data: dict) -> BasicAllowance:
         exp = data.get("expiration")
         return cls(
             spend_limit=Coins.from_data(sl) if sl else None,
-            expiration=exp
+            expiration=exp if exp else None
         )
 
     def to_proto(self) -> BasicAllowance_pb:
         return BasicAllowance_pb(
-            spend_limit=self.spend_limit.to_proto(), expiration=self.expiration
+            spend_limit=self.spend_limit.to_proto() if self.spend_limit else [],
+            expiration=self.expiration
         )