def update_flash(self): """ Since so many methods in this class did the exact same thing, a function was created. This updated the session invoice counter, used in many places (eg. sidebar). """ shared_unpaid_invoices = 0 shared_categories = Category.all_active(self.request).all() for c in shared_categories: unpaid_invoices = Invoice.with_category_all_unpaid(c.id) if unpaid_invoices: shared_unpaid_invoices += len(unpaid_invoices) self.request.session.pop_flash('shared_unpaid_invoices') self.request.session.flash(shared_unpaid_invoices, 'shared_unpaid_invoices') private_unpaid_invoices = 0 private_categories = Category.all_private(self.request).all() for c in private_categories: unpaid_invoices = Invoice.with_category_all_unpaid(c.id) if unpaid_invoices: private_unpaid_invoices += len(unpaid_invoices) self.request.session.pop_flash('private_unpaid_invoices') self.request.session.flash(private_unpaid_invoices, 'private_unpaid_invoices')
def invoices(self): """ Get a paginated list of active invoices for current or given month.""" paid_result = {} unpaid_result = {} year = int(self.request.params.get('year', datetime.now().strftime('%Y'))) month = int(self.request.params.get('month', datetime.now().strftime('%m'))) categories = Category.all_active(self.request).all() for c in categories: paid_invoices = Invoice.with_category_paid(c.id, year, month) if paid_invoices: total = Invoice.with_category_paid(c.id, year, month, total_only=True) paid_result[c.title] = [paid_invoices, total] unpaid_invoices = Invoice.with_category_all_unpaid(c.id) if unpaid_invoices: total = Invoice.with_category_all_unpaid(c.id, total_only=True) unpaid_result[c.title] = [unpaid_invoices, total] return {'paiditems': paid_result, 'unpaiditems': unpaid_result, 'title': 'Invoices', 'month': month, 'year': year, 'nextmonth': self.month_switcher(year, month, next=True), 'prevmonth': self.month_switcher(year, month)}
def expenditure_edit(self): """ Edit expenditure. Method for both post and get request. """ id = int(self.request.matchdict.get('id')) e = Expenditure.by_id(id) if not e: return HTTPNotFound() """ Authorization check. """ if (e.category.private and e.category.user_id is not authenticated_userid(self.request)): return HTTPForbidden() form = ExpenditureEditForm(self.request.POST, e, csrf_context=self.request.session) private = self.request.params.get('private') if private: """ Check if there exists any categories. """ if not Category.first_private(self.request): self.request.session.flash(self.missing_priv_cat, 'error') return HTTPFound(location=self.request .route_url('expenditures')) form.category_id.query = Category.all_private(self.request) else: """ Check if there exists any categories. """ if not Category.first_active(): self.request.session.flash(self.missing_shared_cat, 'error') return HTTPFound(location=self.request .route_url('expenditures')) form.category_id.query = Category.all_active(self.request) if self.request.method == 'POST' and form.validate(): form.populate_obj(e) e.category_id = form.category_id.data.id self.request.session.flash('Expenditure %s updated' % (e.title), 'status') """ A bit ugly, but redirect the user based on private or not. """ if private: return HTTPFound(location= self.request .route_url('expenditures', _query={'private': 1})) return HTTPFound(location=self.request.route_url('expenditures')) form.category_id.data = e.category return {'title': 'Edit private expenditure' if private else 'Edit expenditure', 'form': form, 'id': id, 'action': 'expenditure_edit', 'private': private}
def expenditure_create(self): """ New expenditure. Method for both post and get request. """ form = ExpenditureCreateForm(self.request.POST, csrf_context=self.request.session) private = self.request.params.get('private') if private: """ Check if there exists any private categories. """ if not Category.first_private(self.request): self.request.session.flash(self.missing_priv_cat, 'error') return HTTPFound(location=self.request .route_url('expenditures')) form.category_id.query = Category.all_private(self.request) else: """ Check if there exists any categories. """ if not Category.first_active(): self.request.session.flash(self.missing_shared_cat, 'error') return HTTPFound(location=self.request .route_url('expenditures')) form.category_id.query = Category.all_active(self.request) if self.request.method == 'POST' and form.validate(): e = Expenditure() form.populate_obj(e) e.user_id = authenticated_userid(self.request) e.category_id = form.category_id.data.id DBSession.add(e) self.request.session.flash('Expenditure %s created' % (e.title), 'success') """ A bit ugly, but redirect the user based on private or not. """ if private: return HTTPFound(location= self.request .route_url('expenditures', _query={'private': 1})) return HTTPFound(location=self.request.route_url('expenditures')) return {'title': 'New private expenditure' if private else 'New expenditure', 'form': form, 'action': 'expenditure_new', 'private': private}
def expenditures(self): """ Return a paginated list of active expenditures. Handles both private and shared expenditures. """ result = {} private = self.request.params.get('private') if private: categories = Category.all_private(self.request).all() else: categories = Category.all_active(self.request).all() for c in categories: e = Expenditure.with_category(c.id) if e: s = Expenditure.with_category(c.id, total_only=True) result[c.title] = [e, s] return {'items': result, 'title': 'Private expenditures' if private else 'Shared expenditures', 'private': private}
def invoices_search(self): """ Get a list of invoices based on search arguments. """ page = int(self.request.params.get('page', 1)) form = InvoiceSearchForm(self.request.GET, csrf_context=self.request.session) form.categories.query = Category.all_active(self.request) form.creditors.query = Creditor.all_active(self.request) if self.request.method == 'GET' and form.validate(): q = form.query.data categories = form.categories.data creditors = form.creditors.data fromdate = form.fromdate.data todate = form.todate.data invoices = Invoice.searchpage(self.request, page, qry=q, categories=categories, creditors=creditors, fromdate=fromdate, todate=todate, ) total = Invoice.searchpage(self.request, page, qry=q, categories=categories, creditors=creditors, fromdate=fromdate, todate=todate, total_only=True, ) else: invoices = Invoice.searchpage(self.request, page) total = Invoice.searchpage(self.request, page, total_only=True) return {'paginator': invoices, 'form': form, 'title': 'Search', 'searchpage': True, 'total': total}