scripts/users/handle_users.py


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

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})
        if res is None:
            return {"success": True, "user": None}
        else:
            return {"success": True, "user": user, "password": res["password"]}
    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}