125 lines
3.6 KiB
Python
125 lines
3.6 KiB
Python
from drf_spectacular.utils import extend_schema, extend_schema_view, OpenApiResponse
|
|
|
|
from api.viewsets.common import ModelViewSet
|
|
|
|
# This import only exists so that the migrations can be created
|
|
from project_management.models.project_history import ProjectAuditHistory # pylint: disable=W0611:unused-import
|
|
from project_management.serializers.project import ( # pylint: disable=W0611:unused-import
|
|
Project,
|
|
ProjectImportSerializer,
|
|
ProjectModelSerializer,
|
|
ProjectViewSerializer,
|
|
Organization,
|
|
)
|
|
|
|
|
|
|
|
@extend_schema_view(
|
|
create=extend_schema(
|
|
summary = 'Create a cluster',
|
|
description='',
|
|
responses = {
|
|
201: OpenApiResponse(description='Device created', response=ProjectViewSerializer),
|
|
400: OpenApiResponse(description='Validation failed.'),
|
|
403: OpenApiResponse(description='User is missing create permissions'),
|
|
}
|
|
),
|
|
destroy = extend_schema(
|
|
summary = 'Delete a cluster',
|
|
description = '',
|
|
responses = {
|
|
204: OpenApiResponse(description=''),
|
|
403: OpenApiResponse(description='User is missing delete permissions'),
|
|
}
|
|
),
|
|
list = extend_schema(
|
|
summary = 'Fetch all clusters',
|
|
description='',
|
|
responses = {
|
|
200: OpenApiResponse(description='', response=ProjectViewSerializer),
|
|
403: OpenApiResponse(description='User is missing view permissions'),
|
|
}
|
|
),
|
|
retrieve = extend_schema(
|
|
summary = 'Fetch a single cluster',
|
|
description='',
|
|
responses = {
|
|
200: OpenApiResponse(description='', response=ProjectViewSerializer),
|
|
403: OpenApiResponse(description='User is missing view permissions'),
|
|
}
|
|
),
|
|
update = extend_schema(exclude = True),
|
|
partial_update = extend_schema(
|
|
summary = 'Update a cluster',
|
|
description = '',
|
|
responses = {
|
|
200: OpenApiResponse(description='', response=ProjectViewSerializer),
|
|
403: OpenApiResponse(description='User is missing change permissions'),
|
|
}
|
|
),
|
|
)
|
|
class ViewSet( ModelViewSet ):
|
|
|
|
filterset_fields = [
|
|
'organization',
|
|
'external_system',
|
|
'priority',
|
|
'project_type',
|
|
'state',
|
|
]
|
|
|
|
search_fields = [
|
|
'name',
|
|
'description',
|
|
]
|
|
|
|
model = Project
|
|
|
|
view_description = 'Physical Devices'
|
|
|
|
|
|
def get_serializer_class(self):
|
|
|
|
organization = None
|
|
|
|
if 'organization' in self.request.data:
|
|
|
|
organization = int(self.request.data['organization'])
|
|
|
|
organization = Organization.objects.get( pk = organization )
|
|
|
|
elif self.queryset:
|
|
|
|
if list(self.queryset) == 1:
|
|
|
|
obj = list(self.queryset)[0]
|
|
|
|
organization = obj.organization
|
|
|
|
|
|
if organization:
|
|
|
|
if self.request.tenancy.has_organization_permission(
|
|
organization = organization,
|
|
permissions_required = 'project_management.import_project'
|
|
) or self.request.user.is_superuser:
|
|
|
|
self.serializer_class = globals()[str( self.model._meta.verbose_name) + 'ImportSerializer']
|
|
|
|
return self.serializer_class
|
|
|
|
|
|
if (
|
|
self.action == 'list'
|
|
or self.action == 'retrieve'
|
|
):
|
|
|
|
self.serializer_class = globals()[str( self.model._meta.verbose_name) + 'ViewSerializer']
|
|
|
|
|
|
else:
|
|
|
|
self.serializer_class = globals()[str( self.model._meta.verbose_name) + 'ModelSerializer']
|
|
|
|
return self.serializer_class
|