Beispiel #1
0
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))    
Beispiel #2
0
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))