summaryrefslogtreecommitdiff
path: root/data_collector/utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'data_collector/utils.py')
-rw-r--r--data_collector/utils.py13
1 files changed, 13 insertions, 0 deletions
diff --git a/data_collector/utils.py b/data_collector/utils.py
new file mode 100644
index 0000000..6ca5c69
--- /dev/null
+++ b/data_collector/utils.py
@@ -0,0 +1,13 @@
+from datetime import datetime, timedelta, UTC
+
+from pymongo.collection import Collection
+
+from components.subscriptions.main import Subscription
+from components.subscriptions.typing import SubsDict
+
+def collect_data(subs_collection: Collection[SubsDict]) -> None:
+ for sub_dict in subs_collection.find():
+ sub = Subscription(**sub_dict)
+ sub._collection = subs_collection
+ if datetime.now(tz=UTC) - sub.last_fetch > timedelta(seconds=sub.time_between_fetches):
+ sub.fetch()