Add "open_registration" setting option to control whether registratio…

…n is enabled.

account/urls.py

@@ -0,0 +1,31 @@
+from django.urls import path, re_path, include
+from django.views.generic import TemplateView
+from dj_rest_auth.registration.views import VerifyEmailView, ResendEmailVerificationView
+from .views import RegistrationView
+
+urlpatterns = [
+    path('', include('dj_rest_auth.urls')),
+    path('registration/', RegistrationView.as_view(), name='rest_register'),
+    path('verify-email/', VerifyEmailView.as_view(), name='rest_verify_email'),
+    path('resend-email/', ResendEmailVerificationView.as_view(), name="rest_resend_email"),
+
+    # This url is used by django-allauth and empty TemplateView is
+    # defined just to allow reverse() call inside app, for example when email
+    # with verification link is being sent, then it's required to render email
+    # content.
+
+    # account_confirm_email - You should override this view to handle it in
+    # your API client somehow and then, send post to /verify-email/ endpoint
+    # with proper key.
+    # If you don't want to use API on that step, then just use ConfirmEmailView
+    # view from:
+    # django-allauth https://github.com/pennersr/django-allauth/blob/master/allauth/account/views.py
+    re_path(
+        r'^account-confirm-email/(?P<key>[-:\w]+)/$', TemplateView.as_view(),
+        name='account_confirm_email',
+    ),
+    path(
+        'account-email-verification-sent/', TemplateView.as_view(),
+        name='account_email_verification_sent',
+    ),
+]

account/views.py

@@ -1,4 +1,17 @@
-from django.shortcuts import render
+from rest_framework.response import Response
+from rest_framework import status
+from dj_rest_auth.registration.views import RegisterView
+from chat.models import Setting
 
-# Create your views here.
-from dj_rest_auth.views import LoginView
+
+class RegistrationView(RegisterView):
+    def create(self, request, *args, **kwargs):
+        try:
+            open_registration = Setting.objects.get(name='open_registration').value == 'True'
+        except Setting.DoesNotExist:
+            open_registration = True
+
+        if open_registration:
+            return super().create(request, *args, **kwargs)
+        else:
+            return Response({'detail': 'Registration is not yet open.'}, status=status.HTTP_403_FORBIDDEN)

chatgpt_ui_server/urls.py

@@ -21,7 +21,6 @@
     path('api/chat/', include('chat.urls')),
     path('api/conversation/', conversation, name='conversation'),
     path('api/gen_title/', gen_title, name='gen_title'),
-    path('api/account/', include('dj_rest_auth.urls')),
-    path('api/account/registration/', include('dj_rest_auth.registration.urls')),
+    path('api/account/', include('account.urls')),
     path('admin/', admin.site.urls),
 ]