61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
from enum import Enum
|
|
from sqlalchemy import Index, ForeignKey, Column, String
|
|
from sqlalchemy.dialects.postgresql import UUID
|
|
from sqlalchemy.orm import relationship
|
|
|
|
from atst.models import Base, types, mixins
|
|
|
|
|
|
class CSPRole(Enum):
|
|
NONSENSE_ROLE = "nonsense_role"
|
|
|
|
|
|
class EnvironmentRole(Base, mixins.TimestampsMixin, mixins.AuditableMixin):
|
|
__tablename__ = "environment_roles"
|
|
|
|
id = types.Id()
|
|
environment_id = Column(
|
|
UUID(as_uuid=True), ForeignKey("environments.id"), nullable=False
|
|
)
|
|
environment = relationship("Environment", backref="roles")
|
|
|
|
role = Column(String())
|
|
|
|
user_id = Column(UUID(as_uuid=True), ForeignKey("users.id"), nullable=False)
|
|
user = relationship("User", backref="environment_roles")
|
|
|
|
def __repr__(self):
|
|
return "<EnvironmentRole(role='{}', user='{}', environment='{}', id='{}')>".format(
|
|
self.role, self.user.full_name, self.environment.name, self.id
|
|
)
|
|
|
|
@property
|
|
def history(self):
|
|
return self.get_changes()
|
|
|
|
@property
|
|
def displayname(self):
|
|
return self.role
|
|
|
|
@property
|
|
def event_details(self):
|
|
return {
|
|
"updated_user_name": self.user.displayname,
|
|
"updated_user_id": str(self.user_id),
|
|
"role": self.role,
|
|
"environment": self.environment.displayname,
|
|
"environment_id": str(self.environment_id),
|
|
"project": self.environment.project.name,
|
|
"project_id": str(self.environment.project_id),
|
|
"workspace": self.environment.project.workspace.name,
|
|
"workspace_id": str(self.environment.project.workspace.id),
|
|
}
|
|
|
|
|
|
Index(
|
|
"environments_role_user_environment",
|
|
EnvironmentRole.user_id,
|
|
EnvironmentRole.environment_id,
|
|
unique=True,
|
|
)
|