From 28df0f3c9b0af3b5fd1b7737a575668c3ad786e2 Mon Sep 17 00:00:00 2001 From: DIVYASREE S Date: Thu, 10 Oct 2024 21:17:13 +0530 Subject: [PATCH] Updated proof_of_stake.py --- blockchain/proof_of_stake.py | 29 ++++++++++++----------------- 1 file changed, 12 insertions(+), 17 deletions(-) diff --git a/blockchain/proof_of_stake.py b/blockchain/proof_of_stake.py index 8cc451112..946d7218e 100644 --- a/blockchain/proof_of_stake.py +++ b/blockchain/proof_of_stake.py @@ -1,14 +1,15 @@ import random +from typing import List 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): + def __init__(self, name: str, stake: int): """ Initializes a new validator with a given name and stake. @@ -17,34 +18,28 @@ class Validator: stake (int): The amount of stake the validator has. """ self.name = name - self.stake = stake # Stake defines how much weight the validator has. + self.stake = stake -def choose_validator(validators): +def choose_validator(validators: List[Validator]) -> Validator: """ 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. + validators (List[Validator]): A list of Validator objects. Returns: Validator: The selected validator based on weighted random selection. + + Example: + >>> validators = [Validator("Alice", 50), Validator("Bob", 30), Validator("Charlie", 20)] + >>> chosen = choose_validator(validators) + >>> isinstance(chosen, Validator) + True """ - # 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}")