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))
def upload_data_budgetitems(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: if r[0] == "IN": item = IncomeBudgetItem() item.description = r[1] item.amount_per_item = r[2] item.num_items = r[3] item.amount = r[4] item.budget = Budget.objects.get(position__name=r[5],term=r[6],year=r[7]) item.income_category = IncomeCategory.objects.get(name=r[8]) item.save() elif r[0] == "EX": item = ExpenseBudgetItem() item.description = r[1] item.amount_per_item = r[2] item.num_items = r[3] item.amount = r[4] item.budget = Budget.objects.get(position__name=r[5],term=r[6],year=r[7]) item.expenditure_category = ExpenditureCategory.objects.get(name=r[8]) item.save() return HttpResponseRedirect(reverse('budget_view_budgets')) else: form = UploadDataForm() template['form'] = form return render_to_response('budget/upload_budgetitems.htm',template, context_instance=RequestContext(request))