lib.itmens/users/views.py

203 lines
5.7 KiB
Python
Raw Normal View History

import json
from discord import SyncWebhook
2020-05-05 23:50:48 +08:00
from django.contrib.auth.decorators import login_required
from django.core.exceptions import BadRequest, PermissionDenied
from django.http import HttpResponseRedirect
from django.shortcuts import get_object_or_404, redirect, render
from django.urls import reverse
2020-05-05 23:50:48 +08:00
from django.utils.translation import gettext_lazy as _
from common.config import *
from management.models import Announcement
2020-10-22 21:45:05 +02:00
from mastodon.api import *
from .account import *
from .data import *
from .forms import ReportForm
from .models import Preference, Report, User
2020-05-01 22:46:15 +08:00
2022-05-12 08:14:05 -04:00
def render_user_not_found(request):
2023-07-07 16:54:15 -04:00
sec_msg = _("😖哎呀这位用户好像还没有加入本站快去联邦宇宙呼唤TA来注册吧")
msg = _("未找到该用户")
2022-05-12 08:14:05 -04:00
return render(
request,
"common/error.html",
2022-05-12 08:14:05 -04:00
{
"msg": msg,
"secondary_msg": sec_msg,
},
2022-05-12 08:14:05 -04:00
)
2022-12-27 14:52:03 -05:00
def render_user_blocked(request):
2023-07-07 16:54:15 -04:00
msg = _("没有访问该用户主页的权限")
2022-12-27 14:52:03 -05:00
return render(
request,
"common/error.html",
2022-12-27 14:52:03 -05:00
{
"msg": msg,
},
2022-12-27 14:52:03 -05:00
)
2023-07-07 02:02:48 -04:00
@login_required
def follow(request, user_name):
if request.method != "POST":
raise BadRequest()
user = User.get(user_name)
if request.user.follow(user):
2023-07-07 16:54:15 -04:00
return render(request, "users/profile_actions.html", context={"user": user})
2023-07-07 02:02:48 -04:00
else:
raise BadRequest()
@login_required
def unfollow(request, user_name):
if request.method != "POST":
raise BadRequest()
user = User.get(user_name)
if request.user.unfollow(user):
2023-07-07 16:54:15 -04:00
return render(request, "users/profile_actions.html", context={"user": user})
else:
raise BadRequest()
@login_required
def mute(request, user_name):
if request.method != "POST":
raise BadRequest()
user = User.get(user_name)
if request.user.mute(user):
return render(request, "users/profile_actions.html", context={"user": user})
else:
raise BadRequest()
@login_required
def unmute(request, user_name):
if request.method != "POST":
raise BadRequest()
user = User.get(user_name)
if request.user.unmute(user):
return render(request, "users/profile_actions.html", context={"user": user})
else:
raise BadRequest()
@login_required
def block(request, user_name):
if request.method != "POST":
raise BadRequest()
user = User.get(user_name)
if request.user.block(user):
return render(request, "users/profile_actions.html", context={"user": user})
else:
raise BadRequest()
@login_required
def unblock(request, user_name):
if request.method != "POST":
raise BadRequest()
user = User.get(user_name)
if request.user.unblock(user):
return render(request, "users/profile_actions.html", context={"user": user})
2023-07-07 02:02:48 -04:00
else:
raise BadRequest()
@login_required
def follow_locked(request, user_name):
user = User.get(user_name)
return render(request, "users/follow_locked.html", context={"user": user})
2021-02-17 15:08:16 +01:00
@login_required
def set_layout(request):
if request.method == "POST":
layout = json.loads(request.POST.get("layout"))
2023-04-19 22:31:27 -04:00
if request.POST.get("name") == "profile":
request.user.preference.profile_layout = layout
request.user.preference.save(update_fields=["profile_layout"])
return redirect(request.user.url)
2023-04-19 22:31:27 -04:00
elif request.POST.get("name") == "discover":
request.user.preference.discover_layout = layout
request.user.preference.save(update_fields=["discover_layout"])
return redirect(reverse("catalog:discover"))
raise BadRequest()
2021-02-17 15:08:16 +01:00
2021-02-15 21:27:50 +01:00
2020-05-05 23:50:48 +08:00
@login_required
def report(request):
if request.method == "GET":
user_id = request.GET.get("user_id")
2020-05-07 18:00:39 +08:00
if user_id:
user = get_object_or_404(User, pk=user_id)
form = ReportForm(initial={"reported_user": user})
2020-05-07 18:00:39 +08:00
else:
form = ReportForm()
2020-05-05 23:50:48 +08:00
return render(
request,
"users/report.html",
2020-05-05 23:50:48 +08:00
{
"form": form,
},
2020-05-05 23:50:48 +08:00
)
elif request.method == "POST":
form = ReportForm(request.POST, request.FILES)
2020-05-05 23:50:48 +08:00
if form.is_valid():
form.instance.is_read = False
form.instance.submit_user = request.user
form.save()
dw = settings.DISCORD_WEBHOOKS.get("user-report")
if dw:
webhook = SyncWebhook.from_url(dw) # type: ignore
webhook.send(
f"New report from {request.user} about {form.instance.reported_user} : {form.instance.message}"
)
return redirect(reverse("common:home"))
2020-05-05 23:50:48 +08:00
else:
return render(
request,
"users/report.html",
2020-05-05 23:50:48 +08:00
{
"form": form,
},
2020-05-05 23:50:48 +08:00
)
else:
2023-02-04 00:45:17 -05:00
raise BadRequest()
2020-05-05 23:50:48 +08:00
@login_required
def manage_report(request):
if not request.user.is_staff:
raise PermissionDenied()
if request.method == "GET":
2020-05-05 23:50:48 +08:00
reports = Report.objects.all()
for r in reports.filter(is_read=False):
2020-07-03 15:36:23 +08:00
r.is_read = True
2020-05-05 23:50:48 +08:00
r.save()
return render(
request,
"users/manage_report.html",
2020-05-05 23:50:48 +08:00
{
"reports": reports,
},
2020-05-05 23:50:48 +08:00
)
else:
2023-02-04 00:45:17 -05:00
raise BadRequest()
@login_required
def mark_announcements_read(request):
if request.method == "POST":
try:
request.user.read_announcement_index = Announcement.objects.latest("pk").pk
request.user.save(update_fields=["read_announcement_index"])
except ObjectDoesNotExist:
# when there is no annoucenment
pass
return HttpResponseRedirect(request.META.get("HTTP_REFERER"))