evilchili/random-sets
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

adding ability to set frequency on a DataSourceSet

Browse Source
This commit is contained in:
evilchili 2023-12-24 12:15:48 -08:00
parent e751702610
commit 51d9aa02b4
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
random_sets/sets.py
View File

@ -38,8 +38,15 @@ class DataSourceSet(WeightedSet):
def __init__(self, source: Path): def __init__(self, source: Path):
self.source = DataSource(source.read_text()) self.source = DataSource(source.read_text())
self._populate()
def _populate(self):
super().__init__(*[(key, value) for key, value in self.source.frequencies.items()]) super().__init__(*[(key, value) for key, value in self.source.frequencies.items()])
def set_frequency(self, frequency: str):
self.source.set_frequency(frequency)
self._populate()
def random(self): def random(self):
random_key = super().random() random_key = super().random()
return self.source.as_dict()[random_key] return self.source.as_dict()[random_key]