ManageRoles.py

from pymongo import MongoClient

cluster = MongoClient(
    "mongodb+srv://Logan:lolo123@cluster0.igyjs.mongodb.net/ecole?retryWrites=true&w=majority")
db = cluster.testdb
collection = db.testcollection
roles = db.roles


class ManageRole:
    def __init__(self, username):
        self.username = username

    @property
    def user_role(self):
        user = collection.find({"username": self.username})
        for i in user:
            return i['role']

    def determine_permission(self, role_target):
        self.role_target = role_target
        role_target = roles.find({"Role": self.role_target})
        for i in role_target:
            return i["Permissions"]

    def show_status_user(self, user):
        self.user = user
        user_dic = collection.find({"username": self.user})
        for i in user_dic:
            return "The role of " + str(self.user) + " is " + str(i["role"])

    def manage_role(self, role, option, new_permission):
        self.role = role
        self.option = option
        self.new_permission = new_permission
        self.actual_permission = self.determine_permission(self.role)
        if self.user_role == 'Admin':
            if option == 'remove':
                if self.new_permission not in self.actual_permission:
                    return "Not found in the permission list", False
                else:
                    self.new_permission_list = self.actual_permission
                    self.new_permission_list.remove(self.new_permission)
                    roles.replace_one({'Role': self.role}, {'Role': self.role, 'Permissions': self.new_permission_list})
                    return True
            if option == 'append':
                if self.new_permission in self.actual_permission:
                    return "This permission is already in this role"
                else:
                    self.new_permission_list = self.actual_permission
                    self.new_permission_list.append(self.new_permission)
                    roles.replace_one({'Role': self.role}, {'Role': self.role, 'Permissions': self.new_permission_list})
                    return True

a = ManageRole("Zink")
print(a.show_status_user("Bedus"))