api done
This commit is contained in:
@@ -0,0 +1,4 @@
|
||||
from .connection import MONGODB_CONFIG, MongoDbHandler
|
||||
|
||||
mongo_db_handler = MongoDbHandler(**MONGODB_CONFIG)
|
||||
mongo_db_handler.connect()
|
||||
Binary file not shown.
Binary file not shown.
@@ -0,0 +1,36 @@
|
||||
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
|
||||
Reference in New Issue
Block a user