exp_portal/users.py

# Licensed to the Apache Software Foundation (ASF) under one or more
# contributor license agreements.  See the NOTICE file distributed with
# this work for additional information regarding copyright ownership.
# The ASF licenses this file to You under the Apache License, Version 2.0
# (the "License"); you may not use this file except in compliance with
# the License.  You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from django.shortcuts import render, redirect
from op_tasks.models import UserProfile, Product, Dataset, OpTask, TaskListItem, Experiment
from django.contrib.auth.decorators import login_required
from django.contrib.auth.models import User
from django.http import JsonResponse

@login_required(login_url='/tasking/login')
def view_users(request):
	userprofiles = UserProfile.objects.all().order_by('-user__last_login')
	return render(request, 'users.html', {'userprofiles': userprofiles})


@login_required(login_url='/tasking/login')
def manage_users(request):
	userprofiles = UserProfile.objects.all().order_by('-user__last_login')
	return render(request, 'users.html', {'userprofiles': userprofiles})


@login_required(login_url='/tasking/login')
def edit_user(request, userprofilepk):
	userprofile = UserProfile.objects.get(id=userprofilepk)
	experiment = Experiment.objects.get(name=request.POST['experiment_name'])
	userprofile.experiment = experiment
	userprofile.save()

	return redirect('exp_portal:view_users')


@login_required(login_url='/tasking/login')
def add_user(request):
	experiments = Experiment.objects.all()
	products = Product.objects.all()
	return render(request, 'add_user.html', {'experiments': experiments, 'products':products})


@login_required(login_url='/tasking/login')
def delete_user(request, userprofilepk):
	userprofile = UserProfile.objects.get(id=userprofilepk)
	user = userprofile.user
	usertasks = userprofile.tasklistitem_set.all()

	for tasklistitem in usertasks:
		tasklistitem.delete()

	user.delete()
	userprofile.delete()

	return redirect('exp_portal:view_users')


@login_required(login_url='/tasking/login')
def new_user(request):
	user = User(username=request.POST['username'])
	user.set_password(request.POST['password_1'])
	user.email = user.username
	user.save()

	userprofile = UserProfile()
	userprofile.user = user
	experiment = Experiment.objects.get(name=request.POST['experiment_name'])
	userprofile.experiment = experiment
	userprofile.save()

	# logic for assigning tasks
	if request.POST['product_name'] == 'all':
		# all products all tasks buffet style
		index = 0
		datasets = Dataset.objects.all()
		for dataset in datasets:
			products = dataset.product_set.all()
			for product in products:
				tasks = dataset.optask_set.all()
				tasks = tasks.filter(is_active=True)
				for task in tasks:
					newtasklistitem = TaskListItem()
					newtasklistitem.userprofile = userprofile
					newtasklistitem.op_task = task
					newtasklistitem.product = product
					newtasklistitem.index = index
					index = index + 1
					newtasklistitem.task_active = True
					newtasklistitem.save()
	else:
		# single product - task ordering assumed
		product = Product.objects.get(name=request.POST['product_name'])
		if str(request.POST['taskorder']) == 'b':
			print 'true'
			tasks = product.dataset.optask_set.all().order_by('id').reverse()
		else:
			tasks = product.dataset.optask_set.all().order_by('id')

		tasks = tasks.filter(is_active=True)

		index = 0
		for task in tasks:
			newtasklistitem = TaskListItem()
			newtasklistitem.userprofile = userprofile
			newtasklistitem.op_task = task
			newtasklistitem.product = product
			newtasklistitem.index = index
			# if index == 0 or experiment.sequential_tasks == False:
			# 	newtasklistitem.task_active = True
			# else:
			# 	newtasklistitem.task_active = False
			index = index + 1
			newtasklistitem.save()


	return redirect('exp_portal:view_users')


@login_required(login_url='/tasking/login')
def user_added(request):
	return render(request, 'user_added.html')


@login_required(login_url='/tasking/login')
def view_user_tasks(request, profile):
	userprofile = UserProfile.objects.all().filter(user_hash=profile)[0]
	usertasks = userprofile.tasklistitem_set.all()
	experiments = Experiment.objects.all()
	
	datasets = Dataset.objects.all()
	count = 0
	tasklistitems = []
	for dataset in datasets:
		for product in dataset.product_set.all():
			for task in dataset.optask_set.all():
				count = count + 1

	return render(request, 'user_tasks.html', {'userprofile': userprofile, 'experiments':experiments})


@login_required(login_url='/tasking/login')
def add_user_task(request, userpk):
	userprofile = UserProfile.objects.get(id=userpk)
	datasets = Dataset.objects.all()
	products = Product.objects.all()
	tasks = OpTask.objects.all()
	return render(request, 'add_user_task.html', {'userprofile':userprofile, 
		'datasets': datasets, 'products': products, 'tasks': tasks})


@login_required(login_url='/tasking/login')
def update_user_tasks(request, userpk, datasetpk, productpk, taskpk):
	dataset = Dataset.objects.get(id=datasetpk)
	userprofile = UserProfile.objects.get(id=userpk)
	index = userprofile.tasklistitem_set.count()

	newtasklistitem = TaskListItem()
	newtasklistitem.userprofile = userprofile
	newtasklistitem.op_task = OpTask.objects.get(id=taskpk)
	newtasklistitem.product = Product.objects.get(id=productpk)
	newtasklistitem.index = index
	newtasklistitem.task_active = True
	newtasklistitem.save()

	return redirect('exp_portal:add_user_task', userpk)

@login_required(login_url='/tasking/login')
def view_users_experiment(request, experiment_name):
	experiment = Experiment.objects.get(name=experiment_name)
	userprofiles = experiment.userprofile_set.all()
	user_hashes = []
	for userprofile in userprofiles:
		user_hashes.append(userprofile.user_hash)
	response = JsonResponse(user_hashes, safe=False)
	return response