Advertisement
Guest User

Untitled

a guest
Jan 18th, 2016
102
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 5.45 KB | None | 0 0
  1. from django.http import HttpResponseRedirect
  2. from django.shortcuts import render
  3. from App.models import Ranks
  4. from Libs.UserLib import UserLib
  5. from django.contrib.auth import authenticate, login, logout
  6. from django.contrib.auth.models import User
  7. from django.views.generic import View
  8.  
  9. # This is description for this function. Thats very very long
  10. # use:
  11. # You can use that for create or edit user
  12.  
  13.  
  14. class UserLogin(View):
  15.     def get(self, request):
  16.         if request.user.is_authenticated():
  17.             return HttpResponseRedirect('/profile')
  18.  
  19.         context = {
  20.             'Name': "Kamil",
  21.         }
  22.  
  23.         return render(request, 'login.html', context)
  24.  
  25.     def post(self, request):
  26.         if not request.user.is_authenticated():
  27.             return HttpResponseRedirect('/login')
  28.  
  29.         username = request.POST['username']
  30.         password = request.POST['password']
  31.  
  32.         user = authenticate(username=username, password=password)
  33.  
  34.         if user is not None:
  35.             if user.is_active:
  36.                 login(request, user)
  37.                 return HttpResponseRedirect('/profile/')
  38.             else:
  39.                 return HttpResponseRedirect('/login')
  40.  
  41.         return render(request, 'login.html')
  42.  
  43.  
  44. # This is description for this function. Thats very very long
  45. # use:
  46. # You can use that for create or edit user
  47.  
  48.  
  49. class UserRegister(View):
  50.     def __init__(self):
  51.         self.username = None
  52.         self.password = None
  53.         self.email = None
  54.  
  55.     def get(self, request):
  56.         if request.user.is_authenticated():
  57.             return HttpResponseRedirect('/profile')
  58.  
  59.         context = {
  60.             'Name': "Kamil",
  61.         }
  62.  
  63.         return render(request, 'register.html', context)
  64.  
  65.     def post(self, request):
  66.         if not request.user.is_authenticated():
  67.             return HttpResponseRedirect('/login')
  68.  
  69.         self.username = request.POST['username']
  70.         self.password = request.POST['password']
  71.         self.email = request.POST['email']
  72.  
  73.  
  74.         return render(request, 'register.html')
  75.  
  76.     def create_account(self):
  77.         error = False
  78.  
  79.         if not self.email or not self.username or not self.password:
  80.             error = True
  81.         if self.email \
  82.                 and User.objects.filter(email=self.email).count() or self.username \
  83.                 and User.objects.filter(username=self.username).count() > 0:
  84.             error = True
  85.  
  86.         if error is False:
  87.             user = User.objects.create_user(username=self.username, email=self.email, password=self.password)
  88.             user.save()
  89.  
  90.             getRank = Ranks.objects.get(LeadsRequired=0)
  91.  
  92.             UserLib.CreateUser(self, {'Username': self.username,
  93.                                          'Name': 'null',
  94.                                          'Surname': 'null',
  95.                                          'Pesel': 'null',
  96.                                          'Bank': 'null',
  97.                                          'Rank': getRank.id,
  98.                                          'Money': 0})
  99.  
  100. # This is description for this function. Thats very very long
  101. # use:
  102. # You can use that for create or edit user
  103.  
  104.  
  105. class UserLogout(View):
  106.     def get(self, request):
  107.         if not request.user.is_authenticated():
  108.             return HttpResponseRedirect('/login')
  109.  
  110.         logout(request)
  111.         return HttpResponseRedirect('/login')
  112.  
  113.  
  114. # This is description for this function. Thats very very long
  115. # use:
  116. # You can use that for create or edit user
  117.  
  118.  
  119. class UserProfile(View):
  120.     def get(self, request):
  121.         if not request.user.is_authenticated():
  122.             return HttpResponseRedirect('/login')
  123.  
  124.         context = {
  125.             'Name': request.user.username,
  126.         }
  127.  
  128.         return render(request, 'profile.html', context)
  129.  
  130.  
  131. # This is description for this function. Thats very very long
  132. # use:
  133. # You can use that for create or edit user
  134.  
  135.  
  136. class UserEdit(View):
  137.     def __init__(self):
  138.         self.name = None
  139.         self.surname = None
  140.         self.pesel = None
  141.         self.bank = None
  142.         self.password = None
  143.         self.email = None
  144.  
  145.     def get(self, request):
  146.         if not request.user.is_authenticated():
  147.             return HttpResponseRedirect('/login')
  148.  
  149.         context = {
  150.             'Name': request.user.username,
  151.         }
  152.  
  153.         return render(request, 'edit.html', context)
  154.  
  155.     def post(self, request):
  156.         if not request.user.is_authenticated():
  157.             return HttpResponseRedirect('/login')
  158.  
  159.         self.name = request.POST['name']
  160.         self.surname = request.POST['surname']
  161.         self.pesel = request.POST['pesel']
  162.         self.bank = request.POST['bank']
  163.         self.password = request.POST['password']
  164.         self.email = request.POST['email']
  165.  
  166.         self.change_data(request.user.username)
  167.  
  168.         return render(request, 'edit.html')
  169.  
  170.     def change_data(self, username):
  171.         # change password
  172.         if self.password is not "":
  173.             u = User.objects.get(username=username)
  174.             u.set_password(self.password)
  175.             u.save()
  176.  
  177.         # change password
  178.         u = User.objects.get(username=username)
  179.         u.email = self.email
  180.         u.save()
  181.  
  182.         # change to new data
  183.         ul = UserLib()
  184.         ul.SetUserData(username, {
  185.             'Name': self.name,
  186.             'Surname': self.surname,
  187.             'Pesel': self.pesel,
  188.             'Bank': self.bank,
  189.         })
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement