31 lines
828 B
Python
Executable File
31 lines
828 B
Python
Executable File
import pymongo
|
|
from config import config
|
|
|
|
|
|
class DB:
|
|
client = pymongo.MongoClient(f"mongodb://mongo:{config.DB_PORT}/")
|
|
db = client["rc_coffee"]
|
|
|
|
def __init__(self, collection):
|
|
if collection not in self.db.list_collection_names():
|
|
self.db.create_collection(collection)
|
|
self.collection = self.db[collection]
|
|
|
|
def insert(self, data):
|
|
self.collection.insert_one(data)
|
|
|
|
def find(self, query):
|
|
return list(self.collection.find(query))
|
|
|
|
def find_one(self, query):
|
|
return self.collection.find_one(query)
|
|
|
|
def update(self, query, data):
|
|
self.collection.update_one(query, {"$set": data})
|
|
|
|
def exist(self, query):
|
|
return self.collection.find_one(query) is not None
|
|
|
|
def delete(self, query):
|
|
self.collection.delete_one(query)
|