def test_updateUser(self):
user_service = UserService()
role_list = []
self.assertEqual(
user_service.updateUser('update1', 'amit',
'*****@*****.**', 'amit',
role_list), 'user_updated')
def create_user(request):
msg = ''
msg_level = ''
if request.method == 'POST':
form = CreateUserForm(request.POST)
if form.is_valid():
try:
user_service = UserService()
user_service.addUser(form.cleaned_data['id'],
form.cleaned_data['full_name'],
form.cleaned_data['email_address'],
form.cleaned_data['password'])
msg_level = 'alert-success'
msg = 'The User Successfully Added.'
except:
msg_level = 'alert-danger'
msg = 'There is some error while creating the user.'
else:
initialData = {'form': form}
csrfContext = RequestContext(request, initialData)
return render_to_response('account/register.html', csrfContext)
return render(request, 'account/success_message.html', {
'msg': msg,
'msg_level': msg_level
})
def post(self, request, *args, **kwargs):
form = UserForm(data=request.POST)
if form.is_valid():
user_service = UserService()
user_service.create_user_object(form)
user_service.log_user_in(request, form.instance.username, form.instance.password)
return HttpResponseRedirect(reverse('songthread_list'))
else:
return render_to_response(self.template_name, {'form': form}, context_instance=RequestContext(request))
def get(self, request):
"""
Get user including detail information
:param request:
:return:
"""
token = request.META.get("TOKEN")
# TODO get user_id by token
user_id = 1
user_service = UserService()
user_detail = user_service.get_user_detail_by_id(user_id)
return ResponseHelper.build_success(model_to_dict(user_detail))
def delete(self, request):
"""
Delete user
:param request:
:return:
"""
token = request.META.get("TOKEN")
# TODO get user_id by token
user_id = 1
user_service = UserService()
user_service.delete_user(user_id)
# TODO logout user
return ResponseHelper.build_success()
def recover_password(request):
msg = ''
msg_level = ''
if request.method == 'POST':
id = request.POST.get('id', '')
email_address = request.POST.get('email_address', '')
if (id != '' or email_address != ''):
try:
if (id != ''):
user = User.objects.get(username=id)
user_service = UserService()
user_service.recoverPassword(user)
else:
user = User.objects.filter(email=email_address)
user_service = UserService()
user_service.sendMail(user[0])
msg_level = 'alert-success'
msg = 'Recovery mail has been sent to your registered email address.'
except (ObjectDoesNotExist, MultipleObjectsReturned,
Exception) as e:
print "Django Error ", e
print 'user not exist'
msg_level = 'alert-danger'
msg = 'User ID/Email address does not registered with Kanga. Please Enter Correct User ID/Email address'
else:
return HttpResponse('Please Enter Correct User ID/Email Address')
return render(request, 'account/success_message.html', {
'msg': msg,
'msg_level': msg_level
})
def activate_account(request):
msg = ''
if request.method == 'GET':
hashcode = request.GET.get('hashcode', '')
print hashcode
try:
user_service = UserService()
user_service.activateAccount(hashcode)
msg = 'User has been activated.'
except (ObjectDoesNotExist, MultipleObjectsReturned, Exception) as e:
print "Django Error ", e
msg = 'Something wrong while activating user'
return HttpResponse('Something wrong while activating user')
data = {'message': msg}
form = SuccessMessageForm(data)
return render(request, 'account/success_message.html', {'form': form})
def edit_user(request, query_pk):
if request.method == 'POST':
username = request.POST.get('username')
full_name = request.POST.get('full_name')
email_address = request.POST.get('email_address')
password = request.POST.get('password', '')
role_list = request.POST.getlist('selected_role')
try:
user_service = UserService()
user_service.updateUser(username, full_name, email_address,
password, role_list)
except:
return HttpResponse(
'There is some error while editing user details')
if request.method == 'GET':
user = User.objects.get(username=query_pk)
print user.groups.all()
data = {
'username': str(query_pk),
'full_name': str(user.first_name),
'email_address': str(user.email)
}
form = EditUserForm(data)
auth_service = AuthService()
choices = auth_service.getAllRole()
form.fields['role'].choices = choices
form.fields['selected_role'].choices = auth_service.fetch_user_role(
query_pk)
form.fields['username'].widget.attrs['readonly'] = True
form.fields['full_name'].widget.attrs['class'] = 'form-control'
form.fields['email_address'].widget.attrs['class'] = 'form-control'
form.fields['full_name'].widget.attrs['class'] = 'form-control'
form.fields['role'].widget.attrs['class'] = 'form-control'
form.fields['selected_role'].widget.attrs['class'] = 'form-control'
form.fields['username'].widget.attrs['class'] = 'form-control'
form.fields['password'].widget.attrs['class'] = 'form-control'
form.fields['confirm_password'].widget.attrs['class'] = 'form-control'
return render(request, 'account/edituser.html', {'form': form})
href = reverse('account:user-list')
return HttpResponseRedirect(href)
def password_change(request):
msg = ''
if request.method == 'POST':
new_password = request.POST.get('new_password', '')
hashcode = request.POST.get('hashcode', '')
try:
user_service = UserService()
user_service.changePassword(hashcode, new_password)
msg = 'Password has been changed.'
except (ObjectDoesNotExist, MultipleObjectsReturned, Exception) as e:
print "Django Error ", e
msg = 'Please Enter Correct URL in Browser'
data = {'message': msg}
form = SuccessMessageForm(data)
return render(request, 'account/success_message.html', {'form': form})
return render(request, 'account/passwordchange.html')
def post(self, request):
"""
Register user
:param request:
:return:
"""
data = json.loads(request.body)
email = data.get("email")
email_code = data.get("email_code")
password = data.get("password")
user_service = UserService()
try:
count = user_service.register_user(email, email_code, password)
except RegisterException as e:
return ResponseHelper.build_fail(e.reason)
if count < 1:
return ResponseHelper.build_fail(constants.REGISTER_FAIL)
return ResponseHelper.build_fail(constants.REGISTER_SUCCESS)
def login(request):
redirect_to = request.POST.get('return_uri',
request.GET.get('return_uri', ''))
#Checking if the user is already authenticated and in session
if request.user.is_authenticated():
return HttpResponseRedirect('/account/index.html')
else:
if request.method == 'POST':
id = request.POST.get('username', '')
password = request.POST.get('password', '')
user = authenticate(username=id, password=password)
if user is not None:
if user.is_active:
auth_login(request, user)
permission = user.get_group_permissions()
print permission
user_service = UserService()
user_serach_index = user_service.get_user_serach_index(id)
# print 'user_serach_index ----> ',user_serach_index
request.session["permission"] = str(permission)
request.session["user_serach_index"] = str(
user_serach_index)
return HttpResponseRedirect(redirect_to)
else:
messages.add_message(request, messages.ERROR,
'User is not yet active.')
return render(request, 'account/login_two_columns.html')
else:
messages.add_message(request, messages.ERROR,
'Invalid username or password.')
return render(request, 'account/login_two_columns.html')
if request.method == 'GET':
redirect_to = request.GET.get('return_uri', '')
return render(request, 'account/login_two_columns.html',
{'return_uri': redirect_to})
return render(request, 'account/login_two_columns.html')
def recover_password(request):
msg = ''
msg_level = ''
if request.method == 'POST':
f = ForgotPasswordForm(request.POST or None)
print 'form-->', f
if f.is_valid():
cd = f.cleaned_data
print cd
try:
if not cd['username']:
user = User.objects.get(username=cd['username'])
user_service = UserService()
user_service.recoverPassword(user)
else:
user = User.objects.filter(email=cd['email'])
user_service = UserService()
user_service.recoverPassword(user)
# user_service.sendMail(user[0])
msg_level = 'alert-success'
msg = 'Recovery mail has been sent to your registered email address.'
except (ObjectDoesNotExist, MultipleObjectsReturned,
Exception) as e:
print "Django Error ", e
print 'user not exist'
msg_level = 'alert-danger'
msg = 'User ID/Email address does not registered with Kanga. Please Enter Correct User ID/Email address'
else:
print 'here'
return render(request, 'account/forgotpassword.html', {
'form': f,
'invalid': True
})
# return HttpResponse('Please Enter Correct User ID/Email Address')
return render(request, 'account/success_message.html', {
'msg': msg,
'msg_level': msg_level
})
def test_recover_password(self):
user = User.objects.get(username='******')
user_service = UserService()
self.assertEqual(user_service.recoverPassword(user),
'recover_password_successful')
def get_all_user_permission(request):
user_service = UserService()
l = user_service.get_user_roles('aaa1')
print l
return HttpResponse(l)
def setUp(self):
user_service = UserService()
user_service.addUser('update1', 'amit', '*****@*****.**',
'amit')
def grpc_handlers(server):
account_pb2_grpc.add_UserControllerServicer_to_server(
UserService.as_servicer(), server)
def test_addUser(self):
user_service = UserService()
self.assertEqual(
user_service.addUser('test1', 'amit', '*****@*****.**',
'amit'), 'user_added')
def test_getuser_permission(self):
user_service = UserService()
permisssion_list = Set()
self.assertEqual(user_service.get_user_permission('update1'),
permisssion_list)
def form_valid(self, form):
user_service = UserService()
user_service.create_user_object(form)
user_service.log_user_in(self.request, form.instance.username, form.instance.password)
return HttpResponseRedirect(reverse('songthread_list'))
