47 lines
1.0 KiB
Python
47 lines
1.0 KiB
Python
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, WebhookID
|
|
"""
|
|
|
|
def add(self, **params):
|
|
params['_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)
|
|
|