krosh/rc-tgbot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

some fixs

Browse Source
This commit is contained in:
krosh
2025-03-01 01:49:33 +03:00
parent 2368b5df8c
commit a688a19b4e
13 changed files with 100 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/db.py
Regular → Executable
+4 -2
View File
@@ -3,17 +3,19 @@ from config import config
class DB:
client = pymongo.MongoClient(f"mongodb://localhost:{config.DB_PORT}/")
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 self.collection.find(query)
return list(self.collection.find(query))
def find_one(self, query):
return self.collection.find_one(query)