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
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:
parent
7d05339e05
commit
05be93546b
@ -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}'
|
||||
))
|
||||
@ -112,7 +112,17 @@ class DeviceBindSerializer(serializers.Serializer):
|
||||
|
||||
def validate_mac_address(self, value):
|
||||
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:
|
||||
raise serializers.ValidationError("设备不存在")
|
||||
return value
|
||||
|
||||
# 检查设备是否已被其他用户绑定
|
||||
if UserDevice.objects.filter(device=device).exists():
|
||||
raise serializers.ValidationError("设备已被其他用户绑定")
|
||||
|
||||
return value
|
||||
Loading…
x
Reference in New Issue
Block a user