From d770fa5bf55a8cfeb632301fdc3d09ad484efa36 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 10 Oct 2024 15:54:30 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- blockchain/proof_of_stake.py | 20 ++++++++++++-------- 1 file changed, 12 insertions(+), 8 deletions(-) diff --git a/blockchain/proof_of_stake.py b/blockchain/proof_of_stake.py index 946d7218e..78dc45c6f 100644 --- a/blockchain/proof_of_stake.py +++ b/blockchain/proof_of_stake.py @@ -1,18 +1,20 @@ 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: str, stake: int): """ 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. @@ -20,18 +22,19 @@ class Validator: 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) @@ -40,6 +43,7 @@ def choose_validator(validators: List[Validator]) -> Validator: """ 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]) + selected = random.choices( + [v[0] for v in weighted_validators], weights=[v[1] for v in weighted_validators] + ) return selected[0]