test-auth/hr/views.py

import datetime

from django.http import HttpResponseRedirect
from django.shortcuts import render_to_response, get_object_or_404
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, 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):
    app = get_object_or_404(Application, id=applicationid)

    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, AUDIT_EVENT_ACCEPTED])

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

    posts = []
    for acc in redditacc:
        try: 
            accposts = acc.recent_posts()
        except:
            reddit_error = "Encountered a error while trying to get all Reddit posts"
            accposts = []
        posts.extend(accposts)

    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_application', args=[app.id]))
            
    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):
    rec = get_object_or_404(Recommendation, id=recommendationid, user=request.user)
    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'))

    if 'q' in request.GET:
        query = request.GET['q']
        if 'l' in request.GET:
            limit = request.get['l']
        else:
            limit = 10
        print query, limit
        apps = Application.objects.filter(character__name__icontains=query)[:limit]
    else:
        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, status): 

    hrstaff = (request.user.is_staff or Group.objects.get(name=settings.HR_STAFF_GROUP) in request.user.groups.all())
    app = get_object_or_404(Application, id=applicationid)

    # Allow admins and users that are setting the application as awaiting review
    if hrstaff or (app.user == request.user and int(status) == APPLICATION_STATUS_AWAITINGREVIEW):  
        if not app.status == status:
            app.status = status
            app.save(user=request.user)

    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))

def add_message(request, applicationid):
    if request.method == 'POST':
        obj = Audit(application=Application.objects.get(id=applicationid), user=request.user, event=AUDIT_EVENT_MESSAGE)
        form = NoteForm(request.POST, instance=obj)
        if form.is_valid():
            form.save()
            send_message(obj.application, 'message', note=obj.text)
            return HttpResponseRedirect(reverse('hr.views.view_application', args=[applicationid]))

    form = NoteForm()
    return render_to_response('hr/applications/add_message.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))