29 lines
880 B
Python
29 lines
880 B
Python
from django.contrib.auth import login
|
|
from django.contrib.auth.models import User
|
|
|
|
from SprintLib.BaseView import BaseView
|
|
|
|
|
|
class SetUsernameView(BaseView):
|
|
endpoint = "set_username"
|
|
view_file = "set_username.html"
|
|
required_login = False
|
|
|
|
def get(self):
|
|
if not self.request.user.is_authenticated:
|
|
return "/"
|
|
|
|
def post(self):
|
|
if not self.request.user.is_authenticated:
|
|
return "/"
|
|
user = User.objects.filter(username=self.request.POST['username']).first()
|
|
if user is None:
|
|
self.request.user.username = self.request.POST['username']
|
|
self.request.user.userinfo.verified = True
|
|
self.request.user.save()
|
|
self.request.user.userinfo.save()
|
|
login(self.request, self.request.user)
|
|
return "/"
|
|
else:
|
|
return "/set_username"
|