17 lines
401 B
Python
17 lines
401 B
Python
|
from pymongo import AsyncMongoClient
|
||
|
|
|
||
|
|
|
||
|
|
class DBHandler:
|
||
|
|
@classmethod
|
||
|
|
def set_db(cls, uri, db_name):
|
||
|
|
cls.client = AsyncMongoClient(uri)
|
||
|
|
cls.database = cls.client[db_name]
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def get_cli(cls):
|
||
|
|
return cls.client if hasattr(cls, "client") else None
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def get_db(cls):
|
||
|
|
return cls.database if hasattr(cls, "database") else None
|