from datetime import datetime

from django import shortcuts

from django import http

from django.contrib.auth.decorators import login_required

from django.contrib.auth.models import User

from django.core.context_processors import csrf

from django.core.urlresolvers import reverse

from django.template import RequestContext

from django.utils import simplejson as json

from tagging.models import Tag

from pytask.views import show_msg

from pytask.profile import models as profile_models

from pytask.taskapp import forms as taskapp_forms

from pytask.taskapp import models as taskapp_models

@login_required

def create_task(request):

    user = request.user

    profile = user.get_profile()

    context = {"user": user,

               "profile": profile,

              }

    context.update(csrf(request))

    can_create_task = False if (

      profile.role == profile_models.ROLES_CHOICES[3][0]) else True

    if can_create_task:

        if request.method == "POST":

            form = taskapp_forms.CreateTaskForm(request.POST)

            if form.is_valid():

                data = form.cleaned_data.copy()

                data.update({"created_by": user,

                             "creation_datetime": datetime.now(),

                            })

                task = taskapp_models.Task(**data)

                task.save()

                task_url = reverse('view_task', kwargs={'task_id': task.id})

                return shortcuts.redirect(task_url)

            else:

                context.update({'form':form})

                return shortcuts.render_to_response(

                  'task/edit.html', RequestContext(request, context))

        else:

            form = taskapp_forms.CreateTaskForm()

            context.update({'form': form})

            return shortcuts.render_to_response(

              'task/edit.html', RequestContext(request, context))

    else:

        return show_msg(user, 'You are not authorised to create a task.')

def browse_tasks(request):

    open_tasks = taskapp_models.Task.objects.filter(status=taskapp_models.TASK_STATUS_CHOICES[1][0])

    working_tasks = taskapp_models.Task.objects.filter(status=taskapp_models.TASK_STATUS_CHOICES[3][0])

    comp_tasks = taskapp_models.Task.objects.filter(status=taskapp_models.TASK_STATUS_CHOICES[6][0])

    context = {"open_tasks": open_tasks,

               "working_tasks": working_tasks,

               "comp_tasks": comp_tasks,

              }

    user = request.user

    if not user.is_authenticated():

        return shortcuts.render_to_response("task/browse.html",

                                            RequestContext(request, context))

    profile = user.get_profile()

    can_approve = True if profile.role in [profile_models.ROLES_CHOICES[0][0], profile_models.ROLES_CHOICES[1][0]] else False

    unpub_tasks = taskapp_models.Task.objects.filter(status=taskapp_models.TASK_STATUS_CHOICES[0][0]).exclude(status=taskapp_models.TASK_STATUS_CHOICES[5][0])

    if can_approve:

        context.update({"unpub_tasks": unpub_tasks})

    context.update({"user": user,

                    "profile": profile,

                   })

    return shortcuts.render_to_response("task/browse.html",

                                        RequestContext(request, context))

def view_task(request, task_id):

    """ get the task depending on its task_id and display accordingly if it is a get.

    check for authentication and add a comment if it is a post request.

    """

    task_url = reverse('view_task', kwargs={'task_id': task_id})

    task = shortcuts.get_object_or_404(taskapp_models.Task, pk=task_id)

    user = request.user

    if not user.is_authenticated():

        return shortcuts.render_to_response("task/view.html", {"task": task})

    profile = user.get_profile()

    context = {"user": user,

               "profile": profile,

               "task": task,

              }

    context.update(csrf(request))

    if task.status == taskapp_models.TASK_STATUS_CHOICES[5][0]:

        return show_msg(user, 'This task no longer exists',

                        reverse('browse_tasks'), 'browse the tasks')

    task_viewable = True if ( task.status != taskapp_models.TASK_STATUS_CHOICES[0][0] ) or profile.role != profile_models.ROLES_CHOICES[3][0] \

                         else False

    if not task_viewable:

        return show_msg(user, "You are not authorised to view this task",

                        reverse('browse_tasks'), "browse the tasks")

    reviewers = task.reviewers.all()

    is_reviewer = True if user in task.reviewers.all() else False

    comments = task.comments.filter(

      is_deleted=False).order_by('comment_datetime')

    context.update({'is_reviewer':is_reviewer,

                    'comments':comments,

                    'reviewers':reviewers,

                   })

    selected_users = task.selected_users.all()

    is_creator = True if user == task.created_by else False

    context['selected_users'] = selected_users

    context['is_selected'] = True if user in selected_users else False

    context['can_approve'] = True if task.status == taskapp_models.TASK_STATUS_CHOICES[0][0] and\

                                     profile.role in [profile_models.ROLES_CHOICES[0][0], profile_models.ROLES_CHOICES[1][0]]\

                                     else False

    context['can_edit'] = True if is_creator and task.status == taskapp_models.TASK_STATUS_CHOICES[0][0] else False

    context['can_close'] = True if task.status not in [taskapp_models.TASK_STATUS_CHOICES[0][0], taskapp_models.TASK_STATUS_CHOICES[4][0], taskapp_models.TASK_STATUS_CHOICES[6][0]] and is_reviewer else False

    context['can_delete'] = True if task.status == taskapp_models.TASK_STATUS_CHOICES[0][0] and is_creator else False

    context['can_assign_pynts'] = True if task.status in [taskapp_models.TASK_STATUS_CHOICES[1][0], taskapp_models.TASK_STATUS_CHOICES[3][0]] and is_reviewer else False

    context['task_claimable'] = True if task.status in [taskapp_models.TASK_STATUS_CHOICES[1][0], taskapp_models.TASK_STATUS_CHOICES[3][0]] else False

    context['can_comment'] = True if task.status != taskapp_models.TASK_STATUS_CHOICES[0][0] or\

                                     profile.role != profile_models.ROLES_CHOICES[3][0] else False

    context['can_mod_reviewers'] = True if profile.role in [profile_models.ROLES_CHOICES[0][0], profile_models.ROLES_CHOICES[1][0]] else\

                                   False

    if request.method == 'POST':

        form = taskapp_forms.TaskCommentForm(request.POST)

        if form.is_valid():

            data = form.cleaned_data['data']

            new_comment = taskapp_forms.TaskComment(

              task=task, data=data, commented_by=user,

              comment_datetime=datetime.now())

            new_comment.save()

            return shortcuts.redirect(task_url)

        else:

            context['form'] = form

            return shortcuts.render_to_response(

              'task/view.html', RequestContext(request, context))

    else:

        form = taskapp_forms.TaskCommentForm()

        context['form'] = form

        return shortcuts.render_to_response(

          'task/view.html', RequestContext(request, context))

