mediator_class.py

Go to the documentation of this file.

 
 
 
from .mediator_user_classes import UserList
from .mediator_group_classes import GroupList
 
 
 
class UserGroupsContainer:
 
    
    def __init__(self) -> None:
        self.Users = UserList()
        self.Groups = GroupList()
 
    
 
 
 
 
 
 
class UserGroupMediator:
 
    
    def __init__(self) -> None:
        self._userGroupsContainer = UserGroupsContainer()
 
    
 
    
    
    def AddUser(self, name : str) -> None:
            self._userGroupsContainer.Users.AddUser(name)
 
 
    
    def RemoveUser(self, name : str) -> None:
        self.RemoveUserFromAllGroups(name)
        self._userGroupsContainer.Users.RemoveUser(name)
 
 
    
    def AddGroup(self, name : str) -> None:
        self._userGroupsContainer.Groups.AddGroup(name);
 
 
    
    def RemoveGroup(self, name : str) -> None:
        self._userGroupsContainer.Groups.RemoveGroup(name);
 
 
    
    def AddUserToGroup(self, userName : str, groupName : str) -> None:
        # As mediator, we must verify the user exists because the group
        # has no way to do this (groups have no knowledge of how users
        # are stored, by design).
        if not self._userGroupsContainer.Users.FindUser(userName):
            message = "User '{0}' does not exist.  Cannot add to group '{1}'!".format(userName, groupName)
            raise ValueError(message)
 
        foundGroup = self._userGroupsContainer.Groups.FindGroup(groupName)
        if not foundGroup:
            message = "Cannot add user '{0}' to group '{1}' as that group does not exist!".format(userName, groupName)
            raise ValueError(message)
 
        foundGroup.AddUser(userName)
 
 
    
    def RemoveUserFromGroup(self, userName : str, groupName : str) -> None:
        # As mediator, we must verify the user exists because the group
        # has no way to do this (groups have no knowledge of how users
        # are stored, by design).
        if not self._userGroupsContainer.Users.FindUser(userName):
            message = "User '{0}' does not exist.  Cannot remove from group '{1}'!".format(userName, groupName)
            raise ValueError(message)
 
        foundGroup = self._userGroupsContainer.Groups.FindGroup(groupName)
        if not foundGroup:
            message = "Cannot remove user '{0}' from group '{1}' as that group does not exist!".format(userName, groupName)
            raise ValueError(message)
 
        foundGroup.RemoveUser(userName);
 
 
    
    def RemoveUserFromAllGroups(self, userName : str) -> None:
        # As mediator, we must verify the user exists because the group
        # has no way to do this (groups have no knowledge of how users
        # are stored, by design).
        if not self._userGroupsContainer.Users.FindUser(userName):
            message = "User '{0}' does not exist.  Cannot remove from all groups'!".format(userName)
            raise ValueError(message)
 
        for groupName in self._userGroupsContainer.Groups.GroupNames:
            group = self._userGroupsContainer.Groups.FindGroup(groupName)
            if group:
                if group.ContainsUser(userName):
                    group.RemoveUser(userName)
 
 
    
    def IsUserInGroup(self, userName : str, groupName : str) -> bool:
        userInGroup = False
 
        if not self._userGroupsContainer.Users.FindUser(userName):
            message = "User '{0}' does not exist.  Cannot determine if user is in group '{1}'!".format(userName, groupName)
            raise ValueError(message)
 
        foundGroup = self._userGroupsContainer.Groups.FindGroup(groupName)
        if not foundGroup:
            message = "Cannot determine if user '{0}' is in group '{1}' as that group does not exist!".format(userName, groupName)
            raise ValueError(message)
 
        userInGroup = foundGroup.ContainsUser(userName)
 
        return userInGroup
 
 
    
    def GetGroupsWithUser(self, userName : str) -> list[str]:
        if not self._userGroupsContainer.Users.FindUser(userName):
            message = "User '{0}' does not exist.  Cannot get groups container user!".format(userName)
            raise ValueError(message)
 
        groupNames = []
        for groupName in self._userGroupsContainer.Groups.GroupNames:
            group = self._userGroupsContainer.Groups.FindGroup(groupName)
            if group:
                if group.ContainsUser(userName):
                    groupNames.append(groupName)
 
        return groupNames
 
 
    
    def GetUsersInGroup(self, groupName : str) -> list[str]:
        foundGroup = self._userGroupsContainer.Groups.FindGroup(groupName)
        if not foundGroup:
            message = "Cannot determine get users in group '{0}' as that group does not exist!".format(groupName)
            raise ValueError(message)
 
        return foundGroup.Users
 
 
    
    def GetAllGroups(self) -> list[str]:
        return self._userGroupsContainer.Groups.GroupNames
 
 
    
    def GetAllUsers(self) -> list[str]:
        return self._userGroupsContainer.Users.UserNames