lnbits-exts/satmachineclient
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Add crediting functionality for source wallet in transaction processing: implement credit_source_wallet method to credit the source wallet with the full crypto amount from Lamassu transactions. Update process_transaction method to call this new functionality before calculating DCA distributions, ensuring accurate wallet balance management.

Browse source
This commit is contained in:
padreug 2025-06-19 16:58:44 +02:00
parent 1c1f358d82
commit 033e329870
1 changed files with 32 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

32
transaction_processor.py
View file

@ -23,6 +23,7 @@ except ImportError:
from lnbits.core.services import create_invoice, pay_invoice
from lnbits.core.crud.wallets import get_wallet
from lnbits.core.services import update_wallet_balance
from lnbits.settings import settings
from .crud import (
@ -680,6 +681,31 @@ class LamassuTransactionProcessor:
logger.error(f"Error sending DCA payment to client {client.username or client.user_id}: {e}")
return False
async def credit_source_wallet(self, transaction: Dict[str, Any]) -> bool:
"""Credit the source wallet with the full crypto_atoms amount from Lamassu transaction"""
try:
# Get the configuration to find source wallet
admin_config = await get_active_lamassu_config()
if not admin_config or not admin_config.source_wallet_id:
logger.error("No source wallet configured - cannot credit wallet")
return False
crypto_atoms = transaction["crypto_amount"] # Full amount including commission
transaction_id = transaction["transaction_id"]
# Credit the source wallet with the full crypto_atoms amount
await update_wallet_balance(
wallet_id=admin_config.source_wallet_id,
amount=crypto_atoms * 1000 # Convert sats to millisats
)
logger.info(f"Credited source wallet with {crypto_atoms} sats from transaction {transaction_id}")
return True
except Exception as e:
logger.error(f"Error crediting source wallet for transaction {transaction.get('transaction_id', 'unknown')}: {e}")
return False
async def process_transaction(self, transaction: Dict[str, Any]) -> None:
"""Process a single transaction - calculate and distribute DCA payments"""
try:
@ -693,6 +719,12 @@ class LamassuTransactionProcessor:
logger.info(f"Processing new transaction: {transaction_id}")
# First, credit the source wallet with the full transaction amount
credit_success = await self.credit_source_wallet(transaction)
if not credit_success:
logger.error(f"Failed to credit source wallet for transaction {transaction_id} - skipping distribution")
return
# Calculate distribution amounts
distributions = await self.calculate_distribution_amounts(transaction)