def post(self, request): try: user = self.get_object(request.POST['email']) if user.count() == 0: return render( request, self.template_name, { 'form': forms.LoginForm(), 'error': '1', 'msg': 'User does not exist. SignUp to proceed' }) else: user = user[0] r = utils.get_user(user.db_name, user.email) pass2 = r.get('password') if check_password(request.POST['password'], pass2) == True: request.session.update({ 'email': user.email, 'type': r.get('type') }) return redirect('person:Dashboard') else: return render( request, self.template_name, { 'form': forms.LoginForm(), 'error': '1', 'msg': 'Incorrect Password' }) except: return render( request, self.template_name, { 'form': forms.LoginForm(), 'error': '1', 'msg': 'Incorrect Submission' })
def get(self, request): user = self.get_object(request.GET.get('id')) if user.count() > 0: user = user[0] r = utils.get_user(user.db_name, user.email) if check_password(request.GET.get('token'), r.get('token')) == True: return render(request, self.template_name, {'form': forms.PasswordResetForm()}) else: print('PAGE NOT FOUND') else: print('PAGE NOT FOUND')
def clean_email(self): email = self.cleaned_data.get('email') user = self.get_object(email) if user.count() > 0: user = user[0] user = utils.get_user(user.db_name, user.email) if user.get('type') == 'S': return email else: raise forms.ValidationError('User not a service provider') else: raise forms.ValidationError('User does not exist')
def get_user(self, email): user = models.UserMetaData.objects.filter(email=email) user = user[0] db_name = user.db_name return utils.get_user(db_name, email)
def post(self, request, id): userInfo = models.UserMetaData.objects.get( email=request.session.get('email')) db_name = userInfo.db_name if id[0] == 'H': booking = utils.get_hotel_booking_by_id(id) hotel = models.ServiceMetaData.objects.get( id=booking.get('service_id')) form = forms.DeleteBookingForm(request.POST) if form.is_valid(): user = utils.get_user(db_name, request.session.get('email')) if check_password(form.cleaned_data.get('password'), user.get('password')) == True: r = utils.delete_hotel_booking(id) if r == 200: return redirect('person:Dashboard') else: return render( request, self.template_name, { 'form': forms.DeleteBookingForm(), 'error': '1', 'msg': 'Network Error', 'booking': booking, 'hotel': hotel, 'type': get_type(request) }) else: return render( request, self.template_name, { 'form': forms.DeleteBookingForm(), 'error': '1', 'msg': 'Incorrect Password', 'booking': booking, 'hotel': hotel, 'type': get_type(request) }) else: return render( request, self.template_name, { 'form': forms.DeleteBookingForm(), 'error': '1', 'msg': 'Fields cannot be empty', 'booking': booking, 'hotel': hotel, 'type': get_type(request) }) if id[0] == 'B': booking = utils.get_bus_booking_by_id(id) hotel = models.ServiceMetaData.objects.get( id=booking.get('service_id')) form = forms.DeleteBookingForm(request.POST) if form.is_valid(): user = utils.get_user(db_name, request.session.get('email')) if check_password(form.cleaned_data.get('password'), user.get('password')) == True: r = utils.delete_bus_booking(id) if r == 200: return redirect('person:Dashboard') else: return render( request, self.template_name_1, { 'form': forms.DeleteBookingForm(), 'error': '1', 'msg': 'Network Error', 'booking': booking, 'bus': bus, 'type': get_type(request) }) else: return render( request, self.template_name_1, { 'form': forms.DeleteBookingForm(), 'error': '1', 'msg': 'Incorrect Password', 'booking': booking, 'bus': bus, 'type': get_type(request) }) else: return render( request, self.template_name_1, { 'form': forms.DeleteBookingForm(), 'error': '1', 'msg': 'Fields cannot be empty', 'booking': booking, 'bus': bus, 'type': get_type(request) })
def test_that_query_user_is_123_after_inserting_123_user(): insert_user('123', '123456789') user = get_user('123') assert user['user'] == '123'
def test_that_query_user_is_not_none_after_inserting_a_given_user(): insert_user('123', '123456789') user = get_user('123') assert user != None
def test_that_query_user_is_none(): user = get_user('123') assert user == None