Week08

src/week08/__init__.py

@@ -0,0 +1,3 @@
+from src.utils.paths import get_week_paths
+
+assets_dir, homework_dir, lecture_dir = get_week_paths(__file__)

src/week08/lecture/staking.py

@@ -0,0 +1,43 @@
+from opshin.prelude import *
+
+
+def equal(a: Union[StakingHash, StakingPtr], b: Union[StakingHash, StakingPtr]) -> bool:
+    result = False
+    if isinstance(b, StakingHash):
+        if isinstance(a, StakingHash):
+            result = b == a
+    elif isinstance(b, StakingPtr):
+        if isinstance(a, StakingPtr):
+            result = b == a
+    return result
+
+
+def get_amount(context: ScriptContext, purpose: Rewarding) -> int:
+    amount = 0
+    found = False
+    for staking_cred, a in context.tx_info.wdrl.items():
+        if equal(staking_cred, purpose.staking_credential):
+            found = True
+            amount = a
+    assert found, "withdrawal not found"
+    return amount
+
+
+def paid_to_address(addr: Address, context: ScriptContext) -> int:
+    paid_amount = 0
+    for output in context.tx_info.outputs:
+        if output.address == addr:
+            paid_amount += output.value.get(b"", {b"": 0}).get(b"", 0)
+    return paid_amount
+
+
+def validator(addr: Address, _: None, context: ScriptContext):
+    purpose = context.purpose
+    if isinstance(purpose, Certifying):
+        print("certifying")
+    elif isinstance(purpose, Rewarding):
+        amount = get_amount(context, purpose)
+        paid_amount = paid_to_address(addr, context)
+        assert 2 * paid_amount >= amount, "insufficient reward sharing"
+    else:
+        assert False, "script purpose must be certifying or rewarding"

src/week08/tests/test_lecture.py

@@ -0,0 +1,18 @@
+import pytest
+from opshin import compiler
+
+from src.week08 import lecture_dir
+
+python_files = [
+    "staking.py",
+]
+script_paths = [str(lecture_dir.joinpath(f)) for f in python_files]
+
+
+@pytest.mark.parametrize("path", script_paths)
+def test_lecture_compile(path):
+    with open(path, "r") as f:
+        source_code = f.read()
+    source_ast = compiler.parse(source_code)
+    code = compiler.compile(source_ast)
+    print(code.dumps())