DrCr/drcr/journal/views.py

178 lines
5.9 KiB
Python
Raw Normal View History

2022-12-24 12:31:45 +11:00
# DrCr: Web-based double-entry bookkeeping framework
2023-01-02 18:08:30 +11:00
# Copyright (C) 2022–2023 Lee Yingtong Li (RunasSudo)
2022-12-24 12:31:45 +11:00
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
2023-01-07 19:01:05 +11:00
from flask import abort, redirect, render_template, request, url_for
2022-12-24 12:31:45 +11:00
2022-12-24 17:39:18 +11:00
from .. import AMOUNT_DPS
from ..database import db
2023-01-02 18:08:30 +11:00
from ..models import Amount, Posting, Transaction, TrialBalancer
2022-12-24 17:39:18 +11:00
from ..webapp import all_transactions, app
2023-01-02 18:08:30 +11:00
from .models import BalanceAssertion
from ..statements.models import StatementLineReconciliation
2022-12-24 17:39:18 +11:00
from datetime import datetime
2022-12-24 12:31:45 +11:00
2023-01-02 18:08:30 +11:00
@app.route('/journal')
def journal():
transactions = db.session.scalars(db.select(Transaction).options(db.selectinload(Transaction.postings)).order_by(Transaction.dt)).all()
2022-12-25 17:31:31 +11:00
return render_template(
2023-01-02 18:08:30 +11:00
'journal/journal.html',
2023-01-07 19:01:05 +11:00
commodity_detail=request.args.get('commodity_detail', '0') == '1',
transactions=transactions
2022-12-25 17:31:31 +11:00
)
2023-01-02 18:08:30 +11:00
@app.route('/journal/new-transaction', methods=['GET', 'POST'])
def journal_new_transaction():
if request.method == 'GET':
2023-01-02 18:08:30 +11:00
return render_template('journal/journal_edit_transaction.html', transaction=None)
# New transaction
2023-01-02 18:08:30 +11:00
transaction = Transaction(
dt=datetime.strptime(request.form['dt'], '%Y-%m-%d'),
description=request.form['description'],
postings=[]
)
2022-12-24 19:45:34 +11:00
for account, sign, amount_str in zip(request.form.getlist('account'), request.form.getlist('sign'), request.form.getlist('amount')):
amount = Amount.parse(amount_str)
if sign == 'cr':
2022-12-24 19:45:34 +11:00
amount = -amount
2023-01-02 18:08:30 +11:00
posting = Posting(
account=account,
2022-12-24 19:45:34 +11:00
quantity=amount.quantity,
commodity=amount.commodity
)
transaction.postings.append(posting)
transaction.assert_valid()
db.session.add(transaction)
db.session.commit()
2023-01-07 19:01:05 +11:00
return redirect(url_for('journal'))
2022-12-24 17:39:18 +11:00
2023-01-02 18:08:30 +11:00
@app.route('/journal/edit-transaction', methods=['GET', 'POST'])
def journal_edit_transaction():
transaction = db.session.get(Transaction, request.args['id'])
if not transaction:
abort(404)
if request.method == 'GET':
2023-01-02 18:08:30 +11:00
return render_template('journal/journal_edit_transaction.html', transaction=transaction)
# Edit transaction
transaction.dt = datetime.strptime(request.form['dt'], '%Y-%m-%d')
transaction.description = request.form['description']
2023-01-02 18:08:30 +11:00
new_postings = []
for account, sign, amount_str in zip(request.form.getlist('account'), request.form.getlist('sign'), request.form.getlist('amount')):
amount = Amount.parse(amount_str)
if sign == 'cr':
amount = -amount
2023-01-02 18:08:30 +11:00
posting = Posting(
account=account,
quantity=amount.quantity,
commodity=amount.commodity
)
new_postings.append(posting)
# Fix up reconciliations
for old_posting in transaction.postings:
for reconciliation in StatementLineReconciliation.query.filter(StatementLineReconciliation.posting == old_posting):
# See if there is a corresponding new posting
new_posting = next((p for p in new_postings if p.account == old_posting.account and p.quantity == old_posting.quantity and p.commodity == old_posting.commodity), None)
if new_posting is not None:
# Match up reconciliation
reconciliation.posting = new_posting
else:
# No matching reconciliation
db.session.delete(reconciliation)
transaction.postings = new_postings # This queues the old postings for deletion
transaction.assert_valid()
db.session.commit()
2023-01-07 19:01:05 +11:00
return redirect(url_for('journal'))
2022-12-24 17:39:18 +11:00
@app.route('/balance-assertions')
def balance_assertions():
2023-01-04 18:03:15 +11:00
assertions = db.session.scalars(db.select(BalanceAssertion).order_by(BalanceAssertion.dt)).all()
2022-12-24 17:39:18 +11:00
# Check assertion status
transactions = all_transactions()
assertion_status = {}
for assertion in assertions:
# FIXME: This is very inefficient
balancer = TrialBalancer()
balancer.apply_transactions([t for t in transactions if t.dt <= assertion.dt])
# TODO: Commodities
if balancer.accounts[assertion.account].quantity == assertion.quantity:
assertion_status[assertion] = True
else:
assertion_status[assertion] = False
2023-01-02 18:08:30 +11:00
return render_template('journal/balance_assertions.html', assertions=assertions, assertion_status=assertion_status)
2022-12-24 17:39:18 +11:00
@app.route('/balance-assertions/new', methods=['GET', 'POST'])
def balance_assertions_new():
if request.method == 'GET':
2023-01-02 18:08:30 +11:00
return render_template('journal/balance_assertions_edit.html', assertion=None)
2022-12-24 17:39:18 +11:00
quantity = round(float(request.form['amount']) * (10**AMOUNT_DPS))
if request.form['sign'] == 'cr':
quantity = -quantity
# New balance assertion
assertion = BalanceAssertion(
dt=datetime.strptime(request.form['dt'], '%Y-%m-%d'),
description=request.form['description'],
account=request.form['account'],
quantity=quantity,
commodity='$'
)
db.session.add(assertion)
db.session.commit()
2022-12-24 17:39:18 +11:00
2023-01-07 19:01:05 +11:00
return redirect(url_for('balance_assertions'))
2022-12-24 17:39:18 +11:00
@app.route('/balance-assertions/edit', methods=['GET', 'POST'])
def balance_assertions_edit():
assertion = db.session.get(BalanceAssertion, request.args['id'])
2022-12-24 17:39:18 +11:00
if not assertion:
abort(404)
if request.method == 'GET':
2023-01-02 18:08:30 +11:00
return render_template('journal/balance_assertions_edit.html', assertion=assertion)
2022-12-24 17:39:18 +11:00
quantity = round(float(request.form['amount']) * (10**AMOUNT_DPS))
if request.form['sign'] == 'cr':
quantity = -quantity
# Edit balance assertion
assertion.dt = datetime.strptime(request.form['dt'], '%Y-%m-%d')
assertion.description = request.form['description']
assertion.account = request.form['account']
assertion.quantity = quantity
db.session.commit()
2022-12-24 17:39:18 +11:00
2023-01-07 19:01:05 +11:00
return redirect(url_for('balance_assertions'))