test-auth/hr/views.py

import datetime

from django.http import HttpResponseRedirect
from django.shortcuts import render_to_response
from django.core.urlresolvers import reverse
from django.contrib.auth.models import User, Group
from django.contrib.auth.decorators import login_required
from django.template import RequestContext
from django.template.loader import render_to_string

import settings

from eve_api.models import EVEAccount, EVEPlayerCorporation
from reddit.models import RedditAccount

from hr.forms import CreateRecommendationForm, CreateApplicationForm, CreateApplicationStatusForm, NoteForm
from hr.models import Recommendation, Application, Audit

from app_defines import *

### Shared Functions

def send_message(application, message_type, note=None):
    from django.core.mail import send_mail
    subject = render_to_string('hr/emails/%s_subject.txt' % message_type, { 'app': application })
    subject = ''.join(subject.splitlines())
    message = render_to_string('hr/emails/%s.txt' % message_type, { 'app': application, 'note': note })
    try:
        send_mail(subject, message, settings.DEFAULT_FROM_EMAIL, [application.user.email])
    except:
        pass

    if len(application.user.redditaccount_set.all()) > 0:
        from reddit.api import Inbox
        ib = Inbox(settings.REDDIT_USER, settings.REDDIT_PASSWD)
        ib.send(application.user.redditaccount_set.all()[0].username, subject, message)

### General Views

def index(request):
    if request.user.is_staff or Group.objects.get(name=settings.HR_STAFF_GROUP) in request.user.groups.all():
        hrstaff = True

    return render_to_response('hr/index.html', locals(), context_instance=RequestContext(request))

### Application Management

@login_required
def view_applications(request):
    apps = Application.objects.filter(user=request.user)
    return render_to_response('hr/applications/view_list.html', locals(), context_instance=RequestContext(request))

@login_required
def view_application(request, applicationid):
    try:
        app = Application.objects.get(id=applicationid)
    except Application.DoesNotExist:
        return HttpResponseRedirect(reverse('hr.views.index'))

    if not app.user == request.user and not (request.user.is_staff or Group.objects.get(name=settings.HR_STAFF_GROUP) in request.user.groups.all()):
        return HttpResponseRedirect(reverse('hr.views.index'))

    if request.user.is_staff or Group.objects.get(name=settings.HR_STAFF_GROUP) in request.user.groups.all():
        hrstaff = True
        audit = app.audit_set.all()
    else:
        hrstaff = False
        audit = app.audit_set.filter(event__in=[AUDIT_EVENT_STATUSCHANGE, AUDIT_EVENT_REJECTION_REASON])

    if hrstaff or app.status < 1:
        appform = CreateApplicationStatusForm(hrstaff)
        form = appform(initial={'application': app.id, 'new_status': app.status})

    eveacc = app.user.eveaccount_set.all()
    redditacc = app.user.redditaccount_set.all()
    recs = app.recommendation_set.all()

    posts = []
    for acc in redditacc:
        posts.extend(acc.recent_posts())

    return render_to_response('hr/applications/view.html', locals(), context_instance=RequestContext(request))

@login_required
def add_application(request):

    clsform = CreateApplicationForm(request.user)
    if request.method == 'POST': 
        form = clsform(request.POST) 
        if form.is_valid():

            if form.cleaned_data['character'].corporation == form.cleaned_data['corporation']:
                request.user.message_set.create(message="This character is already a member of %s" % form.cleaned_data['corporation'])
                return HttpResponseRedirect(reverse('hr.views.view_applications'))

            app = Application()

            app.user = request.user
            app.character = form.cleaned_data['character']
            app.corporation = form.cleaned_data['corporation']
            app.save()

            request.user.message_set.create(message="Your application to %s has been created." % app.corporation)
            return HttpResponseRedirect(reverse('hr.views.view_applications'))
            
    else:
        form = clsform() # An unbound form

    if len(EVEPlayerCorporation.objects.filter(applications=True)):
        return render_to_response('hr/applications/add.html', locals(), context_instance=RequestContext(request))
    else:
        return render_to_response('hr/applications/noadd.html', locals(), context_instance=RequestContext(request)) 

### Recommendation Management

@login_required
def view_recommendations(request):
    recs = Recommendation.objects.filter(user=request.user, application__status=0)
    return render_to_response('hr/recommendations/view_list.html', locals(), context_instance=RequestContext(request))

