浅念
16207ae309
GET/POST /api/admin/sites、PATCH /api/admin/sites/<pk> 登录成员可用； DELETE 仅 admin，走软删（is_active=False），物理清除留待未来管理命令 分批删（防大站级联删除锁表），CASCADE 仅完整性兜底。域名归一去协议/ 端口/路径、小写、去空去重；错误响应统一信封（EnvelopeMixin）。 软删后 collect 上报 403 not_allowed（Task7 联动，测试锁定）。 Co-Authored-By: Claude Code <noreply@anthropic.com>
85 lines
3.6 KiB
Python
85 lines
3.6 KiB
Python
from django.shortcuts import get_object_or_404
|
||
from rest_framework.views import APIView
|
||
from rest_framework.response import Response
|
||
from rest_framework.permissions import IsAdminUser
|
||
from tracker.api.envelope import EnvelopeMixin, ok
|
||
from tracker.models import Site
|
||
|
||
|
||
def _norm_domains(raw) -> list[str]:
|
||
"""域名归一：去协议/端口/路径、转小写、去空去重。"""
|
||
out = []
|
||
for d in raw or []:
|
||
d = str(d).strip().lower()
|
||
for pre in ("https://", "http://"):
|
||
if d.startswith(pre):
|
||
d = d[len(pre):]
|
||
d = d.split("/")[0].split(":")[0]
|
||
if d and d not in out:
|
||
out.append(d)
|
||
return out
|
||
|
||
|
||
def _serialize(site: Site) -> dict:
|
||
return {"id": site.pk, "name": site.name, "site_id": site.site_id,
|
||
"domains": site.domains, "allow_no_origin": site.allow_no_origin,
|
||
"is_active": site.is_active, "created_at": site.created_at.isoformat()}
|
||
|
||
|
||
class SiteListView(EnvelopeMixin, APIView):
|
||
def get(self, request):
|
||
return ok([_serialize(s) for s in Site.objects.order_by("-created_at")])
|
||
|
||
def post(self, request):
|
||
data = request.data if isinstance(request.data, dict) else {} # JSON 数组等畸形 body → 空表（.get 会 500）
|
||
name = data.get("name")
|
||
if not isinstance(name, str) or not name.strip():
|
||
return Response({"code": 400, "data": None, "message": "invalid_name"}, status=400)
|
||
domains = _norm_domains(data.get("domains"))
|
||
if not domains:
|
||
return Response({"code": 400, "data": None, "message": "invalid_domains"}, status=400)
|
||
site = Site.objects.create(name=name.strip()[:100], domains=domains,
|
||
allow_no_origin=bool(data.get("allow_no_origin")))
|
||
return ok(_serialize(site), "created")
|
||
|
||
|
||
class SiteDetailView(EnvelopeMixin, APIView):
|
||
def get_permissions(self):
|
||
# DELETE（软删）仅 admin；GET/PATCH 登录成员即可（settings 默认 IsAuthenticated）。
|
||
# 在 initial() 阶段判权限：403 先于 404，且不泄露站点存在性。
|
||
if self.request.method == "DELETE":
|
||
return [IsAdminUser()]
|
||
return super().get_permissions()
|
||
|
||
def _get(self, request, pk):
|
||
return get_object_or_404(Site, pk=pk)
|
||
|
||
def patch(self, request, pk):
|
||
site = self._get(request, pk)
|
||
data = request.data if isinstance(request.data, dict) else {}
|
||
if "name" in data:
|
||
name = data["name"]
|
||
if not isinstance(name, str) or not name.strip():
|
||
return Response({"code": 400, "data": None, "message": "invalid_name"}, status=400)
|
||
site.name = name.strip()[:100]
|
||
if "domains" in data:
|
||
domains = _norm_domains(data["domains"])
|
||
if not domains:
|
||
return Response({"code": 400, "data": None, "message": "invalid_domains"}, status=400)
|
||
site.domains = domains
|
||
if "allow_no_origin" in data:
|
||
site.allow_no_origin = bool(data["allow_no_origin"])
|
||
if "is_active" in data:
|
||
site.is_active = bool(data["is_active"])
|
||
site.save()
|
||
return ok(_serialize(site))
|
||
|
||
def delete(self, request, pk):
|
||
# 软删：默认停用即可（上报侧 collect 见 is_active=False 即 403，Task7 联动）。
|
||
# 物理清除走未来管理命令分批删——大站单语句级联删除有锁表/binlog 风暴风险，
|
||
# CASCADE 仅作完整性兜底，不在请求路径触发。
|
||
site = self._get(request, pk)
|
||
site.is_active = False
|
||
site.save(update_fields=["is_active", "updated_at"])
|
||
return ok(_serialize(site), "deactivated")
|