from app import app from app import db from api.model.accounts import Account from api.model.entries import Entry from flask import json, request from sqlalchemy import func, case, cast, extract, distinct @app.route("/api/accounts", methods=["GET"]) def get_accounts(): """ Returns accounts with their solds. """ session = db.session query = session.query( Account.id.label("id"), Account.name.label("name"), Account.authorized_overdraft.label("authorized_overdraft"), func.sum(Entry.value).label("future"), func.sum(case([(Entry.pointed, Entry.value,)], else_=cast(0, db.Numeric(15, 2)))).label("pointed"), func.sum(case([(Entry.operation_date < func.now(), Entry.value,)], else_=cast(0, db.Numeric(15, 2)))).label("current") ).outerjoin(Entry).group_by(Account.id).order_by(Account.id) return json.dumps([{ "id": i.id, "name": i.name, "authorized_overdraft": i.authorized_overdraft, "current": str(i.current), "pointed": str(i.pointed), "future": str(i.future) } for i in query.all()]) @app.route("/api/accounts//months") def get_months(account_id): session = db.session query = session.query( distinct(extract("year", Entry.operation_date)).label("year"), extract("month", Entry.operation_date).label("month") ).filter(Entry.account_id == account_id).order_by("year", "month") return json.dumps([{ "year": i.year, "month": i.month } for i in query.all()]) @app.route("/api/accounts", methods=["PUT"]) def add_account(): session = db.session try: account = Account(request.json['name'], request.json['authorized_overdraft']) session.add(account) session.commit() return json.dumps("Account added.") except: session.rollback() raise @app.route("/api/accounts/", methods=["PUT"]) def update_account(account_id): session = db.session try: account = session.query(Account).filter(Account.id == account_id).first() account.name = request.json['name'] account.authorized_overdraft = request.json['authorized_overdraft'] session.merge(account) session.commit() return json.dumps("Account #%s updated." % account_id) except: session.rollback() raise @app.route("/api/accounts/", methods=["DELETE"]) def delete_account(account_id): session = db.session try: account = session.query(Account).filter(Account.id == account_id).first() session.delete(account) session.commit() return json.dumps("Account #%s deleted." % account_id) except: session.rollback() raise