From ee6d91d811c0169508f38ed691f493530ec5e00c Mon Sep 17 00:00:00 2001
From: RunasSudo <runassudo@yingtongli.me>
Date: Sat, 24 Dec 2022 13:55:19 +1100
Subject: [PATCH] Display existing statement line transaction when editing

---
 drcr/statements/views.py                      | 15 ++++----
 .../statement_line_edit_transaction.html      | 35 +++++++++++++++----
 drcr/templates/statement_lines.html           | 17 +++++----
 3 files changed, 47 insertions(+), 20 deletions(-)

diff --git a/drcr/statements/views.py b/drcr/statements/views.py
index e08bf53..1019580 100644
--- a/drcr/statements/views.py
+++ b/drcr/statements/views.py
@@ -62,16 +62,19 @@ def statement_line_edit_transaction():
 		abort(404)
 	
 	if request.method == 'GET':
-		# FIXME: Show existing transaction details
-		return render_template('statement_line_edit_transaction.html', statement_line=statement_line)
+		if len(statement_line.postings) == 0:
+			return render_template('statement_line_edit_transaction.html', statement_line=statement_line, transaction=None)
+		
+		if len(statement_line.postings) > 1:
+			# NYI
+			raise Exception('Cannot display a StatementLine with >2 postings')
+		
+		# Get existing transaction
+		return render_template('statement_line_edit_transaction.html', statement_line=statement_line, transaction=statement_line.postings[0].transaction)
 	
 	# Delete existing postings
 	if len(statement_line.postings) > 0:
 		for posting in statement_line.postings:
-			if len(posting.transaction.postings) > 2:
-				# Complex posting
-				raise Exception('Cannot automatically delete a StatementLineTransaction with >2 postings')
-			
 			# Queue for deletion
 			db_session.delete(posting.transaction)
 	
diff --git a/drcr/templates/statement_line_edit_transaction.html b/drcr/templates/statement_line_edit_transaction.html
index 0696b7f..3459cd0 100644
--- a/drcr/templates/statement_line_edit_transaction.html
+++ b/drcr/templates/statement_line_edit_transaction.html
@@ -73,13 +73,34 @@
 						{% if statement_line.quantity >= 0 %}<td class="has-amount">{{ statement_line.amount().format() }}</td>{% else %}<td></td>{% endif %}
 						{% if statement_line.quantity < 0 %}<td class="has-amount">{{ (statement_line.amount()|abs).format() }}</td>{% else %}<td></td>{% endif %}
 					</tr>
