diff options
| author | 2020-09-22 12:51:16 -0500 | |
|---|---|---|
| committer | 2020-09-22 12:51:16 -0500 | |
| commit | cf0c7c2289bc06ce9ce29128ecb6d6c55f695431 (patch) | |
| tree | 38c94578633b5cdde188845c8be4f109cfdca477 /bot/utils | |
| parent | Changed Wikipedia to WikipediaSearch (diff) | |
| parent | Merge pull request #430 from Anubhav1603/master (diff) | |
Merge branch 'master' into bug_wiki
Diffstat (limited to 'bot/utils')
| -rw-r--r-- | bot/utils/randomization.py | 23 | 
1 files changed, 23 insertions, 0 deletions
| diff --git a/bot/utils/randomization.py b/bot/utils/randomization.py new file mode 100644 index 00000000..8f47679a --- /dev/null +++ b/bot/utils/randomization.py @@ -0,0 +1,23 @@ +import itertools +import random +import typing as t + + +class RandomCycle: +    """ +    Cycles through elements from a randomly shuffled iterable, repeating indefinitely. + +    The iterable is reshuffled after each full cycle. +    """ + +    def __init__(self, iterable: t.Iterable) -> None: +        self.iterable = list(iterable) +        self.index = itertools.cycle(range(len(iterable))) + +    def __next__(self) -> t.Any: +        idx = next(self.index) + +        if idx == 0: +            random.shuffle(self.iterable) + +        return self.iterable[idx] | 
