Allow reconciling two statement lines together as a transfer between accounts

drcr/statements/models.py

@@ -17,7 +17,7 @@
 from sqlalchemy import Column, DateTime, ForeignKey, Integer, String
 from sqlalchemy.orm import relationship
 
-from ..database import Base
+from ..database import Base, db_session
 from ..models import Amount, Posting, Transaction
 
 class StatementLine(Base):
@@ -64,6 +64,13 @@ class StatementLine(Base):
 		
 		return False
 	
+	def delete_postings(self):
+		"""Delete existing StatementLineTransactions with postings for this StatementLine"""
+		
+		for posting in self.postings:
+			# TODO: Will be wonky if transaction covers multiple StatementLines
+			db_session.delete(posting.transaction)
+
 class StatementLineTransaction(Base, Transaction):
 	__tablename__ = 'statement_line_transactions'
 	

drcr/statements/views.py

@@ -37,14 +37,7 @@ def statement_line_charge():
 		abort(404)
 	
 	# Delete existing postings
-	if len(statement_line.postings) > 0:
+	statement_line.delete_postings()
-		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)
 	
 	transaction = StatementLineTransaction(
 		dt=statement_line.dt,
@@ -78,10 +71,7 @@ def statement_line_edit_transaction():
 		return render_template('statements/statement_line_edit_transaction.html', statement_line=statement_line, transaction=statement_line.postings[0].transaction)
 	
 	# Delete existing postings
-	if len(statement_line.postings) > 0:
+	statement_line.delete_postings()
-		for posting in statement_line.postings:
-			# Queue for deletion
-			db_session.delete(posting.transaction)
 	
 	if len(request.form.getlist('charge-account')) == 1:
 		# Simple transaction
@@ -113,3 +103,33 @@ def statement_line_edit_transaction():
 	db_session.commit()
 	
 	return redirect('/statement-lines')
+
+@app.route('/statement-lines/reconcile-transfer', methods=['POST'])
+def statement_line_reconcile_transfer():
+	line_ids = request.form.getlist('sel-line-id')
+	if len(line_ids) != 2:
+		raise Exception('Must select exactly 2 statement lines')
+	
+	line1 = db_session.get(StatementLine, line_ids[0])
+	line2 = db_session.get(StatementLine, line_ids[1])
+	
+	# Check same amount
+	if line1.quantity != -line2.quantity or line1.commodity != line2.commodity:
+		raise Exception('Selected statement line debits/credits must equal')
+	
+	# Delete existing postings
+	line1.delete_postings()
+	line2.delete_postings()
+	
+	transaction = StatementLineTransaction(
+		dt=line1.dt,
+		description=line1.description,
+		postings=[
+			StatementLinePosting(statement_line=line1, description=line1.description, account=line1.source_account, quantity=line1.quantity, commodity=line1.commodity),
+			StatementLinePosting(statement_line=line2, description=line2.description, account=line2.source_account, quantity=line2.quantity, commodity=line2.commodity)
+		]
+	)
+	db_session.add(transaction)
+	db_session.commit()
+	
+	return redirect('/statement-lines')

drcr/templates/statements/statement_lines.html

@@ -22,9 +22,15 @@
 	<div class="container">
 		<h1 class="h2 mt-4 mb-4">Statement lines</h1>
 		
+		<form method="POST">
+			<div class="mb-2">
+				<button type="submit" class="btn btn-outline-secondary" formaction="/statement-lines/reconcile-transfer">Reconcile selected as transfer</button>
+			</div>
+			
 			<table class="table">
 				<thead>
 					<tr>
+						<th></th>
 						<th>Source account</th>
 						<th>Date</th>
 						<th>Description</th>
@@ -37,6 +43,7 @@
 				<tbody>
 					{% for line in statement_lines %}
 						<tr data-line-id="{{ line.id }}">
+							<td><input type="checkbox" name="sel-line-id" value="{{ line.id }}"></td>
 							<td>{{ line.source_account }}</td>
 							<td>{{ line.dt.strftime('%Y-%m-%d') }}</td>
 							<td>{{ line.description }}</td>
@@ -57,6 +64,7 @@
 					{% endfor %}
 				</tbody>
 			</table>
+		</form>
 	</div>
 	
 	<script>