-					<tr data-side="charge">
-						{# Charge line #}
-						<td></td>
-						<td></td>
-						<td><i>{{ 'Cr' if statement_line.quantity >= 0 else 'Dr' }}</i> <input type="text" name="charge-account"> <a class="text-primary" href="#" onclick="addPosting(this);return false;"><i class="bi bi-plus-circle-fill"></i></a></td>
-						{% if statement_line.quantity < 0 %}<td class="has-amount">{{ (statement_line.amount()|abs).format() }}</td>{% else %}<td></td>{% endif %}
-						{% if statement_line.quantity >= 0 %}<td class="has-amount">{{ statement_line.amount().format() }}</td>{% else %}<td></td>{% endif %}
+					{# Charge lines #}
+					{% if transaction == None %}
+						<tr data-side="charge">
+							<td></td>
+							<td></td>
+							<td><i>{{ 'Cr' if statement_line.quantity >= 0 else 'Dr' }}</i> <input type="text" name="charge-account"> <a class="text-primary" href="#" onclick="addPosting(this);return false;"><i class="bi bi-plus-circle-fill"></i></a></td>
+							{% if statement_line.quantity < 0 %}<td class="has-amount">{{ (statement_line.amount()|abs).format() }}</td>{% else %}<td></td>{% endif %}
+							{% if statement_line.quantity >= 0 %}<td class="has-amount">{{ statement_line.amount().format() }}</td>{% else %}<td></td>{% endif %}
+						</tr>
+					{% elif transaction.postings|length == 2 %}
+						<tr data-side="charge">
+							<td></td>
+							<td></td>
+							<td><i>{{ 'Cr' if statement_line.quantity >= 0 else 'Dr' }}</i> <input type="text" name="charge-account" value="{{ transaction.charge_account(statement_line.source_account) }}"> <a class="text-primary" href="#" onclick="addPosting(this);return false;"><i class="bi bi-plus-circle-fill"></i></a></td>
+							{% if statement_line.quantity < 0 %}<td class="has-amount">{{ (statement_line.amount()|abs).format() }}</td>{% else %}<td></td>{% endif %}
+							{% if statement_line.quantity >= 0 %}<td class="has-amount">{{ statement_line.amount().format() }}</td>{% else %}<td></td>{% endif %}
+						</tr>
+					{% else %}
+						{% for posting in transaction.postings if posting.account != statement_line.source_account %}
+							<tr data-side="charge">
+								<td></td>
+								<td></td>
+								<td><i>{{ 'Cr' if statement_line.quantity >= 0 else 'Dr' }}</i> <input type="text" name="charge-account" value="{{ posting.account }}"> <a class="text-primary" href="#" onclick="addPosting(this);return false;"><i class="bi bi-plus-circle-fill"></i></a></td>
+								{% if statement_line.quantity < 0 %}<td class="has-amount">{{ posting.commodity }}<input type="number" name="charge-amount" step="0.01" value="{{ posting.amount().quantity_string() }}"></td>{% else %}<td></td>{% endif %}
+								{% if statement_line.quantity >= 0 %}<td class="has-amount">{{ posting.commodity }}<input type="number" name="charge-amount" step="0.01" value="{{ (posting.amount()|abs).quantity_string() }}"></td>{% else %}<td></td>{% endif %}
+							</tr>
+						{% endfor %}
+					{% endif %}
 					</tr>
 				</tbody>
 			</table>
diff --git a/drcr/templates/statement_lines.html b/drcr/templates/statement_lines.html
index 679e178..f4711d2 100644
--- a/drcr/templates/statement_lines.html
+++ b/drcr/templates/statement_lines.html
@@ -40,13 +40,16 @@
 						<td>{{ line.source_account }}</td>
 						<td>{{ line.dt.strftime('%Y-%m-%d') }}</td>
 						<td>{{ line.description }}</td>
-						{% if line.postings|length == 0 %}
-							<td class="charge-account"><a href="#" class="text-danger" onclick="classifyLine({{ line.id }});return false;">Unclassified</a> <a href="/statement-lines/edit-transaction?line-id={{ line.id }}" class="text-muted"><i class="bi bi-pencil"></i></a></td>
-						{% elif line.is_complex() %}
-							<td class="charge-account"><i>(Complex)</i></td>
-						{% else %}
-							<td class="charge-account"><a href="#" class="text-body" onclick="classifyLine({{ line.id }});return false;">{{ line.postings[0].transaction.charge_account(line.source_account) }}</a> <a href="/statement-lines/edit-transaction?line-id={{ line.id }}" class="text-muted"><i class="bi bi-pencil"></i></a></td>
-						{% endif %}
+						<td class="charge-account">
+							{% if line.postings|length == 0 %}
+								<a href="#" class="text-danger" onclick="classifyLine({{ line.id }});return false;">Unclassified</a>
+							{% elif line.is_complex() %}
+								<i>(Complex)</i>
+							{% else %}
+								<a href="#" class="text-body" onclick="classifyLine({{ line.id }});return false;">{{ line.postings[0].transaction.charge_account(line.source_account) }}</a>
+							{% endif %}
+							<a href="/statement-lines/edit-transaction?line-id={{ line.id }}" class="text-muted"><i class="bi bi-pencil"></i></a>
+						</td>
 						<td>{{ line.amount().format() if line.quantity >= 0 else '' }}</td>
 						<td>{{ (line.amount()|abs).format() if line.quantity < 0 else '' }}</td>
 						<td>{{ line.balance or '' }}</td>