From 51d9aa02b496845d9bc724eb1992fa1fc951ba18 Mon Sep 17 00:00:00 2001
From: evilchili <evilchili@gmail.com>
Date: Sun, 24 Dec 2023 12:15:48 -0800
Subject: [PATCH] adding ability to set frequency on a DataSourceSet

---
 random_sets/sets.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/random_sets/sets.py b/random_sets/sets.py
index 0dc2cea..d17bcfa 100644
--- a/random_sets/sets.py
+++ b/random_sets/sets.py
@@ -38,8 +38,15 @@ class DataSourceSet(WeightedSet):
 
     def __init__(self, source: Path):
         self.source = DataSource(source.read_text())
+        self._populate()
+
+    def _populate(self):
         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):
         random_key = super().random()
         return self.source.as_dict()[random_key]