From 611cc0355e5112657920e8d65f8f13aacf47c913 Mon Sep 17 00:00:00 2001
From: doubaniux <goodsir@vivaldi.net>
Date: Sun, 25 Oct 2020 23:40:58 +0100
Subject: [PATCH] Add bad request response

---
 users/templates/users/register.html | 2 +-
 users/views.py                      | 5 ++++-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/users/templates/users/register.html b/users/templates/users/register.html
index 5f479aa0..de999594 100644
--- a/users/templates/users/register.html
+++ b/users/templates/users/register.html
@@ -26,7 +26,7 @@
                 BTW欧盟惯例本站使用了Cookie，请理解！
             </p>
             <p>
-                此外里瓣书影音现处于“公开阿尔法测试”阶段，您的数据存在丢失的可能。使用过程中遇到的问题或者Bug欢迎向<a href="https://donotban.com/@whitiewhite">作者</a>提出。
+                此外NiceDB书影音现处于“公开阿尔法测试”阶段，您的数据存在丢失的可能。使用过程中遇到的问题或者Bug欢迎向<a href="https://donotban.com/@whitiewhite">作者</a>提出。
             </p>
             <form action="{% url 'users:register' %}" method="post">
                 {% csrf_token %}
diff --git a/users/views.py b/users/views.py
index df11c3b7..097ba9da 100644
--- a/users/views.py
+++ b/users/views.py
@@ -34,7 +34,10 @@ def OAuth2_login(request):
         site = request.COOKIES.get('mastodon_domain')
 
         # Network IO
-        token = obtain_token(site, request, code)
+        try:
+            token = obtain_token(site, request, code)
+        except ObjectDoesNotExist:
+            return HttpResponseBadRequest("Mastodon site not registered")
         if token:
             # oauth is completed when token aquired
             user = authenticate(request, token=token, site=site)