def upload_data_budget(request):
template = dict()
if request.method == 'POST':
form = UploadDataForm(request.POST, request.FILES)
if form.is_valid():
directory = MEDIA_ROOT + "/test_data/" + request.FILES["file"].name
reader = csv.reader(open(directory))
for r in reader:
budget = Budget()
budget.position = Position.objects.get(name=r[0])
budget.term = r[1]
budget.year = r[2]
budget.stream = r[3]
if r[4] == "TRUE":
budget.approved = True
budget.creator = request.user
budget.start_date = start_date(int(budget.year), budget.term)
budget.end_date = end_date(int(budget.year), budget.term)
budget.save()
return HttpResponseRedirect(reverse('budget_view_budgets'))
else:
form = UploadDataForm()
template['form'] = form
return render_to_response('budget/upload_budget.htm',template, context_instance=RequestContext(request))