@login_required
def view_recommendation(request, recommendationid):
    try:
        rec = Recommendation.objects.get(id=recommendationid, user=request.user)
    except Recommendation.DoesNotExist:
        return HttpResponseRedirect(reverse('hr.views.index'))
    return render_to_response('hr/recommendations/view.html', locals(), context_instance=RequestContext(request))

@login_required
def add_recommendation(request):

    clsform = CreateRecommendationForm(request.user)

    if request.method == 'POST': 
        form = clsform(request.POST) 
        if form.is_valid():
            rec = Recommendation()

            rec.user = request.user
            rec.user_character = form.cleaned_data['character']
            rec.application = form.cleaned_data['application']
            rec.created_by = request.user
            rec.last_updated_by = request.user
            rec.save()

            request.user.message_set.create(message="Recommendation added to %s's application" % rec.application )
            return HttpResponseRedirect(reverse('hr.views.view_recommendations'))
            
    else:
        form = clsform() # An unbound form

    return render_to_response('hr/recommendations/add.html', locals(), context_instance=RequestContext(request))

@login_required
def admin_applications(request):
    if not (request.user.is_staff or Group.objects.get(name=settings.HR_STAFF_GROUP) in request.user.groups.all()):
        return HttpResponseRedirect(reverse('hr.views.index'))

    view_status = [APPLICATION_STATUS_AWAITINGREVIEW, APPLICATION_STATUS_ACCEPTED, APPLICATION_STATUS_QUERY]
    apps = Application.objects.filter(status__in=view_status)
    return render_to_response('hr/applications/admin/view_list.html', locals(), context_instance=RequestContext(request))

@login_required
def update_application(request, applicationid):   
    if request.method == 'POST':
        appform = CreateApplicationStatusForm(True)
        form = appform(request.POST)
        if form.is_valid():
            app = Application.objects.get(id=form.cleaned_data['application'])

            hrstaff = (request.user.is_staff or Group.objects.get(name=settings.HR_STAFF_GROUP) in request.user.groups.all())
            if not hrstaff and int(form.cleaned_data['new_status']) > 1:
                return HttpResponseRedirect(reverse('hr.views.index'))

            if not app.status == form.cleaned_data['new_status']:

                app.status = form.cleaned_data['new_status']
                app.save(user=request.user)

                if int(app.status) == APPLICATION_STATUS_ACCEPTED:
                    send_message(app, 'accepted')

    return HttpResponseRedirect(reverse('hr.views.view_application', args=[applicationid]))

@login_required
def add_note(request, applicationid):   
    if request.method == 'POST':
        obj = Audit(application=Application.objects.get(id=applicationid), user=request.user, event=AUDIT_EVENT_NOTE)
        form = NoteForm(request.POST, instance=obj)
        if form.is_valid():
            form.save()
            return HttpResponseRedirect(reverse('hr.views.view_application', args=[applicationid]))

    form = NoteForm()
    return render_to_response('hr/applications/add_note.html', locals(), context_instance=RequestContext(request))

@login_required
def reject_application(request, applicationid):   
    if request.method == 'POST':
        obj = Audit(application=Application.objects.get(id=applicationid), user=request.user, event=AUDIT_EVENT_REJECTION)
        form = NoteForm(request.POST, instance=obj)
        if form.is_valid():
            obj = form.save()
            obj.application.status = APPLICATION_STATUS_REJECTED
            obj.application.save(user=request.user)
            send_message(obj.application, 'rejected', note=obj.text)
            return HttpResponseRedirect(reverse('hr.views.view_application', args=[applicationid]))

    form = NoteForm()
    return render_to_response('hr/applications/reject.html', locals(), context_instance=RequestContext(request))

@login_required
def accept_application(request, applicationid):   
    if request.method == 'POST':
        obj = Audit(application=Application.objects.get(id=applicationid), user=request.user, event=AUDIT_EVENT_ACCEPTED)
        form = NoteForm(request.POST, instance=obj)
        if form.is_valid():
            obj = form.save()
            obj.application.status = APPLICATION_STATUS_ACCEPTED
            obj.application.save(user=request.user)
            send_message(obj.application, 'accepted', note=obj.text)
            return HttpResponseRedirect(reverse('hr.views.view_application', args=[applicationid]))

    form = NoteForm()
    return render_to_response('hr/applications/accept.html', locals(), context_instance=RequestContext(request))