1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
from scripts.database.mongo import mongo_database
mongo_collection = mongo_database.get_collection("users")
def user_exists(user):
try:
res = mongo_collection.find_one({"name": user}, {"password": 1, "isAiSubscriber": 1})
if res is None:
return {"success": True, "user": None}
else:
# Fix this - there is a better way
return {"success": True, "user": user, "password": res["password"], "isAiSubscriber": res["isAiSubscriber"]}
except:
return {"success": False, "error": "Request error, try again"}
def add_user(user, password):
try:
res = mongo_collection.insert_one({"name": user, "password": password})
return {"success": True, "user": user}
except:
print("error")
def delete_single_user(user):
try:
mongo_collection.delete_one({"name": user})
return {"success": True, "user": user}
except:
return {"success": False, "user": user}
def update_user(user, user_prefs):
print(user, user_prefs, "in user handler")
try:
res = mongo_collection.update_one({"name": user}, {"$set": user_prefs})
return {"success": True, "message": "user preferences successfully updated"}
except Exception as e:
return {"success": False, "message": str(e)}
|
