check for restricted transfer, transfer_from -> transfer

ayshptk · ayshptk · commit c95fa0dde95d · 2022-01-04T16:18:13.000+05:30
diff --git a/thirdweb/errors/__init__.py b/thirdweb/errors/__init__.py
@@ -21,3 +21,7 @@ class UploadError(Exception):
     def __init__(self, message: str):
         super().__init__(f"There was an error while uploading the image : {message}")
 
+
+class RestrictedTransferError(Exception):
+    def __init__(self, address: str = None):
+        super().__init__(f"Failed to transfer asset, transfer is restricted.${f'Address : {address}' if address is not None else ''}")
diff --git a/thirdweb/modules/currency.py b/thirdweb/modules/currency.py
@@ -1,6 +1,7 @@
 """
 Interact with the Currency module of the app.
 """
+from thirdweb.errors import RestrictedTransferError
 from thirdweb_web3 import Web3
 
 from ..abi.coin import Coin
@@ -156,7 +157,9 @@ def transfer(self, to_address: str, amount: int):
 
         Transfers the given amount from the current address
         """
-        return self.execute_tx(self.__abi_module.transfer_from.build_transaction(
+        if(self.__abi_module.is_restricted_transfer()):
+            raise RestrictedTransferError(self.address)
+        return self.execute_tx(self.__abi_module.transfer.build_transaction(
             self.get_signer_address(), to_address, amount, self.get_transact_opts()
         ))        
 
