Implement AI workspace and mitigation workflow
This commit is contained in:
@@ -1,12 +1,14 @@
|
||||
from django.contrib.auth.mixins import LoginRequiredMixin
|
||||
from mixins.views import AbortSave, ObjectCreate, ObjectDelete, ObjectList, ObjectUpdate
|
||||
from django.db import IntegrityError
|
||||
from mixins.views import AbortSave, ObjectCreate, ObjectDelete, ObjectList, ObjectUpdate
|
||||
|
||||
from core.forms import MessageForm
|
||||
from core.models import Message
|
||||
from core.util import logs
|
||||
|
||||
log = logs.get_logger(__name__)
|
||||
|
||||
|
||||
class MessagePermissionMixin:
|
||||
def set_extra_args(self, user):
|
||||
self.extra_permission_args = {
|
||||
@@ -14,6 +16,7 @@ class MessagePermissionMixin:
|
||||
"session__pk": self.kwargs["session"],
|
||||
}
|
||||
|
||||
|
||||
class MessageList(LoginRequiredMixin, MessagePermissionMixin, ObjectList):
|
||||
list_template = "partials/message-list.html"
|
||||
model = Message
|
||||
@@ -52,6 +55,7 @@ class MessageCreate(LoginRequiredMixin, MessagePermissionMixin, ObjectCreate):
|
||||
log.error(f"Session {self.kwargs['session']} does not exist")
|
||||
raise AbortSave("session does not exist or you don't have access")
|
||||
|
||||
|
||||
class MessageUpdate(LoginRequiredMixin, MessagePermissionMixin, ObjectUpdate):
|
||||
model = Message
|
||||
form_class = MessageForm
|
||||
@@ -62,5 +66,3 @@ class MessageUpdate(LoginRequiredMixin, MessagePermissionMixin, ObjectUpdate):
|
||||
|
||||
class MessageDelete(LoginRequiredMixin, MessagePermissionMixin, ObjectDelete):
|
||||
model = Message
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user