122 lines
4.8 KiB
Python
122 lines
4.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
DRF 序列化器。
|
|
"""
|
|
from rest_framework import serializers
|
|
|
|
from server.models import BossAccount, TaskLog, FilterConfig, ChatScript, ContactRecord, SystemConfig
|
|
|
|
|
|
# ────────────────────────── 账号 ──────────────────────────
|
|
|
|
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)
|
|
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.FloatField()
|
|
updated_at = serializers.FloatField()
|
|
|
|
|
|
# ────────────────────────── 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 FilterConfigSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = FilterConfig
|
|
fields = "__all__"
|
|
read_only_fields = ["id", "created_at", "updated_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"]
|