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")
|