Added Setup/DB Handler
This commit is contained in:
46
core/database.py
Normal file
46
core/database.py
Normal file
@@ -0,0 +1,46 @@
|
||||
from pymongo import MongoClient
|
||||
from core.files import Data
|
||||
|
||||
client = MongoClient(Data('config').yaml_read()['mongo'])
|
||||
|
||||
class Servers:
|
||||
def __init__(self, server=None):
|
||||
self.server = server
|
||||
self.col = client["BytesBump"]["servers"]
|
||||
|
||||
def get(self):
|
||||
return self.col.find_one({"_id":self.server})
|
||||
|
||||
def get_all(self):
|
||||
return self.col.find({})
|
||||
|
||||
"""
|
||||
Add a server to the database.
|
||||
Expected params: Desc, Color, Listing, Invite
|
||||
"""
|
||||
|
||||
def add(self, **params):
|
||||
kwargs['_id'] = self.server
|
||||
self.col.insert_one(params)
|
||||
|
||||
"""
|
||||
Update a server in the database.
|
||||
"""
|
||||
|
||||
def update(self, **checks):
|
||||
self.col.update_one({
|
||||
"_id": self.server
|
||||
}, {
|
||||
"$set": params
|
||||
})
|
||||
|
||||
"""
|
||||
Remove a server from the database.
|
||||
"""
|
||||
|
||||
def delete(self, **checks):
|
||||
if self.server:
|
||||
self.col.delete_one({'_id': self.server})
|
||||
else:
|
||||
self.col.delete_one(checks)
|
||||
|
||||
Reference in New Issue
Block a user