implement no type checking flag (#230)

Author: juliusfrost

pycardano/backend/base.py

@@ -3,12 +3,11 @@
 from dataclasses import dataclass
 from typing import Dict, List, Union
 
-from typeguard import typechecked
-
 from pycardano.address import Address
 from pycardano.network import Network
 from pycardano.plutus import ExecutionUnits
 from pycardano.transaction import Transaction, UTxO
+from pycardano.types import typechecked
 
 __all__ = [
     "GenesisParameters",

pycardano/serialization.py

@@ -26,13 +26,13 @@
 
 from cbor2 import CBOREncoder, CBORSimpleValue, CBORTag, dumps, loads, undefined
 from pprintpp import pformat
-from typeguard import check_type, typechecked
 
 from pycardano.exception import (
     DeserializeException,
     InvalidArgumentException,
     SerializeException,
 )
+from pycardano.types import typechecked, check_type
 
 __all__ = [
     "default_encoder",

pycardano/transaction.py

@@ -11,7 +11,6 @@
 from cbor2 import CBORTag
 from nacl.encoding import RawEncoder
 from nacl.hash import blake2b
-from typeguard import typechecked
 
 from pycardano.address import Address
 from pycardano.certificate import Certificate
@@ -39,6 +38,7 @@
     default_encoder,
     list_hook,
 )
+from pycardano.types import typechecked
 from pycardano.witness import TransactionWitnessSet
 
 __all__ = [

pycardano/types.py

@@ -1,4 +1,22 @@
+import os
+from functools import partial
 from typing import Any, Dict
 
+import typeguard
+
 # https://github.com/python/typing/issues/182#issuecomment-199532520
 JsonDict = Dict[str, Any]
+
+
+def typechecked(func=None, *args, **kwargs):
+    if os.getenv("PYCARDANO_NO_TYPE_CHECK", "False").lower() in ("true", "1"):
+        if func is None:
+            return partial(typechecked, *args, **kwargs)
+        return func
+    return typeguard.typechecked(func, *args, **kwargs)
+
+
+def check_type(*args, **kwargs):
+    if os.getenv("PYCARDANO_NO_TYPE_CHECK", "False").lower() in ("true", "1"):
+        return None
+    return typeguard.check_type(*args, **kwargs)

test/pycardano/test_types.py

@@ -0,0 +1,20 @@
+import typeguard
+
+from pycardano.types import typechecked, check_type
+
+
+def test_types(monkeypatch):
+    monkeypatch.setenv("PYCARDANO_NO_TYPE_CHECK", "true")
+
+    assert typeguard.typechecked != typechecked
+    assert typeguard.check_type != check_type
+
+    @typechecked
+    def func1():
+        pass
+
+    @typechecked()
+    def func2():
+        pass
+
+    check_type()