quay/endpoints/api/permission_models_interface.py

import sys
from abc import ABCMeta, abstractmethod
from collections import namedtuple

from six import add_metaclass


class SaveException(Exception):
    def __init__(self, other):
        self.traceback = sys.exc_info()
        super(SaveException, self).__init__(str(other))


class DeleteException(Exception):
    def __init__(self, other):
        self.traceback = sys.exc_info()
        super(DeleteException, self).__init__(str(other))


class Role(namedtuple("Role", ["role_name"])):
    def to_dict(self):
        return {
            "role": self.role_name,
        }


class UserPermission(
    namedtuple(
        "UserPermission",
        ["role_name", "username", "is_robot", "avatar", "is_org_member", "has_org",],
    )
):
    def to_dict(self):
        perm_dict = {
            "role": self.role_name,
            "name": self.username,
            "is_robot": self.is_robot,
            "avatar": self.avatar,
        }
        if self.has_org:
            perm_dict["is_org_member"] = self.is_org_member
        return perm_dict


class RobotPermission(
    namedtuple("RobotPermission", ["role_name", "username", "is_robot", "is_org_member",])
):
    def to_dict(self, user=None, team=None, org_members=None):
        return {
            "role": self.role_name,
            "name": self.username,
            "is_robot": True,
            "is_org_member": self.is_org_member,
        }


class TeamPermission(namedtuple("TeamPermission", ["role_name", "team_name", "avatar",])):
    def to_dict(self):
        return {
            "role": self.role_name,
            "name": self.team_name,
            "avatar": self.avatar,
        }


@add_metaclass(ABCMeta)
class PermissionDataInterface(object):
    """
    Data interface used by permissions API.
    """

    @abstractmethod
    def get_repo_permissions_by_user(self, namespace_name, repository_name):
        """
    
    Args:
      namespace_name: string
      repository_name: string

    Returns:
      list(UserPermission)
    """

    @abstractmethod
    def get_repo_roles(self, username, namespace_name, repository_name):
        """
    
    Args:
      username: string
      namespace_name: string
      repository_name: string

    Returns:
      list(Role) or None
    """

    @abstractmethod
    def get_repo_permission_for_user(self, username, namespace_name, repository_name):
        """
    
    Args:
      username: string
      namespace_name: string
      repository_name: string

    Returns:
      UserPermission
    """

    @abstractmethod
    def set_repo_permission_for_user(self, username, namespace_name, repository_name, role_name):
        """
    
    Args:
      username: string
      namespace_name: string
      repository_name: string
      role_name: string

    Returns:
      UserPermission
      
    Raises:
      SaveException
    """

    @abstractmethod
    def delete_repo_permission_for_user(self, username, namespace_name, repository_name):
        """
    
    Args:
      username: string
      namespace_name: string
      repository_name: string

    Returns:
      void
      
    Raises:
      DeleteException
    """

    @abstractmethod
    def get_repo_permissions_by_team(self, namespace_name, repository_name):
        """
    
    Args:
      namespace_name: string
      repository_name: string

    Returns:
      list(TeamPermission)
    """

    @abstractmethod
    def get_repo_role_for_team(self, team_name, namespace_name, repository_name):
        """
    
    Args:
      team_name: string
      namespace_name: string
      repository_name: string

    Returns:
      Role
    """

    @abstractmethod
    def set_repo_permission_for_team(self, team_name, namespace_name, repository_name, permission):
        """
    
    Args:
      team_name: string
      namespace_name: string
      repository_name: string
      permission: string

    Returns:
      TeamPermission
      
    Raises:
      SaveException
    """

    @abstractmethod
    def delete_repo_permission_for_team(self, team_name, namespace_name, repository_name):
        """
    
    Args:
      team_name: string
      namespace_name: string
      repository_name: string

    Returns:
      TeamPermission
      
    Raises:
      DeleteException
    """