@login_required

def edit_task(request, task_id):

    """ only creator gets to edit the task and that too only before it gets

    approved.

    """

    user = request.user

    profile = user.get_profile()

    task_url = reverse('view_task', kwargs={'task_id': task_id})

    task = shortcuts.get_object_or_404(taskapp_models.Task, pk=task_id)

    is_creator = True if user == task.created_by else False

    can_edit = True if task.status == taskapp_models.TASK_STATUS_CHOICES[0][0] and is_creator else False

    if not can_edit:

        raise http.Http404

    context = {"user": user,

               "profile": profile,

               "task": task,

              }

    context.update(csrf(request))

    if request.method == "POST":

        form = taskapp_forms.EditTaskForm(request.POST, instance=task)

        if form.is_valid():

            form.save()

            return shortcuts.redirect(task_url)

        else:

            context.update({"form": form})

            return shortcuts.render_to_response(

              "task/edit.html", RequestContext(request, context))

    else:

        form = taskapp_forms.EditTaskForm(instance=task)

        context.update({"form": form})

        return shortcuts.render_to_response("task/edit.html",

                                            RequestContext(request, context))

@login_required

def approve_task(request, task_id):

    user = request.user

    profile = user.get_profile()

    task = shortcuts.get_object_or_404(taskapp_models.Task, pk=task_id)

    if profile.role not in [profile_models.ROLES_CHOICES[0][0], profile_models.ROLES_CHOICES[1][0]] or task.status != taskapp_models.TASK_STATUS_CHOICES[0][0]:

        raise http.Http404

    context = {"user": user,

               "profile": profile,

               "task": task,

              }

    return shortcuts.render_to_response(

      "task/confirm_approval.html", RequestContext(request, context))

@login_required

def approved_task(request, task_id):

    user = request.user

    profile = user.get_profile()

    task = shortcuts.get_object_or_404(taskapp_models.Task, pk=task_id)

    if profile.role not in [profile_models.ROLES_CHOICES[0][0], profile_models.ROLES_CHOICES[1][0]] or task.status != taskapp_models.TASK_STATUS_CHOICES[0][0]:

        raise http.Http404

    task.approved_by = user

    task.approval_datetime = datetime.now()

    task.status = taskapp_models.TASK_STATUS_CHOICES[1][0]

    task.save()

    context = {"user": user,

               "profile": profile,

               "task": task,

              }

    return shortcuts.render_to_response(

      "task/approved_task.html", RequestContext(request, context))

@login_required

def addreviewer(request, task_id):

    user = request.user

    profile = user.get_profile()

    task_url = reverse('view_task', kwargs={'task_id': task_id})

    task = shortcuts.get_object_or_404(taskapp_models.Task, pk=task_id)

    can_mod_reviewers = True if profile.role in [profile_models.ROLES_CHOICES[0][0], profile_models.ROLES_CHOICES[1][0]] else False

    if not can_mod_reviewers:

        raise http.Http404

    context = {"user": user,

               "profile": profile,

               "task": task,

              }

    context.update(csrf(request))

    # This part has to be made better

    reviewer_choices = User.objects.filter(is_active=True).\

                                           exclude(reviewing_tasks__id=task_id).\

                                           exclude(claimed_tasks__id=task_id).\

                                           exclude(selected_tasks__id=task_id).\

                                           exclude(created_tasks__id=task_id)

    choices = ((a_user.id,a_user.username) for a_user in reviewer_choices)

    label = "Reviewer"

    if request.method == "POST":

        form = taskapp_forms.ChoiceForm(choices, data=request.POST, label=label)

        if form.is_valid():

            data = form.cleaned_data.copy()

            uid = data['choice']

            reviewer = User.objects.get(id=uid)

            task.reviewers.add(reviewer)

            return shortcuts.redirect(task_url)

        else:

            context.update({"form": form})

            return shortcuts.render_to_response(

              "task/addreviewer.html", RequestContext(request, context))

    else:

        form = taskapp_forms.ChoiceForm(choices, label=label)

        context.update({"form": form})

        return shortcuts.render_to_response(

          "task/addreviewer.html", RequestContext(request, context))

