From f467cec98a1678f7eabd99d1f8f5fc8df5fb9578 Mon Sep 17 00:00:00 2001
From: DIVYASREE S <divyasrees.cse.41@gmail.com>
Date: Thu, 10 Oct 2024 21:12:51 +0530
Subject: [PATCH] Created proof_of_stake.py

---
 blockchain/proof_of_stake.py | 50 ++++++++++++++++++++++++++++++++++++
 1 file changed, 50 insertions(+)
 create mode 100644 blockchain/proof_of_stake.py

diff --git a/blockchain/proof_of_stake.py b/blockchain/proof_of_stake.py
new file mode 100644
index 000000000..8cc451112
--- /dev/null
+++ b/blockchain/proof_of_stake.py
@@ -0,0 +1,50 @@
+import random
+
+class Validator:
+    """
+    Represents a validator in a Proof of Stake system.
+
+    Attributes:
+        name (str): The name of the validator.
+        stake (int): The amount of stake (coins) the validator holds.
+    """
+    def __init__(self, name, stake):
+        """
+        Initializes a new validator with a given name and stake.
+        
+        Args:
+            name (str): The name of the validator.
+            stake (int): The amount of stake the validator has.
+        """
+        self.name = name
+        self.stake = stake  # Stake defines how much weight the validator has.
+
+def choose_validator(validators):
+    """
+    Selects a validator to create the next block based on the weight of their stake.
+    
+    The higher the stake, the greater the chance to be selected.
+    
+    Args:
+        validators (list): A list of Validator objects.
+    
+    Returns:
+        Validator: The selected validator based on weighted random selection.
+    """
+    # Total stake of all validators
+    total_stake = sum(v.stake for v in validators)
+    
+    # Create a list of validators with weights (probability of being chosen)
+    weighted_validators = [(v, v.stake / total_stake) for v in validators]
+    
+    # Randomly select a validator based on their stake weight
+    selected = random.choices([v[0] for v in weighted_validators],
+                              weights=[v[1] for v in weighted_validators])
+    return selected[0]
+
+# Example of validators with different stakes
+validators = [Validator("Alice", 50), Validator("Bob", 30), Validator("Charlie", 20)]
+
+# Select a validator based on their stake
+chosen_validator = choose_validator(validators)
+print(f"Chosen validator: {chosen_validator.name}")