def view_add(request):
user = User()
errCode = user.add(request.POST['user'], request.POST['password'])
data = { 'errCode': errCode }
if (errCode == User.SUCCESS):
newUser = User.objects.get(user=request.POST['user'], password=request.POST['password'])
data['count'] = newUser.count
if 'count' not in data:
return render(request, 'index.html', {'error_message': user.errorMessage(errCode)})
return render(request, 'login.html', {'user': request.POST['user'], 'count': newUser.count})
def user_add(request):
"""
Add User
"""
if request.content_type != 'application/json':
return Response("Unrecognized request", status = status.HTTP_404_NOT_FOUND)
user = User()
errCode = user.add(request.DATA['user'], request.DATA['password'])
data = { 'errCode': errCode }
if (errCode == User.SUCCESS):
newUser = User.objects.get(user=request.DATA['user'], password=request.DATA['password'])
data['count'] = newUser.count
return Response(data)
def user_login(request):
"""
Login User
"""
if request.content_type != 'application/json':
# fix this up at some point
return Response("Unrecognized request", status = status.HTTP_404_NOT_FOUND)
user = User()
errCode = user.login(request.DATA['user'], request.DATA['password'])
data = { 'errCode': errCode }
if (errCode == User.SUCCESS):
thisUser = User.objects.get(user=request.DATA['user'], password=request.DATA['password'])
data['count'] = thisUser.count
return Response(data)