def view_work(request, task_id):

    task = shortcuts.get_object_or_404(taskapp_models.Task, pk=task_id)

    user = request.user

    old_reports = task.reports.all()

    context = {"task": task,

               "old_reports": old_reports,

              }

    if not user.is_authenticated():

        return shortcuts.render_to_response(

          "task/view_work.html", RequestContext(request, context))

    profile = user.get_profile()

    context.update({"user": user,

                    "profile": profile,

                   })

    context.update(csrf(request))

    working_users = task.selected_users.all()

    is_working = True if user in working_users else False

    context.update({"is_working": is_working})

    return shortcuts.render_to_response("task/view_work.html",

                                        RequestContext(request, context))

@login_required

def view_report(request, report_id):

    report = shortcuts.get_object_or_404(taskapp_models.WorkReport,

                                         pk=report_id)

    user = request.user

    context = {"report": report,

               "user": user,

              }

    if not user.is_authenticated():

        return shortcuts.render_to_response(

          "task/view_report.html", RequestContext(request, context))

    profile = user.get_profile()

    context.update({"profile": profile})

    return shortcuts.render_to_response("task/view_report.html",

                                        RequestContext(request, context))

@login_required

def submit_report(request, task_id):

    """ Check if the work is in WR state and the user is in assigned_users.

    """

    task_url = reverse('view_task', kwargs={'task_id': task_id})

    task = shortcuts.get_object_or_404(taskapp_models.Task, pk=task_id)

    user = request.user

    old_reports = task.reports.all()

    if not task.status == taskapp_models.TASK_STATUS_CHOICES[3][0]:

        raise http.Http404

    can_upload = True if user in task.selected_users.all() else False

    context = {

        'user': user,

        'task': task,

        'can_upload': can_upload,

    }

    context.update(csrf(request))

    if request.method == "POST":

        if not can_upload:

            return show_msg(user, "You are not authorised to upload data to this task", task_url, "view the task")

        form = taskapp_forms.WorkReportForm(request.POST, request.FILES)

        if form.is_valid():

            data = form.cleaned_data.copy()

            data.update({"task":task,

                         "revision": old_reports.count(),

                         "submitted_by": user,

                         "submitted_at": datetime.now(),

                        })

            r = taskapp_models.WorkReport(**data)

            r.save()

            report_url = reverse('view_report', kwargs={'report_id': r.id})

            return shortcuts.redirect(report_url)

        else:

            context.update({"form":form})

            return shortcuts.render_to_response(

              'task/submit_report.html', RequestContext(request, context))

    else:

        form = taskapp_forms.WorkReportForm()

        context.update({"form":form})

        return shortcuts.render_to_response(

          'task/submit_report.html', RequestContext(request, context))

@login_required

def create_textbook(request):

    user = request.user

    profile = user.get_profile()

    can_create = True if profile.role != profile_models.ROLES_CHOICES[3][0] else False

    if not can_create:

        raise http.Http404

    context = {"user": user,

               "profile": profile,

              }

    context.update(csrf(request))

    if request.method == "POST":

        form = taskapp_forms.CreateTextbookForm(request.POST)

        if form.is_valid():

            data = form.cleaned_data.copy()

            data.update({"created_by": user,

                         "creation_datetime": datetime.now()})

            del data['chapters']

            new_textbook = taskapp_models.TextBook(**data)

            new_textbook.save()

            new_textbook.chapters = form.cleaned_data['chapters']

            textbook_url = reverse(

              'view_textbook', kwargs={'task_id': new_textbook.id})

            return shortcuts.redirect(textbook_url)

        else:

            context.update({"form": form})

            return shortcuts.render_to_response(

              "task/edit.html", RequestContext(request, context))

    else:

        form = taskapp_forms.CreateTextbookForm()

        context.update({"form": form})

        return shortcuts.render_to_response(

          "task/edit.html", RequestContext(request, context))

def view_textbook(request, task_id):

    textbook = shortcuts.get_object_or_404(taskapp_models.TextBook, pk=task_id)

    chapters = textbook.chapters.all()

    user = request.user

    context = {"user": user,

               "textbook": textbook,

               "chapters": chapters,

              }

    if not user.is_authenticated():

        return shortcuts.render_to_response("task/view_textbook.html",

                                            RequestContext(request, context))

    profile = user.get_profile()

    context.update({"profile": profile,

                    "textbook": textbook,

                   })

    context.update(csrf(request))

    can_edit = True if user == textbook.created_by and textbook.status == taskapp_models.TB_STATUS_CHOICES[0][0]\

                       else False