boss_dp/server/api/contacts.py

# -*- coding: utf-8 -*-
"""
联系记录 API（需要登录）：
- GET    /api/contacts         -> 查询联系记录（支持分页、搜索）
- GET    /api/contacts/query   -> 按电话号码或微信号查询（数据展示）
- GET    /api/contacts/export  -> 导出联系记录为 Excel 文件（返回下载链接）
- POST   /api/contacts         -> 创建联系记录
- GET    /api/contacts/{id}    -> 查询单条记录
- PUT    /api/contacts/{id}    -> 更新记录
- DELETE /api/contacts/{id}    -> 删除记录
"""
import os
import uuid
from datetime import datetime

from django.db.models import Q
from django.conf import settings

from rest_framework import status
from rest_framework.decorators import api_view

from openpyxl import Workbook
from openpyxl.styles import Font, Alignment, PatternFill

from server.core.response import api_success, api_error
from server.models import ContactRecord
from server.serializers import ContactRecordSerializer


def _normalize_media_url(media_url: str) -> str:
    """Normalize MEDIA_URL so it always ends with '/' and is safe for URL拼接."""
    media_url = (media_url or "/media/").strip()
    if media_url.startswith(("http://", "https://")):
        return f"{media_url.rstrip('/')}/"
    return f"/{media_url.strip('/')}/"


@api_view(["GET", "POST"])
def contact_list(request):
    if request.method == "GET":
        search = request.query_params.get("search", "")
        page = int(request.query_params.get("page", 1))
        page_size = int(request.query_params.get("page_size", 10))
        reply_status = request.query_params.get("reply_status")
        wechat = request.query_params.get("wechat_exchanged")
        start_date = request.query_params.get("start_date")
        end_date = request.query_params.get("end_date")

        qs = ContactRecord.objects.all()
        if search:
            qs = qs.filter(
                Q(name__icontains=search) |
                Q(position__icontains=search) |
                Q(contact__icontains=search)
            )
        if reply_status:
            qs = qs.filter(reply_status=reply_status)
        if wechat is not None:
            qs = qs.filter(wechat_exchanged=wechat.lower() in ("true", "1"))
        if start_date:
            qs = qs.filter(created_at__gte=start_date)
        if end_date:
            qs = qs.filter(created_at__lte=f"{end_date} 23:59:59")

        total = qs.count()
        start = (page - 1) * page_size
        end = start + page_size
        records = qs[start:end]

        return api_success({
            "total": total,
            "page": page,
            "page_size": page_size,
            "results": ContactRecordSerializer(records, many=True).data,
        })

    ser = ContactRecordSerializer(data=request.data)
    ser.is_valid(raise_exception=True)
    ser.save()
    return api_success(ser.data, http_status=status.HTTP_201_CREATED)


@api_view(["GET"])
def contact_query(request):
    """
    按电话号码或微信号查询联系记录（数据展示接口）。
    传入 contact 参数（电话号码或微信号），返回匹配到的所有记录。
    """
    contact = (request.query_params.get("contact") or "").strip()
    if not contact:
        return api_error(status.HTTP_400_BAD_REQUEST, "请提供 contact 参数（电话号码或微信号）")
    page = int(request.query_params.get("page", 1))
    page_size = int(request.query_params.get("page_size", 20))

    qs = ContactRecord.objects.filter(contact__icontains=contact).order_by("-created_at")
    total = qs.count()
    start = (page - 1) * page_size
    end = start + page_size
    records = qs[start:end]

    return api_success({
        "total": total,
        "page": page,
        "page_size": page_size,
        "contact_keyword": contact,
        "results": ContactRecordSerializer(records, many=True).data,
    })


