Load plutus script from json file (#453)

* Load plutus script from json file

* fix qa

Author: cffls

pycardano/plutus.py

@@ -1064,25 +1064,52 @@ def plutus_script_hash(script: Union[NativeScript, PlutusScript]) -> ScriptHash:
     return script_hash(script)
 
 
-class PlutusScript(bytes):
+class PlutusScript(CBORSerializable, bytes):
+    """
+    Plutus script class.
+
+    This class is a base class for all Plutus script versions.
+
+    Example - Load a Plutus script from `test/resources/scriptV2.plutus <https://github.com/Python-Cardano/pycardano/blob/main/test/resources/scriptV2.plutus>`_ and get its address: # noqa: E501
+
+
+        >>> from pycardano import Address, Network
+        >>> script = PlutusV2Script.load("test/resources/scriptV2.plutus")
+        >>> Address(plutus_script_hash(script), network=Network.TESTNET).encode()
+        'addr_test1wrmz3pjz4dmfxj0fc0a0eyw69tp6h7mpndzf9g3kttq9cqqqw47ym'
+    """
+
     @property
     def version(self) -> int:
         raise NotImplementedError("")
 
+    def to_shallow_primitive(self) -> bytes:
+        return bytes(self)
+
+    @classmethod
+    def from_primitive(
+        cls: Type[PlutusScript], value: Any, type_args: Optional[tuple] = None
+    ) -> PlutusScript:
+        if not isinstance(value, (bytes, bytearray)):
+            raise DeserializeException(f"Expect bytes, got {type(value)} instead.")
+        return cls(value)
+
     @classmethod
     def from_version(cls, version: int, script_data: bytes) -> "PlutusScript":
-        if version == 1:
-            return PlutusV1Script(script_data)
-        elif version == 2:
-            return PlutusV2Script(script_data)
-        elif version == 3:
-            return PlutusV3Script(script_data)
-        else:
+        class_name = f"PlutusV{version}Script"
+        script_class = globals().get(class_name)
+
+        if script_class is None:
             raise ValueError(f"No Plutus script class found for version {version}")
 
+        return script_class(script_data)
+
     def get_script_hash_prefix(self) -> bytes:
         raise NotImplementedError("")
 
+    def __repr__(self):
+        return f"{self.__class__.__name__}({self.hex()})"
+
 
 class PlutusV1Script(PlutusScript):
     def get_script_hash_prefix(self) -> bytes:

pycardano/serialization.py

@@ -720,14 +720,6 @@ def _restore_typed_primitive(
         if not isinstance(v, bytes):
             raise DeserializeException(f"Expected type bytes but got {type(v)}")
         return ByteString(v)
-    elif isclass(t) and t.__name__ in [
-        "PlutusV1Script",
-        "PlutusV2Script",
-        "PlutusV3Script",
-    ]:
-        if not isinstance(v, bytes):
-            raise DeserializeException(f"Expected type bytes but got {type(v)}")
-        return t(v)
     elif hasattr(t, "__origin__") and (t.__origin__ is dict):
         t_args = t.__args__
         if len(t_args) != 2:

test/pycardano/test_plutus.py

@@ -9,13 +9,14 @@
 import pytest
 from cbor2 import CBORTag
 
-from pycardano import TransactionWitnessSet
+from pycardano import Address, Network, TransactionWitnessSet
 from pycardano.exception import DeserializeException
 from pycardano.plutus import (
     COST_MODELS,
     Datum,
     ExecutionUnits,
     PlutusData,
+    PlutusV2Script,
     RawPlutusData,
     Redeemer,
     RedeemerKey,
@@ -593,3 +594,15 @@ def test_empty_map_deser():
     serialized = witness.to_primitive()
     deserialized = TransactionWitnessSet.from_primitive(serialized)
     assert deserialized.redeemer == empty_map
+
+
+def test_load_plutus_script():
+    script = PlutusV2Script.load("test/resources/scriptV2.plutus")
+    assert (
+        script.to_cbor_hex()
+        == "581a581801000022232632498cd5ce2481064255524e542100120011"
+    )
+    assert (
+        Address(plutus_script_hash(script), network=Network.TESTNET).encode()
+        == "addr_test1wrmz3pjz4dmfxj0fc0a0eyw69tp6h7mpndzf9g3kttq9cqqqw47ym"
+    )

test/resources/scriptV2.plutus

@@ -0,0 +1,5 @@
+{
+    "type": "PlutusScriptV2",
+    "description": "",
+    "cborHex": "581a581801000022232632498cd5ce2481064255524e542100120011"
+}