Leahnaya/DeathGarden_API_Rebirth
0
0
Fork 0
mirror of https://github.com/wolfswolke/DeathGarden_API_Rebirth.git synced 2026-10-26 08:09:45 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fixed message System

Browse Source
This commit is contained in:
ZKWolf
2023-11-24 00:28:03 +01:00
parent 3379a868cc
commit c2106b2d73
2 changed files with 22 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/logic/mongodb_handler.py
View File

@@ -138,17 +138,23 @@ class Mongo:
logger.graylog_logger(level="error", handler="mongo_get_data_with_list", message=e)
return None
def write_data_with_list(self, steamid, items_dict):
def write_data_with_list(self, login, login_steam, items_dict):
try:
steam_id = str(steamid)
client = pymongo.MongoClient(self.dyn_server)
dyn_client_db = client[self.dyn_db]
dyn_collection = dyn_client_db[self.dyn_collection]
existing_document = dyn_collection.find_one({'steamid': steamid})
if login_steam:
steam_id = str(login)
existing_document = dyn_collection.find_one({'steamid': steam_id})
else:
user_id = str(login)
existing_document = dyn_collection.find_one({"userId": user_id})
if existing_document:
update_query = {'$set': items_dict}
dyn_collection.update_one({'steamid': steamid}, update_query)
if login_steam:
dyn_collection.update_one({'steamid': steam_id}, update_query)
else:
dyn_collection.update_one({'userId': user_id}, update_query)
client.close()
return {"status": "success", "message": "Data updated"}
else: