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:
Poker Design Developer
2026-05-31 15:33:50 +08:00
parent 48629736f4
commit 0370e4018a
12 changed files with 749 additions and 4 deletions

View File

@@ -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}"

View File

@@ -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:

View File

@@ -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'),
]

View File

@@ -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)