37 lines
938 B
Python
37 lines
938 B
Python
from pymongo import MongoClient
|
|
|
|
MONGODB_CONFIG = {
|
|
'host': 'localhost',
|
|
'port': 27017,
|
|
'dbname': 'local',
|
|
'collection': 'messenger'
|
|
}
|
|
|
|
|
|
class MongoDbHandler:
|
|
def __init__(self, dbname, collection, host, port):
|
|
self.host = host
|
|
self.port = port
|
|
self.dbname = dbname
|
|
self.collection_name = collection
|
|
self.client = None
|
|
self.collection = None
|
|
|
|
def connect(self):
|
|
try:
|
|
self.client = MongoClient(self.host, self.port)
|
|
db = self.client[self.dbname]
|
|
self.collection = db[self.collection_name]
|
|
print("Connected to MongoDB")
|
|
except Exception as e:
|
|
self.collection = None
|
|
|
|
def close_client(self):
|
|
if self.client:
|
|
self.client.close()
|
|
self.collection = None
|
|
print("MongoDB client closed.")
|
|
|
|
def get_collection(self):
|
|
return self.collection
|