Implement asset upload API and utility functions
- Add Asset and CardLayer model updates - Create asset upload API endpoints - Add AssetUploadDialog component - Create card layout algorithms - Implement symmetry generation utils - Add template configurations
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
from django.db import models
|
||||
import uuid
|
||||
|
||||
|
||||
class Project(models.Model):
|
||||
"""项目配置模型"""
|
||||
id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
|
||||
@@ -24,8 +25,11 @@ class Project(models.Model):
|
||||
|
||||
class Asset(models.Model):
|
||||
"""项目素材模型"""
|
||||
project = models.ForeignKey(Project, on_delete=models.CASCADE, related_name='assets')
|
||||
asset_type = models.CharField(max_length=20) # 'suit_symbol', 'face_card', 'joker', 'back', 'border'
|
||||
asset_key = models.CharField(max_length=50) # 如 'spade', 'heart-J', 'big_joker'
|
||||
file_path = models.CharField(max_length=255) # 相对于media目录
|
||||
file_name = models.CharField(max_length=100)
|
||||
width = models.IntegerField(null=True)
|
||||
height = models.IntegerField(null=True)
|
||||
uploaded_at = models.DateTimeField(auto_now_add=True)
|
||||
@@ -33,9 +37,13 @@ class Asset(models.Model):
|
||||
def __str__(self):
|
||||
return f"{self.asset_type}:{self.asset_key}"
|
||||
|
||||
class Meta:
|
||||
ordering = ['-uploaded_at']
|
||||
|
||||
|
||||
class CardLayer(models.Model):
|
||||
"""牌面图层配置模型"""
|
||||
project = models.ForeignKey(Project, on_delete=models.CASCADE, related_name='layers')
|
||||
card_type = models.CharField(max_length=20) # 'number', 'face', 'joker'
|
||||
card_key = models.CharField(max_length=30) # 'spade-A', 'heart-K', 'big_joker'
|
||||
layer_name = models.CharField(max_length=50)
|
||||
@@ -47,7 +55,6 @@ class CardLayer(models.Model):
|
||||
|
||||
# 图层属性(JSON存储)
|
||||
properties = models.JSONField(default=dict)
|
||||
file_ref = models.ForeignKey(Asset, on_delete=models.SET_NULL, null=True, related_name='layers')
|
||||
|
||||
def __str__(self):
|
||||
return f"{self.card_key}-{self.layer_name}"
|
||||
|
||||
@@ -1,4 +1,5 @@
|
||||
from rest_framework import serializers
|
||||
from django.conf import settings
|
||||
from .models import Project, Asset, CardLayer
|
||||
|
||||
|
||||
@@ -9,10 +10,20 @@ class ProjectSerializer(serializers.ModelSerializer):
|
||||
|
||||
|
||||
class AssetSerializer(serializers.ModelSerializer):
|
||||
file_url = serializers.SerializerMethodField()
|
||||
|
||||
class Meta:
|
||||
model = Asset
|
||||
fields = '__all__'
|
||||
|
||||
def get_file_url(self, obj):
|
||||
if obj.file_path:
|
||||
request = self.context.get('request')
|
||||
if request:
|
||||
return request.build_absolute_uri(f'{settings.MEDIA_URL}{obj.file_path}')
|
||||
return f'{settings.MEDIA_URL}{obj.file_path}'
|
||||
return None
|
||||
|
||||
|
||||
class CardLayerSerializer(serializers.ModelSerializer):
|
||||
class Meta:
|
||||
|
||||
@@ -1,7 +1,9 @@
|
||||
from django.urls import path
|
||||
from .views import project_list, project_detail
|
||||
from .views import project_list, project_detail, asset_list, asset_detail
|
||||
|
||||
urlpatterns = [
|
||||
path('', project_list, name='project-list'),
|
||||
path('<str:pk>/', project_detail, name='project-detail'),
|
||||
path('<str:project_pk>/assets/', asset_list, name='asset-list'),
|
||||
path('<str:project_pk>/assets/<str:asset_pk>/', asset_detail, name='asset-detail'),
|
||||
]
|
||||
|
||||
@@ -1,8 +1,12 @@
|
||||
from rest_framework.decorators import api_view
|
||||
from rest_framework.response import Response
|
||||
from rest_framework import status
|
||||
from .models import Project
|
||||
from .serializers import ProjectSerializer, ProjectDetailSerializer
|
||||
from django.core.files.storage import default_storage
|
||||
from django.conf import settings
|
||||
from PIL import Image
|
||||
import os
|
||||
from .models import Project, Asset
|
||||
from .serializers import ProjectSerializer, ProjectDetailSerializer, AssetSerializer
|
||||
|
||||
|
||||
@api_view(['GET', 'POST'])
|
||||
@@ -42,4 +46,81 @@ def project_detail(request, pk):
|
||||
|
||||
elif request.method == 'DELETE':
|
||||
project.delete()
|
||||
return Response(status=status.HTTP_204_NO_CONTENT)
|
||||
|
||||
|
||||
@api_view(['GET', 'POST'])
|
||||
def asset_list(request, project_pk):
|
||||
"""获取项目素材列表或上传新素材"""
|
||||
try:
|
||||
project = Project.objects.get(pk=project_pk)
|
||||
except Project.DoesNotExist:
|
||||
return Response({'error': 'Project not found'}, status=status.HTTP_404_NOT_FOUND)
|
||||
|
||||
if request.method == 'GET':
|
||||
assets = project.assets.all()
|
||||
serializer = AssetSerializer(assets, many=True)
|
||||
return Response(serializer.data)
|
||||
|
||||
elif request.method == 'POST':
|
||||
if 'file' not in request.FILES:
|
||||
return Response({'error': 'No file provided'}, status=status.HTTP_400_BAD_REQUEST)
|
||||
|
||||
file = request.FILES['file']
|
||||
asset_type = request.POST.get('asset_type', 'unknown')
|
||||
asset_key = request.POST.get('asset_key', 'unknown')
|
||||
|
||||
# 创建项目素材目录
|
||||
project_media_dir = os.path.join('projects', str(project.id), asset_type)
|
||||
full_dir = os.path.join(settings.MEDIA_ROOT, project_media_dir)
|
||||
os.makedirs(full_dir, exist_ok=True)
|
||||
|
||||
# 保存文件
|
||||
file_name = f"{asset_key}_{file.name}"
|
||||
file_path = os.path.join(project_media_dir, file_name)
|
||||
saved_path = default_storage.save(file_path, file)
|
||||
|
||||
# 获取图片尺寸
|
||||
try:
|
||||
img = Image.open(file)
|
||||
width, height = img.size
|
||||
except:
|
||||
width, height = None, None
|
||||
|
||||
# 创建Asset记录
|
||||
asset = Asset.objects.create(
|
||||
project=project,
|
||||
asset_type=asset_type,
|
||||
asset_key=asset_key,
|
||||
file_path=saved_path,
|
||||
file_name=file_name,
|
||||
width=width,
|
||||
height=height
|
||||
)
|
||||
|
||||
serializer = AssetSerializer(asset)
|
||||
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
||||
|
||||
|
||||
@api_view(['GET', 'DELETE'])
|
||||
def asset_detail(request, project_pk, asset_pk):
|
||||
"""获取或删除单个素材"""
|
||||
try:
|
||||
project = Project.objects.get(pk=project_pk)
|
||||
asset = project.assets.get(pk=asset_pk)
|
||||
except (Project.DoesNotExist, Asset.DoesNotExist):
|
||||
return Response({'error': 'Asset not found'}, status=status.HTTP_404_NOT_FOUND)
|
||||
|
||||
if request.method == 'GET':
|
||||
serializer = AssetSerializer(asset)
|
||||
return Response(serializer.data)
|
||||
|
||||
elif request.method == 'DELETE':
|
||||
# 删除文件
|
||||
if asset.file_path:
|
||||
file_full_path = os.path.join(settings.MEDIA_ROOT, asset.file_path)
|
||||
if os.path.exists(file_full_path):
|
||||
os.remove(file_full_path)
|
||||
|
||||
asset.delete()
|
||||
return Response(status=status.HTTP_204_NO_CONTENT)
|
||||
Reference in New Issue
Block a user