from drf_spectacular.utils import extend_schema
from rest_framework import status
from rest_framework.response import Response
from rest_framework.views import APIView
from anthias_app.models import Asset
from api.helpers import (
AssetCreationException,
get_active_asset_ids,
save_active_assets_ordering,
)
from api.serializers import (
AssetSerializer,
UpdateAssetSerializer,
)
from api.serializers.v1_2 import CreateAssetSerializerV1_2
from api.views.mixins import DeleteAssetViewMixin
from lib.auth import authorized
class AssetListViewV1_2(APIView):
serializer_class = AssetSerializer
@extend_schema(
summary='List assets',
responses={
200: AssetSerializer(many=True)
}
)
@authorized
def get(self, request):
queryset = Asset.objects.all()
serializer = self.serializer_class(queryset, many=True)
return Response(serializer.data)
@extend_schema(
summary='Create asset',
request=CreateAssetSerializerV1_2,
responses={
201: AssetSerializer
}
)
@authorized
def post(self, request):
try:
serializer = CreateAssetSerializerV1_2(
data=request.data, unique_name=True)
if not serializer.is_valid():
raise AssetCreationException(serializer.errors)
except AssetCreationException as error:
return Response(error.errors, status=status.HTTP_400_BAD_REQUEST)
active_asset_ids = get_active_asset_ids()
asset = Asset.objects.create(**serializer.data)
if asset.is_active():
active_asset_ids.insert(asset.play_order, asset.asset_id)
save_active_assets_ordering(active_asset_ids)
asset.refresh_from_db()
return Response(
AssetSerializer(asset).data,
status=status.HTTP_201_CREATED,
)
class AssetViewV1_2(APIView, DeleteAssetViewMixin):
serializer_class = AssetSerializer
@extend_schema(summary='Get asset')
@authorized
def get(self, request, asset_id):
asset = Asset.objects.get(asset_id=asset_id)
serializer = self.serializer_class(asset)
return Response(serializer.data)
def update(self, request, asset_id, partial=False):
asset = Asset.objects.get(asset_id=asset_id)
serializer = UpdateAssetSerializer(
asset, data=request.data, partial=partial)
if serializer.is_valid():
serializer.save()
else:
return Response(
serializer.errors, status=status.HTTP_400_BAD_REQUEST)
active_asset_ids = get_active_asset_ids()
asset.refresh_from_db()
try:
active_asset_ids.remove(asset.asset_id)
except ValueError:
pass
if asset.is_active():
active_asset_ids.insert(asset.play_order, asset.asset_id)
save_active_assets_ordering(active_asset_ids)
asset.refresh_from_db()
return Response(AssetSerializer(asset).data)
@extend_schema(
summary='Update asset',
request=UpdateAssetSerializer,
responses={
200: AssetSerializer
}
)
@authorized
def patch(self, request, asset_id):
return self.update(request, asset_id, partial=True)
@extend_schema(
summary='Update asset',
request=UpdateAssetSerializer,
responses={
200: AssetSerializer
}
)
@authorized
def put(self, request, asset_id):
return self.update(request, asset_id, partial=False)
