158 lines
6.3 KiB
Python
158 lines
6.3 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
DRF 序列化器。
|
||
"""
|
||
from rest_framework import serializers
|
||
|
||
from server.models import (
|
||
BossAccount, TaskLog, RecruitFilterSnapshot, ChatScript, ContactRecord, SystemConfig,
|
||
FollowUpConfig, FollowUpScript, FollowUpRecord
|
||
)
|
||
|
||
|
||
# ────────────────────────── 账号 ──────────────────────────
|
||
|
||
class BossAccountSerializer(serializers.ModelSerializer):
|
||
"""账号序列化器（读取用）。"""
|
||
# 运行时补充的字段（非数据库字段）
|
||
worker_name = serializers.CharField(read_only=True, default="")
|
||
worker_online = serializers.BooleanField(read_only=True, default=False)
|
||
|
||
class Meta:
|
||
model = BossAccount
|
||
fields = [
|
||
"id", "worker_id", "browser_id", "browser_name",
|
||
"boss_username", "boss_id", "is_logged_in",
|
||
"current_task_id", "current_task_status",
|
||
"checked_at", "created_at", "updated_at",
|
||
"worker_name", "worker_online",
|
||
]
|
||
|
||
|
||
class AccountBindSerializer(serializers.Serializer):
|
||
"""添加账号请求。"""
|
||
browser_name = serializers.CharField(max_length=128)
|
||
worker_id = serializers.CharField(max_length=64)
|
||
|
||
|
||
# ────────────────────────── 任务 ──────────────────────────
|
||
|
||
class TaskCreateSerializer(serializers.Serializer):
|
||
"""提交任务请求。"""
|
||
task_type = serializers.CharField(max_length=64)
|
||
id = serializers.IntegerField(required=False, allow_null=True, default=None) # 账号 ID
|
||
account_id = serializers.IntegerField(required=False, allow_null=True, default=None) # 同上，别名
|
||
boss_id = serializers.CharField(max_length=64, required=False, allow_blank=True, default="") # 即 /api/accounts 返回的 id（账号主键）
|
||
worker_id = serializers.CharField(max_length=64, required=False, allow_blank=True, default="")
|
||
account_name = serializers.CharField(max_length=128, required=False, allow_blank=True, default="")
|
||
params = serializers.JSONField(required=False, default=dict)
|
||
|
||
|
||
class TaskOutSerializer(serializers.Serializer):
|
||
"""任务信息响应。"""
|
||
task_id = serializers.CharField()
|
||
task_type = serializers.CharField()
|
||
status = serializers.CharField()
|
||
worker_id = serializers.CharField(allow_null=True)
|
||
account_name = serializers.CharField(allow_null=True)
|
||
params = serializers.DictField()
|
||
progress = serializers.CharField(allow_null=True)
|
||
result = serializers.JSONField(allow_null=True)
|
||
error = serializers.CharField(allow_null=True)
|
||
created_at = serializers.CharField()
|
||
updated_at = serializers.CharField()
|
||
|
||
|
||
# ────────────────────────── Worker ──────────────────────────
|
||
|
||
class BrowserProfileSerializer(serializers.Serializer):
|
||
"""浏览器窗口信息。"""
|
||
id = serializers.CharField()
|
||
name = serializers.CharField()
|
||
remark = serializers.CharField()
|
||
|
||
|
||
class WorkerOutSerializer(serializers.Serializer):
|
||
"""Worker 信息响应。"""
|
||
worker_id = serializers.CharField()
|
||
worker_name = serializers.CharField()
|
||
browsers = BrowserProfileSerializer(many=True)
|
||
online = serializers.BooleanField()
|
||
current_task_id = serializers.CharField(allow_null=True)
|
||
|
||
|
||
# ────────────────────────── 认证 ──────────────────────────
|
||
|
||
class LoginSerializer(serializers.Serializer):
|
||
"""登录请求。"""
|
||
username = serializers.CharField(max_length=64)
|
||
password = serializers.CharField(max_length=128)
|
||
|
||
|
||
# ────────────────────────── 话术 ──────────────────────────
|
||
|
||
class RecruitFilterSnapshotSerializer(serializers.ModelSerializer):
|
||
class Meta:
|
||
model = RecruitFilterSnapshot
|
||
fields = "__all__"
|
||
read_only_fields = ["id", "synced_at", "created_at"]
|
||
|
||
|
||
class ChatScriptSerializer(serializers.ModelSerializer):
|
||
script_type_display = serializers.CharField(source="get_script_type_display", read_only=True)
|
||
|
||
class Meta:
|
||
model = ChatScript
|
||
fields = "__all__"
|
||
read_only_fields = ["id", "created_at", "updated_at"]
|
||
|
||
|
||
# ────────────────────────── 联系记录 ──────────────────────────
|
||
|
||
class ContactRecordSerializer(serializers.ModelSerializer):
|
||
class Meta:
|
||
model = ContactRecord
|
||
fields = "__all__"
|
||
read_only_fields = ["id", "created_at"]
|
||
|
||
|
||
# ────────────────────────── 系统配置 ──────────────────────────
|
||
|
||
class SystemConfigSerializer(serializers.ModelSerializer):
|
||
class Meta:
|
||
model = SystemConfig
|
||
fields = "__all__"
|
||
read_only_fields = ["updated_at"]
|
||
|
||
# ────────────────────────── 复聊配置 ──────────────────────────
|
||
|
||
class FollowUpScriptSerializer(serializers.ModelSerializer):
|
||
"""复聊话术序列化器。"""
|
||
class Meta:
|
||
model = FollowUpScript
|
||
fields = "__all__"
|
||
read_only_fields = ["id", "created_at"]
|
||
|
||
|
||
class FollowUpConfigSerializer(serializers.ModelSerializer):
|
||
"""复聊配置序列化器（包含关联的话术列表）。"""
|
||
scripts = serializers.SerializerMethodField()
|
||
|
||
class Meta:
|
||
model = FollowUpConfig
|
||
fields = "__all__"
|
||
read_only_fields = ["id", "created_at", "updated_at"]
|
||
|
||
def get_scripts(self, obj):
|
||
"""获取该配置下的所有话术。"""
|
||
scripts = FollowUpScript.objects.filter(config_id=obj.id, is_active=True).order_by('day_number', 'order')
|
||
return FollowUpScriptSerializer(scripts, many=True).data
|
||
|
||
|
||
class FollowUpRecordSerializer(serializers.ModelSerializer):
|
||
"""复聊记录序列化器。"""
|
||
class Meta:
|
||
model = FollowUpRecord
|
||
fields = "__all__"
|
||
read_only_fields = ["id", "sent_at"]
|