fix(test-round): 测试清单一批 bug 修复 + 团队月限额持久化

后端:
- accounts: 团队级月限额持久化(Team.monthly_credit_limit 三态 + PATCH/GET
  /api/auth/team/settings/,刷新不丢,PMC#12);头像上传 500→502 可读错误;
  设备下线真失效(删并重建 token,旧 token 401)
- assets: 审核类目加 model_portrait,消除三视图"无需审核"误报
- ai/projects/products: 模特已有三视图复用、产品三视图同步回商品库
  (metadata.view=three_view)、真人模特三视图回写

前端:
- 商品图删除判断改用真实图片数;团队成员弹窗禁点外部关闭
- 图片预览骨架铺满占位;平台套图左侧栏折叠改导航同款;收件箱长文换行对齐
- 团队月限额改真落库(乐观更新+失败回滚)

测试:新增 TeamSettingsTests(5 条),accounts 全套 36 tests 通过;前端 build 0 error

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
zyc
2026-06-29 18:50:22 +08:00
co-authored by Claude Opus 4.8
parent ca0c3e2300
commit b0b3ac2348
20 changed files with 476 additions and 57 deletions
@@ -0,0 +1,20 @@
# Generated by Django 5.1.15 on 2026-06-29 10:44
from django.db import migrations, models
class Migration(migrations.Migration):
dependencies = [
("accounts", "0006_teammember_daily_credit_limit_and_more"),
]
operations = [
migrations.AddField(
model_name="team",
name="monthly_credit_limit",
field=models.DecimalField(
blank=True, decimal_places=2, default=None, max_digits=12, null=True
),
),
]
+3
View File
@@ -32,6 +32,9 @@ class Team(TimeStampedModel):
name = models.CharField(max_length=128)
status = models.CharField(max_length=24, choices=Status.choices, default=Status.ACTIVE)
owner = models.ForeignKey(User, on_delete=models.PROTECT, related_name="owned_teams")
# 团队级月限额(超管在团队页设置,自然月重置)。语义三态,与成员级 0=不限 不同:
# None = 未设置(前端按成员月度额度累加作团队月限额)· -1 = 不限 · >=0 = 固定上限。
monthly_credit_limit = models.DecimalField(max_digits=12, decimal_places=2, null=True, blank=True, default=None)
def __str__(self) -> str:
return self.name
+1 -1
View File
@@ -38,7 +38,7 @@ class LoginSessionSerializer(serializers.ModelSerializer):
class TeamSerializer(serializers.ModelSerializer):
class Meta:
model = Team
fields = ["id", "name", "status", "owner", "created_at", "updated_at"]
fields = ["id", "name", "status", "owner", "monthly_credit_limit", "created_at", "updated_at"]
read_only_fields = ["id", "status", "owner", "created_at", "updated_at"]
+168
View File
@@ -404,3 +404,171 @@ class ValidateInviteTests(TestCase):
self.assertEqual(r.status_code, 200)
self.assertFalse(r.data["valid"])
class AvatarUploadTests(TestCase):
"""bug(9):上传新头像点「确认使用」弹 500。真因之一=对象存储上传异常裸抛 500;
现改为捕获后回 502 + 人话,成功路径回 200 + 新 avatar_url。"""
def setUp(self):
from rest_framework.authtoken.models import Token
self.user = User.objects.create_user(username="av-user", password="strong-password")
self.token = Token.objects.create(user=self.user)
self.client = APIClient()
self.client.credentials(HTTP_AUTHORIZATION=f"Token {self.token.key}")
def _png(self):
from django.core.files.uploadedfile import SimpleUploadedFile
# 1x1 PNG 字节(够 multipart 上传,内容无所谓,storage 被 mock)
png = (
b"\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR\x00\x00\x00\x01\x00\x00\x00\x01"
b"\x08\x06\x00\x00\x00\x1f\x15\xc4\x89\x00\x00\x00\nIDATx\x9cc\x00"
b"\x01\x00\x00\x05\x00\x01\r\n-\xb4\x00\x00\x00\x00IEND\xaeB`\x82"
)
return SimpleUploadedFile("avatar.png", png, content_type="image/png")
def test_upload_success_returns_200_and_sets_url(self):
from unittest import mock
with mock.patch("apps.assets.storage.TosStorage") as MockStorage:
inst = MockStorage.return_value
inst.upload_fileobj.return_value = None
inst.public_url.return_value = "https://bucket.example.com/users/x/avatar/abc.png"
r = self.client.post("/api/auth/me/avatar/", {"file": self._png()}, format="multipart")
self.assertEqual(r.status_code, 200, r.data)
self.user.refresh_from_db()
self.assertEqual(self.user.avatar_url, "https://bucket.example.com/users/x/avatar/abc.png")
def test_upload_storage_failure_returns_502_not_bare_500(self):
from unittest import mock
with mock.patch("apps.assets.storage.TosStorage") as MockStorage:
MockStorage.return_value.upload_fileobj.side_effect = RuntimeError("TOS unreachable")
r = self.client.post("/api/auth/me/avatar/", {"file": self._png()}, format="multipart")
self.assertEqual(r.status_code, 502)
self.assertIn("detail", r.data)
def test_upload_without_file_returns_400(self):
r = self.client.post("/api/auth/me/avatar/", {}, format="multipart")
self.assertEqual(r.status_code, 400)
class DeviceRevokeTests(TestCase):
"""bug(10):在用设备点「下线」提示成功但实际没下线。真因=只标记 LoginSession.revoked_at
没动 token(单 token 体系),被下线设备旧 token 仍有效。现改为旋转 token,旧 token 立即 401。"""
def setUp(self):
from rest_framework.authtoken.models import Token
self.user = User.objects.create_user(username="dev-user", password="strong-password")
# 模拟登录拿到 token(被下线的「目标设备」用的就是这个)
self.token = Token.objects.create(user=self.user)
def test_revoke_session_invalidates_old_token(self):
from apps.accounts.models import LoginSession
session = LoginSession.objects.create(user=self.user, user_agent="OtherDevice", ip_address="1.1.1.1")
old_key = self.token.key
client = APIClient()
client.credentials(HTTP_AUTHORIZATION=f"Token {old_key}")
r = client.post(f"/api/auth/me/sessions/{session.id}/revoke/")
self.assertEqual(r.status_code, 200)
# 必须真正吊销:旧 token 已失效,再用它访问受保护接口应 401
self.assertIn("token", r.data)
self.assertNotEqual(r.data["token"], old_key)
stale = APIClient()
stale.credentials(HTTP_AUTHORIZATION=f"Token {old_key}")
self.assertEqual(stale.get("/api/auth/me/").status_code, 401)
# 会话被标记下线
session.refresh_from_db()
self.assertIsNotNone(session.revoked_at)
def test_revoke_unknown_session_no_op(self):
import uuid as _uuid
client = APIClient()
client.credentials(HTTP_AUTHORIZATION=f"Token {self.token.key}")
r = client.post(f"/api/auth/me/sessions/{_uuid.uuid4()}/revoke/")
self.assertEqual(r.status_code, 200)
self.assertEqual(r.data.get("revoked"), 0)
class TeamSettingsTests(TestCase):
"""PMC#12:团队月限额持久化。设置后落库,刷新(重新 GET)仍在;成员不可改;清空回到「未设置」。"""
def _register(self, client, username, **extra):
return client.post(
"/api/auth/register/",
{"username": username, "password": "strong-password", **extra},
format="json",
)
def setUp(self):
self.owner_client = APIClient()
r = self._register(self.owner_client, "ts-owner", team_name="TS Team", invite_code=make_create_team_code())
self.assertEqual(r.status_code, 201)
self.team = Team.objects.get(name="TS Team")
self.owner_client.credentials(HTTP_AUTHORIZATION=f"Token {r.data['token']}")
# 注册返回体里团队月限额默认未设置(null)
self.assertIsNone(r.data["team"].get("monthly_credit_limit"))
def test_owner_sets_limit_persists(self):
res = self.owner_client.patch(
"/api/auth/team/settings/", {"monthly_credit_limit": 3000}, format="json"
)
self.assertEqual(res.status_code, 200, res.content)
self.assertEqual(Decimal(res.data["monthly_credit_limit"]), Decimal("3000"))
# 落库
self.team.refresh_from_db()
self.assertEqual(self.team.monthly_credit_limit, Decimal("3000"))
# 「刷新」= 重新 GET,值仍在(PMC#12 核心)
got = self.owner_client.get("/api/auth/team/settings/")
self.assertEqual(Decimal(got.data["monthly_credit_limit"]), Decimal("3000"))
def test_unlimited_minus_one_stored_as_is(self):
res = self.owner_client.patch(
"/api/auth/team/settings/", {"monthly_credit_limit": -1}, format="json"
)
self.assertEqual(res.status_code, 200, res.content)
self.team.refresh_from_db()
self.assertEqual(self.team.monthly_credit_limit, Decimal("-1"))
def test_empty_clears_to_unset(self):
self.owner_client.patch("/api/auth/team/settings/", {"monthly_credit_limit": 3000}, format="json")
res = self.owner_client.patch(
"/api/auth/team/settings/", {"monthly_credit_limit": ""}, format="json"
)
self.assertEqual(res.status_code, 200, res.content)
self.team.refresh_from_db()
self.assertIsNone(self.team.monthly_credit_limit)
def test_invalid_number_rejected(self):
res = self.owner_client.patch(
"/api/auth/team/settings/", {"monthly_credit_limit": "abc"}, format="json"
)
self.assertEqual(res.status_code, 400)
def test_member_cannot_change_limit(self):
self.owner_client.post(
"/api/auth/team/members/",
{"username": "ts-member", "password": "strong-password", "role": "member"},
format="json",
)
member_client = APIClient()
login = member_client.post(
"/api/auth/login/", {"username": "ts-member", "password": "strong-password"}, format="json"
)
member_client.credentials(HTTP_AUTHORIZATION=f"Token {login.data['token']}")
# 成员可读
self.assertEqual(member_client.get("/api/auth/team/settings/").status_code, 200)
# 但不可改
res = member_client.patch(
"/api/auth/team/settings/", {"monthly_credit_limit": 9999}, format="json"
)
self.assertEqual(res.status_code, 403)
self.team.refresh_from_db()
self.assertIsNone(self.team.monthly_credit_limit)
+2
View File
@@ -15,6 +15,7 @@ from .views import (
team_member_detail,
team_member_password,
team_members,
team_settings,
update_avatar,
validate_invite,
)
@@ -32,6 +33,7 @@ urlpatterns = [
path("me/sessions/", login_sessions, name="auth-sessions"),
path("me/sessions/revoke-others/", revoke_other_sessions, name="auth-sessions-revoke-others"),
path("me/sessions/<uuid:session_id>/revoke/", revoke_login_session, name="auth-session-revoke"),
path("team/settings/", team_settings, name="team-settings"),
path("team/members/", team_members, name="team-members"),
path("team/members/<uuid:member_id>/", team_member_detail, name="team-member-detail"),
path("team/members/<uuid:member_id>/password/", team_member_password, name="team-member-password"),
+58 -8
View File
@@ -213,12 +213,24 @@ def update_avatar(request):
user = request.user
suffix = Path(upload.name or "").suffix.lower() or ".png"
object_key = f"users/{user.id}/avatar/{uuid.uuid4()}{suffix}"
storage = TosStorage()
storage.upload_fileobj(
fileobj=upload.file,
object_key=object_key,
content_type=upload.content_type or "image/png",
)
# 对象存储上传是唯一会抛异常的环节(凭证/网络/桶不可达)。不捕获则裸抛 500,
# 前端只看到「服务器错误 500」查不到真因(与全站「raise_for_status 吞 body」同类坑)。
# 捕获后回 502 + 人话,既不掩盖问题(日志仍有栈)又让前端 toast 可读。
try:
storage = TosStorage()
storage.upload_fileobj(
fileobj=upload.file,
object_key=object_key,
content_type=upload.content_type or "image/png",
)
except Exception as exc: # noqa: BLE001 — 对象存储故障归一成可读错误,不裸 500
import logging
logging.getLogger(__name__).exception("avatar upload to TOS failed: %s", exc)
return Response(
{"detail": "头像上传失败,请稍后重试(对象存储不可用)"},
status=status.HTTP_502_BAD_GATEWAY,
)
# 桶公读,存稳定的虚拟主机式直链(全站资产同款 public_url)——不存预签名 URL:
# 预签名 URL 带一长串 SigV4 查询参数(常 >300 字符),会超出 avatar_url 这个 URLField
# 的默认 max_length=200,MySQL 严格模式下 save 抛 DataError(Data too long)→ 500;
@@ -320,6 +332,32 @@ def team_members(request):
return Response(TeamMemberSerializer(member).data, status=status.HTTP_201_CREATED)
@api_view(["GET", "PATCH"])
@permission_classes([IsAuthenticated])
def team_settings(request):
"""团队级设置:目前承载「设置月限额」(团队页超管设置,刷新后持久化 · PMC#12)。
GET 任意成员可读;PATCH 仅超管/团管(can_manage_team)。
monthly_credit_limit 三态:null=未设置(按成员累加)· -1=不限 · >=0=固定上限。"""
team = get_current_team(request.user)
if request.method == "GET":
return Response(TeamSerializer(team).data)
if not can_manage_team(request.user, team):
return Response({"detail": "permission denied"}, status=status.HTTP_403_FORBIDDEN)
if "monthly_credit_limit" in request.data:
raw = request.data.get("monthly_credit_limit")
if raw is None or (isinstance(raw, str) and not raw.strip()):
team.monthly_credit_limit = None # 清空 = 回到「未设置(按成员累加)」
else:
try:
team.monthly_credit_limit = Decimal(str(raw))
except (InvalidOperation, ValueError, TypeError):
return Response({"monthly_credit_limit": ["invalid number"]}, status=status.HTTP_400_BAD_REQUEST)
team.save(update_fields=["monthly_credit_limit", "updated_at"])
return Response(TeamSerializer(team).data)
@api_view(["GET", "POST"])
@permission_classes([IsAuthenticated])
def team_invitations(request):
@@ -462,7 +500,14 @@ def login_sessions(request):
@permission_classes([IsAuthenticated])
def revoke_login_session(request, session_id):
"""下线单个设备:把同一台设备(UA + IP)下的所有未下线会话一并下线,
否则去重展示的一台设备点「下线」后,底层其它重复会话仍存活会再次冒出来。"""
否则去重展示的一台设备点「下线」后,底层其它重复会话仍存活会再次冒出来。
关键:DRF TokenAuthentication 是「一用户一 token」(authtoken 共享一行),
只标记 LoginSession.revoked_at 根本不动 token —— 被下线的设备仍拿着有效 token,
继续访问畅通无阻(bug:提示「设备已下线」但实际没下线)。单 token 体系下无法
只吊销某一台,唯一能真正下线的做法 = 旋转 token(删旧 + 发新),令所有端旧 token
立即失效,再把新 token 发回当前设备(同 revoke_other_sessions 的思路)。
被下线的目标设备下次请求即 401,真正离线。"""
from django.utils import timezone
target = LoginSession.objects.filter(user=request.user, id=session_id).first()
@@ -474,7 +519,12 @@ def revoke_login_session(request, session_id):
ip_address=target.ip_address,
revoked_at__isnull=True,
).update(revoked_at=timezone.now())
return Response({"revoked": updated})
# 真正吊销:旋转用户 token(令被下线设备旧 token 立即失效),再发新 token 给当前设备。
Token.objects.filter(user=request.user).delete()
token, _ = Token.objects.get_or_create(user=request.user)
# 刷新当前设备这条会话(旋转后当前设备等价于「重新登录」),避免它被自己下线后又冒出来
record_login_session(request, request.user)
return Response({"revoked": updated, "token": token.key})
@api_view(["POST"])
+30 -11
View File
@@ -1317,16 +1317,28 @@ def run_base_asset_task(*, task_id: str) -> None:
BaseAssetGroup.Kind.PERSON: Asset.Category.PERSON,
BaseAssetGroup.Kind.SCENE: Asset.Category.SCENE,
}[kind]
# 商品基础资产 = 商品三视图(prompt 即「生成同一件商品的三视图」):名字带「三视图」并打
# metadata.view=three_view + product_id,让商品库的三视图查询(products is_triview / ?product 过滤)
# 能认出并回填——否则视频项目生成的商品三视图同步不回商品库,商品库永远显示「尚未生成」(ZWQ#5)。
is_product = kind == BaseAssetGroup.Kind.PRODUCT
asset_name = f"{project.name}-商品三视图" if is_product else f"{project.name}-{kind}"
asset = _store_generated_media(
team=project.team,
user=user,
project=project,
task=task,
media=media,
name=f"{project.name}-{kind}",
name=asset_name,
category=category,
asset_type=Asset.Type.IMAGE,
)
if is_product:
meta = dict(asset.metadata or {})
meta["view"] = "three_view"
if project.product_id:
meta["product_id"] = str(project.product_id)
asset.metadata = meta
asset.save(update_fields=["metadata", "updated_at"])
# 复用同实体的组:追加候选 + 采用最新(版本=candidate_assets,采用=adopted_asset);无则新建
group = _find_entity_group(project, kind, label, group_id)
if group is None:
@@ -1423,19 +1435,26 @@ def run_triview_task(*, task_id: str) -> None:
group.candidate_assets.add(asset)
group.adopted_asset = asset
group.save(update_fields=["adopted_asset", "updated_at"])
# B 路(期3):视频流程新生成的「角色」立绘 + 三视图成套 → 自动加入模特库(团队级可复用)。
# 幂等:同一立绘只建一次 Model;best-effort,不影响主链路。
# B 路(期3):视频流程新生成的「角色」立绘 + 三视图成套 → 自动加入/更新模特库(团队级可复用)。
# 幂等:同一立绘只建一次 Model;已存在该立绘的 Model(尤其「真人上传」模特,upload 时只建了 portrait、
# 还没 triview)→ 回写 triview_asset,否则模特库永远显示「无三视图」(ZWQ#7:AI 角色能更新、真人上传不更新)。
from apps.assets.models import Model as ModelEntity
portrait = Asset.objects.filter(id=asset_key).first()
if portrait is not None and not ModelEntity.objects.filter(portrait_asset=portrait).exists():
ModelEntity.objects.create(
team=project.team, created_by=user,
name=(portrait.name or f"{project.name}-角色").split("·")[0].strip()[:255] or "角色",
source=ModelEntity.Source.AI,
portrait_asset=portrait, triview_asset=asset,
metadata={"from_project": str(project.id), "auto_enrolled": True},
)
if portrait is not None:
existing_model = ModelEntity.objects.filter(portrait_asset=portrait, is_deleted=False).first()
if existing_model is None:
ModelEntity.objects.create(
team=project.team, created_by=user,
name=(portrait.name or f"{project.name}-角色").split("·")[0].strip()[:255] or "角色",
source=ModelEntity.Source.AI,
portrait_asset=portrait, triview_asset=asset,
metadata={"from_project": str(project.id), "auto_enrolled": True},
)
elif existing_model.triview_asset_id != asset.id:
# 真人上传 / 已有模特补三视图:回写新三视图(应用后模特库即刷新三视图)
existing_model.triview_asset = asset
existing_model.save(update_fields=["triview_asset", "updated_at"])
# 合规(期3):三视图含人脸,视频生成前必须过火山审核 → 事务提交后静默送审(best-effort)
from apps.assets.review import submit_asset_for_review
+4 -2
View File
@@ -34,10 +34,12 @@ class Asset(TeamOwnedModel):
FREE_CREATE = "free_create", "Free Create" # 自由创作(图片趴)
STORYBOARD = "storyboard", "Storyboard" # 分镜图(视频趴·送审)
# 送审范围(火山人像审核):视频趴含人脸 = 角色定妆照 / 三视图 / 分镜图 / 场景 / 商品图。
# 送审范围(火山人像审核):视频趴含人脸 = 角色定妆照 / 模特形象图 / 三视图 / 分镜图 / 场景 / 商品图。
# model_portrait(模特库形象图,尤其真人上传)与 person 同样是真人脸,必须可送审——否则新增模特后在
# 基础资产点「待审核」会被判「该素材无需审核」(ZWQ#6:AI 角色 person 可审、上传模特 model_portrait 报无需审核)。
# 场景图与商品图也可能出现真人(模特出镜/上身),不送审则视频路只能传原始直链 → 火山判「疑似真人」拒;
# 送审后拿到 remote_id,视频路换 asset:// 素材库引用即可放行。图片趴(上身图/套图/创作)仍不送审。
REVIEW_CATEGORIES = ("person", "tri_view", "storyboard", "scene", "product_image")
REVIEW_CATEGORIES = ("person", "model_portrait", "tri_view", "storyboard", "scene", "product_image")
name = models.CharField(max_length=255)
asset_type = models.CharField(max_length=24, choices=Type.choices)
+12 -7
View File
@@ -182,7 +182,8 @@ class SubmitReviewTests(TestCase):
self.client = APIClient()
self.client.force_authenticate(self.user)
self.person = Asset.objects.create(team=self.team, name="角色", asset_type="image", source="ai_generated", category=Asset.Category.PERSON)
self.product = Asset.objects.create(team=self.team, name="商品图", asset_type="image", source="upload", category=Asset.Category.PRODUCT_IMAGE)
# 图片趴(模特上身图)= 非送审类,用于校验 submit-review 对其拒绝(product_image/scene 现已属送审类)
self.tryon = Asset.objects.create(team=self.team, name="上身图", asset_type="image", source="ai_generated", category=Asset.Category.MODEL_TRYON)
def test_submit_review_person_ok(self):
with patch("apps.assets.review.submit_asset_for_review") as sub:
@@ -192,7 +193,7 @@ class SubmitReviewTests(TestCase):
self.assertIn("review_status", res.json())
def test_submit_review_rejects_non_review_category(self):
res = self.client.post(f"/api/assets/{self.product.id}/submit-review/")
res = self.client.post(f"/api/assets/{self.tryon.id}/submit-review/")
self.assertEqual(res.status_code, 400)
def test_submit_review_503_when_submission_did_not_start(self):
@@ -218,22 +219,26 @@ class SubmitReviewTests(TestCase):
class ReviewScopeTests(TestCase):
"""送审范围 = 角色定妆照(person)/ 三视图(tri_view)/ 分镜图(storyboard);图片趴不送审。"""
"""送审范围 = 含人脸资产:角色定妆照(person)/ 模特形象图(model_portrait)/ 三视图(tri_view)/
分镜图(storyboard)/ 场景(scene)/ 商品图(product_image);图片趴(上身图/套图/创作)不送审"""
def test_review_categories_constant(self):
self.assertEqual(Asset.REVIEW_CATEGORIES, ("person", "tri_view", "storyboard"))
self.assertEqual(
Asset.REVIEW_CATEGORIES,
("person", "model_portrait", "tri_view", "storyboard", "scene", "product_image"),
)
def test_poll_team_reviews_only_scans_review_categories(self):
from apps.assets import review
_, team = _mk_team("uc", "TeamC")
keep = {}
for cat in ("person", "tri_view", "storyboard"):
for cat in ("person", "model_portrait", "tri_view", "storyboard", "scene", "product_image"):
a = Asset.objects.create(team=team, name=cat, asset_type="image", source="ai_generated",
category=cat, review_status="processing")
keep[cat] = str(a.id)
# 图片趴 / 场景:不应进送审队列
for cat in ("model_tryon", "model_portrait", "scene"):
# 图片趴:不应进送审队列
for cat in ("model_tryon", "platform_kit", "free_create"):
Asset.objects.create(team=team, name=cat, asset_type="image", source="ai_generated",
category=cat, review_status="processing")
with patch("apps.assets.review.assets_client.is_enabled", return_value=False):
+5 -1
View File
@@ -82,8 +82,12 @@ class ProductViewSet(TeamScopedViewSetMixin, ModelViewSet):
def ref_count(asset):
return BaseAssetGroup.objects.filter(Q(adopted_asset=asset) | Q(candidate_assets=asset)).values("project").distinct().count()
# 三视图:tri_view 类 / 老数据(metadata.view=three_view 或名字含三视图)
# 人物三视图(给角色卡配对用):tri_view 类 / 老数据(metadata.view=three_view 或名字含三视图)
# 排除 product_image —— 商品三视图(ZWQ#5)也带 view=three_view,但它属商品图而非人物角色,
# 否则会被当成孤儿角色卡误列进「角色」区(同时它已在 product_images 区正常展示)。
def is_triview(a):
if a.category == "product_image":
return False
return a.category == "tri_view" or (a.metadata or {}).get("view") == "three_view" or "三视图" in (a.name or "")
triviews = [a for a in assets if is_triview(a)]
+49
View File
@@ -596,9 +596,58 @@ class ProjectViewSet(TeamScopedViewSetMixin, ModelViewSet):
from apps.assets.review import submit_asset_for_review
transaction.on_commit(lambda a=asset: submit_asset_for_review(a))
# ZWQ#4 · 沿用模特库已有三视图:选了「已带三视图」的模特(官方/自有)作角色时,
# 把该模特的三视图也挂进项目(triview_of=本组采用立绘 id),前端即判定「已有三视图」、
# 不再要求点「AI 生成三视图」。仅对人物组生效;asset_id 传的是模特的 portrait_asset(原始 id),
# 故按原始 portrait_asset 反查 Model;官方模特三视图同立绘一样克隆进本团队再挂(team 隔离)。
if asset.category in (Asset.Category.PERSON, Asset.Category.MODEL_PORTRAIT):
self._attach_existing_model_triview(project, group, asset, source_portrait_id=asset_id, user=request.user)
promote_base_asset_stage_if_ready(project)
return Response(BaseAssetGroupSerializer(group).data)
def _attach_existing_model_triview(self, project, portrait_group, portrait_asset, *, source_portrait_id, user):
"""若被采用的立绘来自「已带三视图」的模特,把该三视图挂进项目并采用(triview_of=立绘 id)。
幂等:已存在该立绘的三视图组就跳过,不重复挂best-effort,出错不阻断主采用流程"""
from apps.assets.models import Model as ModelEntity
# source_portrait_id = 前端传的资产 id(模特的原始 portrait_asset);官方模特会被克隆,
# 故 portrait_asset 可能是克隆体。两边都查一遍找到对应 Model。
candidate_ids = {str(portrait_asset.id)}
if source_portrait_id:
candidate_ids.add(str(source_portrait_id))
model = (
ModelEntity.objects.filter(portrait_asset_id__in=candidate_ids, is_deleted=False)
.exclude(triview_asset__isnull=True)
.select_related("triview_asset")
.first()
)
if model is None or model.triview_asset is None:
return
portrait_key = str(portrait_asset.id)
# 已有该立绘的三视图组?幂等跳过。
existing = next(
(g for g in project.base_asset_groups.filter(kind=BaseAssetGroup.Kind.PERSON)
if (g.metadata or {}).get("triview_of") == portrait_key),
None,
)
if existing is not None:
return
tri_asset = model.triview_asset
# 官方模特三视图跨团队 → 克隆进本团队再挂(与立绘同款);本团队三视图直接用。
if tri_asset.team_id != project.team_id:
tri_asset = _clone_asset_into_team(tri_asset, team=project.team, user=user)
tri_group = BaseAssetGroup.objects.create(
project=project, kind=BaseAssetGroup.Kind.PERSON, prompt="",
metadata={"label": "·三视图", "triview_of": portrait_key, "adopt": "adopted"},
)
tri_group.candidate_assets.add(tri_asset)
tri_group.adopted_asset = tri_asset
tri_group.save(update_fields=["adopted_asset", "updated_at"])
if tri_asset.category in Asset.REVIEW_CATEGORIES and tri_asset.review_status != "active":
from apps.assets.review import submit_asset_for_review
transaction.on_commit(lambda a=tri_asset: submit_asset_for_review(a))
@action(detail=True, methods=["post"], url_path="generate-triview")
def generate_triview(self, request, pk=None):
"""流程步骤4 · 据某一版立绘资产生成它配套的三视图(image_edit 锁角色一致性,三视图绑定该立绘)。"""