blob: 72f63e66add4d947698f0df10de482d07596d58a (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
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()
|
