Create consensus.py

KOSASIH · Aug 6, 2024 · 0b0c24f · 0b0c24f
1 parent 77093f1
commit 0b0c24f
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/blockchain/consensus/consensus.py b/blockchain/consensus/consensus.py
@@ -0,0 +1,13 @@
+from .pos import ProofOfStake
+from .dpos import DelegatedProofOfStake
+
+def create_consensus(consensus_type: str, validators: list, block_time: int = 10):
+    if consensus_type == "pos":
+        return ProofOfStake(validators, block_time)
+    elif consensus_type == "dpos":
+        return DelegatedProofOfStake(validators, block_time)
+    else:
+        raise ValueError("Invalid consensus type")
+
+def run_consensus(consensus):
+    consensus.run_consensus()