From 489717ad7716f9453d46d6ebafc587622f5988e7 Mon Sep 17 00:00:00 2001
From: Egor Matveev <emmtvv@icloud.com>
Date: Sat, 20 Nov 2021 23:17:34 +0300
Subject: [PATCH] refactor

---
 Main/views/SendCodeView.py | 2 ++
 Main/views/__init__.py     | 1 -
 Sprint/urls.py             | 4 +---
 3 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/Main/views/SendCodeView.py b/Main/views/SendCodeView.py
index 6d732d0..9ce11c2 100644
--- a/Main/views/SendCodeView.py
+++ b/Main/views/SendCodeView.py
@@ -25,6 +25,8 @@ class SendCodeView(BaseView):
             return {"success": False, "message": "Пользователя с таким именем не существует"}
         code = int(self.request.POST["code"])
         if code == user.userinfo.code:
+            user.userinfo.code = None
+            user.userinfo.save()
             login(self.request, user)
             return {"success": True, "message": "Успешно"}
         else:
diff --git a/Main/views/__init__.py b/Main/views/__init__.py
index ba31df6..56db9a6 100644
--- a/Main/views/__init__.py
+++ b/Main/views/__init__.py
@@ -1,5 +1,4 @@
 from Main.views.EnterView import EnterView
-from Main.views.RegisterView import RegisterView
 from Main.views.MainView import MainView
 from Main.views.TasksView import TasksView
 from Main.views.AccountView import AccountView
diff --git a/Sprint/urls.py b/Sprint/urls.py
index dab40e9..d2d6e93 100644
--- a/Sprint/urls.py
+++ b/Sprint/urls.py
@@ -1,12 +1,10 @@
-from django.conf.urls.static import static
 from django.contrib import admin
 from django.urls import path
+
 from Main.views import *
-from Sprint import settings
 
 urlpatterns = [
     path("enter", EnterView.as_view()),
-    path("register", RegisterView.as_view()),
     path("rating", RatingView.as_view()),
     path("tasks", TasksView.as_view()),
     path("account", AccountView.as_view()),