source: treasury/vouchers/views.py @ e0689aa

client
Last change on this file since e0689aa was e0689aa, checked in by Alex Dehnert <adehnert@…>, 16 years ago

Lazily initialize CommitteesField?

Otherwise, any page view will throw an exception
if the Accounts -> Assets -> Budget account
doesn't exist, which is a bit annoying

  • Property mode set to 100644
File size: 4.4 KB
Line 
1import vouchers.models
2from finance_core.models import BudgetTerm, BudgetArea
3from vouchers.models import ReimbursementRequest
4
5from django.contrib.auth.decorators import user_passes_test
6from django.shortcuts import render_to_response, get_object_or_404
7from django.http import Http404, HttpResponseRedirect
8from django.forms import Form
9from django.forms import ModelForm
10from django.forms import ModelChoiceField
11from django.core.urlresolvers import reverse
12
13class RequestForm(ModelForm):
14    class Meta:
15        model = ReimbursementRequest
16        fields = (
17            'name',
18            'description',
19            'amount',
20            'budget_area',
21            'check_to_name',
22            'check_to_email',
23            'check_to_addr',
24        )
25
26class CommitteesField(ModelChoiceField):
27    def __init__(self, *args, **kargs):
28        base_area = BudgetArea.get_by_path(['Accounts', 'Assets', 'Budget', ])
29        self.strip_levels = base_area.depth
30        areas = (base_area.get_descendants()
31            .filter(depth__lte=base_area.depth+2)
32            .exclude(name='Holding')
33        )
34        ModelChoiceField.__init__(self, queryset=areas,
35            help_text='Select the appropriate committe or other budget area',
36            *args, **kargs)
37
38    def label_from_instance(self, obj,):
39        return obj.indented_name(strip_levels=self.strip_levels)
40
41class SelectRequestBasicsForm(Form):
42    area = None
43    term = ModelChoiceField(queryset = BudgetTerm.objects.all())
44    def __init__(self, *args, **kargs):
45        self.area = CommitteesField()
46
47@user_passes_test(lambda u: u.is_authenticated())
48def select_request_basics(http_request, ):
49    if http_request.method == 'POST': # If the form has been submitted...
50        form = SelectRequestBasicsForm(http_request.POST) # A form bound to the POST data
51        if form.is_valid(): # All validation rules pass
52            term = form.cleaned_data['term'].slug
53            area = form.cleaned_data['area'].id
54            return HttpResponseRedirect(reverse(submit_request, args=[term, area],)) # Redirect after POST
55    else:
56        form = SelectRequestBasicsForm() # An unbound form
57
58    context = {
59        'form':form,
60    }
61    return render_to_response('vouchers/select.html', context)
62
63class CommitteeBudgetAreasField(ModelChoiceField):
64    def __init__(self, base_area, *args, **kargs):
65        self.strip_levels = base_area.depth
66        areas = base_area.get_descendants()
67        ModelChoiceField.__init__(self, queryset=areas,
68            help_text='In general, this should be a fully indented budget area, not one with children',
69            *args, **kargs)
70
71    def label_from_instance(self, obj,):
72        return obj.indented_name(strip_levels=self.strip_levels)
73
74@user_passes_test(lambda u: u.is_authenticated())
75def submit_request(http_request, term, committee):
76    term_obj = get_object_or_404(BudgetTerm, slug=term)
77    comm_obj = get_object_or_404(BudgetArea, pk=committee)
78
79    new_request = ReimbursementRequest()
80    new_request.submitter = http_request.user.username
81    new_request.budget_term = term_obj
82
83    if http_request.method == 'POST': # If the form has been submitted...
84        form = RequestForm(http_request.POST, instance=new_request) # A form bound to the POST data
85        form.fields['budget_area'] = CommitteeBudgetAreasField(comm_obj)
86        if form.is_valid(): # All validation rules pass
87            form.save()
88            return HttpResponseRedirect(reverse(review_request, args=[new_request.pk],)) # Redirect after POST
89    else:
90        form = RequestForm(instance=new_request) # An unbound form
91        form.fields['budget_area'] = CommitteeBudgetAreasField(comm_obj)
92
93    context = {
94        'term':term_obj,
95        'comm':comm_obj,
96        'form':form,
97    }
98    return render_to_response('vouchers/submit.html', context)
99
100@user_passes_test(lambda u: u.is_authenticated())
101def review_request(http_request, object_id):
102    request_obj = get_object_or_404(ReimbursementRequest, pk=object_id)
103    if not (http_request.user.has_perm('vouchers.view_requests')
104        or http_request.user.username == request_obj.submitter):
105        # I'd probably use a 403, but that requires like writing
106        # a new template and stuff
107        # So I'm going to call this "don't leak information"
108        # and let it be
109        raise Http404
110    context = {
111        'rr':request_obj,
112    }
113    return render_to_response('vouchers/ReimbursementRequest_review.html', context)
114
Note: See TracBrowser for help on using the repository browser.