feat: update device interaction serializers and add fallback device command
All checks were successful
Build and Deploy LTY / build-and-deploy (push) Successful in 50m22s

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
pmc 2026-03-30 14:03:03 +08:00
parent 7d05339e05
commit 05be93546b
2 changed files with 114 additions and 2 deletions

View File

@ -0,0 +1,102 @@
from django.core.management.base import BaseCommand
from django.utils import timezone
from device_interaction.models import DeviceType, DeviceBatch, Device, UserDevice
from userapp.models import ParadiseUser
FALLBACK_MAC = '00:55:11:44:f3:22'
class Command(BaseCommand):
help = '创建 Unity Editor 测试用兜底设备MAC: 00:55:11:44:f3:22并绑定到指定用户'
def add_arguments(self, parser):
parser.add_argument(
'--username',
type=str,
default='',
help='绑定到哪个用户username不填则自动选第一个超级管理员'
)
def handle(self, *args, **options):
# 1. 找用户
username = options['username']
if username:
try:
user = ParadiseUser.objects.get(username=username)
except ParadiseUser.DoesNotExist:
self.stdout.write(self.style.ERROR(f'用户 "{username}" 不存在'))
return
else:
user = ParadiseUser.objects.filter(is_superuser=True).first()
if not user:
user = ParadiseUser.objects.first()
if not user:
self.stdout.write(self.style.ERROR('数据库中没有任何用户,请先创建用户'))
return
self.stdout.write(f'绑定目标用户: {user.username} (id={user.id})')
# 2. 获取或创建设备类型
device_type, created = DeviceType.objects.get_or_create(
code='T01',
defaults={
'name': '测试设备类型',
'description': 'Unity Editor 本地测试用'
}
)
if created:
self.stdout.write(self.style.SUCCESS(f'创建设备类型: {device_type.name} (T01)'))
else:
self.stdout.write(f'使用已有设备类型: {device_type.name} (T01)')
# 3. 获取或创建批次
batch, created = DeviceBatch.objects.get_or_create(
batch_number='BT01-TEST',
defaults={
'device_type': device_type,
'production_date': timezone.now().date(),
'quantity': 1,
'description': 'Unity Editor 测试批次'
}
)
if created:
self.stdout.write(self.style.SUCCESS(f'创建批次: {batch.batch_number}'))
else:
self.stdout.write(f'使用已有批次: {batch.batch_number}')
# 4. 获取或创建设备
device, created = Device.objects.get_or_create(
mac_address=FALLBACK_MAC,
defaults={
'device_type': device_type,
'batch': batch,
'serial_number': 'TEST-001',
'is_active': True,
'activated_at': timezone.now(),
}
)
if created:
self.stdout.write(self.style.SUCCESS(f'创建设备: {device.device_code} (MAC: {FALLBACK_MAC})'))
else:
self.stdout.write(f'设备已存在: {device.device_code} (MAC: {FALLBACK_MAC})')
# 确保已激活
if not device.is_active:
device.is_active = True
device.activated_at = timezone.now()
device.save()
self.stdout.write(self.style.SUCCESS('设备已更新为激活状态'))
# 5. 绑定用户
user_device, created = UserDevice.objects.get_or_create(
user=user,
device=device,
defaults={'nickname': 'Editor测试设备', 'is_primary': True}
)
if created:
self.stdout.write(self.style.SUCCESS(f'设备已绑定到用户 {user.username}'))
else:
self.stdout.write(f'设备已绑定过用户 {user.username},跳过')
self.stdout.write(self.style.SUCCESS(
f'\n完成!现在可以用以下地址测试:\n'
f'https://qy-lty.airlabs.art/api/device/rtc-token/get_by_mac/?mac_address={FALLBACK_MAC}'
))

View File

@ -112,7 +112,17 @@ class DeviceBindSerializer(serializers.Serializer):
def validate_mac_address(self, value): def validate_mac_address(self, value):
try: try:
Device.objects.get(mac_address=value) device = Device.objects.get(mac_address=value)
# 检查设备是否已被激活
if not device.is_active:
device.is_active = True
device.activated_at = timezone.now()
device.save()
except Device.DoesNotExist: except Device.DoesNotExist:
raise serializers.ValidationError("设备不存在") raise serializers.ValidationError("设备不存在")
return value
# 检查设备是否已被其他用户绑定
if UserDevice.objects.filter(device=device).exists():
raise serializers.ValidationError("设备已被其他用户绑定")
return value