123 lines
4.9 KiB
Python
123 lines
4.9 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)
|
||
id = serializers.IntegerField(required=False, allow_null=True, default=None) # 账号 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.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"]
|