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