@api_view(["GET"])
def contact_export(request):
    """
    导出联系记录为 Excel 文件。
    支持与列表接口相同的筛选参数：search, reply_status, wechat_exchanged, start_date, end_date
    返回下载链接而不是直接返回文件流
    """
    search = request.query_params.get("search", "")
    reply_status = request.query_params.get("reply_status")
    wechat = request.query_params.get("wechat_exchanged")
    start_date = request.query_params.get("start_date")
    end_date = request.query_params.get("end_date")

    # 构建查询
    qs = ContactRecord.objects.all()
    if search:
        qs = qs.filter(
            Q(name__icontains=search) |
            Q(position__icontains=search) |
            Q(contact__icontains=search)
        )
    if reply_status:
        qs = qs.filter(reply_status=reply_status)
    if wechat is not None:
        qs = qs.filter(wechat_exchanged=wechat.lower() in ("true", "1"))
    if start_date:
        qs = qs.filter(created_at__gte=start_date)
    if end_date:
        qs = qs.filter(created_at__lte=f"{end_date} 23:59:59")

    # 创建 Excel 工作簿
    wb = Workbook()
    ws = wb.active
    ws.title = "联系记录"

    # 设置表头样式
    header_fill = PatternFill(start_color="4472C4", end_color="4472C4", fill_type="solid")
    header_font = Font(bold=True, color="FFFFFF", size=11)
    header_alignment = Alignment(horizontal="center", vertical="center")

    # 定义表头
    headers = ["ID", "姓名", "岗位", "联系方式", "回复状态", "是否交换微信", "Worker ID", "备注", "联系时间", "创建时间"]
    ws.append(headers)

    # 应用表头样式
    for cell in ws[1]:
        cell.fill = header_fill
        cell.font = header_font
        cell.alignment = header_alignment

    # 写入数据
    for record in qs.order_by("-created_at"):
        ws.append([
            record.id,
            record.name,
            record.position,
            record.contact,
            record.reply_status,
            "是" if record.wechat_exchanged else "否",
            record.worker_id,
            record.notes,
            record.contacted_at.strftime("%Y-%m-%d %H:%M:%S") if record.contacted_at else "",
            record.created_at.strftime("%Y-%m-%d %H:%M:%S"),
        ])

    # 调整列宽
    column_widths = [8, 12, 15, 18, 12, 15, 15, 30, 20, 20]
    for i, width in enumerate(column_widths, start=1):
        ws.column_dimensions[chr(64 + i)].width = width

    # 设置数据行样式
    for row in ws.iter_rows(min_row=2, max_row=ws.max_row):
        for cell in row:
            cell.alignment = Alignment(vertical="center")

    # 确保导出目录存在
    export_dir = settings.EXPORT_ROOT
    os.makedirs(export_dir, exist_ok=True)

    # 生成唯一文件名
    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
    unique_id = uuid.uuid4().hex[:8]
    filename = f"contacts_export_{timestamp}_{unique_id}.xlsx"
    filepath = os.path.join(export_dir, filename)

    # 保存文件
    wb.save(filepath)

    # 生成下载链接（始终返回可直接访问的完整 URL）
    media_url = _normalize_media_url(getattr(settings, "MEDIA_URL", "/media/"))
    download_path = f"{media_url}exports/{filename}"
    download_url = request.build_absolute_uri(download_path)

    return api_success({
        "download_url": download_url,
        "filename": filename,
        "total_records": qs.count(),
        "generated_at": datetime.now().strftime("%Y-%m-%d %H:%M:%S")
    })


@api_view(["GET", "PUT", "DELETE"])
def contact_detail(request, pk):
    try:
        obj = ContactRecord.objects.get(pk=pk)
    except ContactRecord.DoesNotExist:
        return api_error(status.HTTP_404_NOT_FOUND, "联系记录不存在")

    if request.method == "GET":
        return api_success(ContactRecordSerializer(obj).data)

    if request.method == "PUT":
        ser = ContactRecordSerializer(obj, data=request.data, partial=True)
        ser.is_valid(raise_exception=True)
        ser.save()
        return api_success(ser.data)

    obj.delete()
    return api_success(msg="联系记录已删除")