From df4c4440958f930b064fd4b10d2a882c4281bb90 Mon Sep 17 00:00:00 2001 From: DIVYASREE S Date: Thu, 10 Oct 2024 21:40:33 +0530 Subject: [PATCH] Created and updated proof_of_stake.py --- blockchain/proof_of_stake.py | 41 ++++++++++++++++++++++++++++++++++++ 1 file changed, 41 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..568824921 --- /dev/null +++ b/blockchain/proof_of_stake.py @@ -0,0 +1,41 @@ +import random + +class Validator: + def __init__(self, name: str, stake: int) -> None: + """ + 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 + +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[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 = sum(v.stake for v in validators) + weighted_validators = [ + (v, v.stake / total_stake) for v in validators + ] + selected = random.choices( + [v[0] for v in weighted_validators], + weights=[v[1] for v in weighted_validators] + ) + return selected[0]