From 69bcc9822c82ee6d726c98deafbdd0a20527a55a Mon Sep 17 00:00:00 2001 From: A Farzat Date: Thu, 21 Aug 2025 09:41:56 +0300 Subject: Move components to root directory Some components might be shared with other applications such as the data analyser later on. --- components/database.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) create mode 100644 components/database.py (limited to 'components/database.py') diff --git a/components/database.py b/components/database.py new file mode 100644 index 0000000..72f63e6 --- /dev/null +++ b/components/database.py @@ -0,0 +1,16 @@ +import atexit +from typing import Any, Dict +from pymongo import MongoClient +from pymongo.database import Database +from pymongo.collection import Collection +from components.subscriptions.typing import SubsDict +from components.users.typing import UserDict + +client: MongoClient[Any] = MongoClient("mongodb://localhost", tz_aware=True) +database: Database[Any] = client.get_database("youtube") +subscriptions: Collection[SubsDict] = database.get_collection("subscriptions") +users: Collection[UserDict] = database.get_collection("users") + +@atexit.register +def _cleanup() -> None: + client.close() -- cgit v1.2.3-70-g09d2