# -*- coding: utf-8 -*- # Part of Odoo, Flectra. See LICENSE file for full copyright and licensing details. import logging from flectra import api, fields, models, _ from flectra.exceptions import UserError from flectra.tools import float_compare _logger = logging.getLogger(__name__) class PaymentTransaction(models.Model): _inherit = 'payment.transaction' # YTI FIXME: The auto_join seems useless sale_order_id = fields.Many2one('sale.order', string='Sales Order', auto_join=True) so_state = fields.Selection('sale.order', string='Sale Order State', related='sale_order_id.state') @api.model def form_feedback(self, data, acquirer_name): """ Override to confirm the sales order, if defined, and if the transaction is done. """ tx = None res = super(PaymentTransaction, self).form_feedback(data, acquirer_name) # fetch the tx tx_find_method_name = '_%s_form_get_tx_from_data' % acquirer_name if hasattr(self, tx_find_method_name): tx = getattr(self, tx_find_method_name)(data) _logger.info('<%s> transaction processed: tx ref:%s, tx amount: %s', acquirer_name, tx.reference if tx else 'n/a', tx.amount if tx else 'n/a') if tx and tx.sale_order_id: # Auto-confirm SO if necessary tx._confirm_so() return res # -------------------------------------------------- # Sale management # -------------------------------------------------- def _confirm_so(self): """ Check tx state, confirm the potential SO """ self.ensure_one() if self.sale_order_id.state not in ['draft', 'sent', 'sale']: _logger.warning('<%s> transaction STATE INCORRECT for order %s (ID %s, state %s)', self.acquirer_id.provider, self.sale_order_id.name, self.sale_order_id.id, self.sale_order_id.state) return 'pay_sale_invalid_doc_state' if not float_compare(self.amount, self.sale_order_id.amount_total, 2) == 0: _logger.warning( '<%s> transaction AMOUNT MISMATCH for order %s (ID %s): expected %r, got %r', self.acquirer_id.provider, self.sale_order_id.name, self.sale_order_id.id, self.sale_order_id.amount_total, self.amount, ) self.sale_order_id.message_post( subject=_("Amount Mismatch (%s)") % self.acquirer_id.provider, body=_("The sale order was not confirmed despite response from the acquirer (%s): SO amount is %r but acquirer replied with %r.") % ( self.acquirer_id.provider, self.sale_order_id.amount_total, self.amount, ) ) return 'pay_sale_tx_amount' if self.state == 'authorized' and self.acquirer_id.capture_manually: _logger.info('<%s> transaction authorized, auto-confirming order %s (ID %s)', self.acquirer_id.provider, self.sale_order_id.name, self.sale_order_id.id) if self.sale_order_id.state in ('draft', 'sent'): self.sale_order_id.with_context(send_email=True).action_confirm() elif self.state == 'done': _logger.info('<%s> transaction completed, auto-confirming order %s (ID %s)', self.acquirer_id.provider, self.sale_order_id.name, self.sale_order_id.id) if self.sale_order_id.state in ('draft', 'sent'): self.sale_order_id.with_context(send_email=True).action_confirm() elif self.state not in ['cancel', 'error'] and self.sale_order_id.state == 'draft': _logger.info('<%s> transaction pending/to confirm manually, sending quote email for order %s (ID %s)', self.acquirer_id.provider, self.sale_order_id.name, self.sale_order_id.id) self.sale_order_id.force_quotation_send() else: _logger.warning('<%s> transaction MISMATCH for order %s (ID %s)', self.acquirer_id.provider, self.sale_order_id.name, self.sale_order_id.id) return 'pay_sale_tx_state' return True def _generate_and_pay_invoice(self): self.sale_order_id._force_lines_to_invoice_policy_order() # force company to ensure journals/accounts etc. are correct # company_id needed for default_get on account.journal # force_company needed for company_dependent fields ctx_company = {'company_id': self.sale_order_id.company_id.id, 'force_company': self.sale_order_id.company_id.id} # We might fail to create the invoice because there is no invoiceable lines. This will # raise a UserError and break the workflow. Better catch the error. try: created_invoice = self.sale_order_id.with_context(**ctx_company).action_invoice_create() created_invoice = self.env['account.invoice'].browse(created_invoice).with_context(**ctx_company) except UserError: _logger.warning('<%s> transaction completed, could not auto-generate invoice for %s (ID %s)', self.acquirer_id.provider, self.sale_order_id.name, self.sale_order_id.id, exc_info=True) return if created_invoice: _logger.info('<%s> transaction completed, auto-generated invoice %s (ID %s) for %s (ID %s)', self.acquirer_id.provider, created_invoice.name, created_invoice.id, self.sale_order_id.name, self.sale_order_id.id) created_invoice.action_invoice_open() if not self.acquirer_id.journal_id: default_journal = self.env['account.journal'].search([('type', '=', 'bank')], limit=1) if not default_journal: _logger.warning('<%s> transaction completed, could not auto-generate payment for %s (ID %s) (no journal set on acquirer)', self.acquirer_id.provider, self.sale_order_id.name, self.sale_order_id.id) return False self.acquirer_id.journal_id = default_journal # TDE Note: in post-v11 we could probably add an explicit parameter + update account.payment tx directly at creation, not afterwards created_invoice.with_context(default_currency_id=self.currency_id.id).pay_and_reconcile(self.acquirer_id.journal_id, pay_amount=created_invoice.amount_total) if created_invoice.payment_ids: created_invoice.payment_ids[0].payment_transaction_id = self self._post_process_after_done(invoice_id=created_invoice) else: _logger.warning('<%s> transaction completed, could not auto-generate invoice for %s (ID %s)', self.acquirer_id.provider, self.sale_order_id.name, self.sale_order_id.id) # -------------------------------------------------- # Tools for payment # -------------------------------------------------- def confirm_sale_token(self): """ Confirm a transaction token and call SO confirmation if it is a success. :return: True if success; error string otherwise """ self.ensure_one() if self.payment_token_id and self.partner_id == self.sale_order_id.partner_id: try: s2s_result = self.s2s_do_transaction() except Exception as e: _logger.warning( _("<%s> transaction (%s) failed: <%s>") % (self.acquirer_id.provider, self.id, str(e))) return 'pay_sale_tx_fail' valid_state = 'authorized' if self.acquirer_id.capture_manually else 'done' if not s2s_result or self.state != valid_state: _logger.warning( _("<%s> transaction (%s) invalid state: %s") % (self.acquirer_id.provider, self.id, self.state_message)) return 'pay_sale_tx_state' try: return self._confirm_so() except Exception as e: _logger.warning( _("<%s> transaction (%s) order confirmation failed: <%s>") % (self.acquirer_id.provider, self.id, str(e))) return 'pay_sale_tx_confirm' return 'pay_sale_tx_token' def _check_or_create_sale_tx(self, order, acquirer, payment_token=None, tx_type='form', add_tx_values=None, reset_draft=True): tx = self if not tx: tx = self.search([('reference', '=', order.name)], limit=1) if tx.state in ['error', 'cancel']: # filter incorrect states tx = False if (tx and tx.acquirer_id != acquirer) or (tx and tx.sale_order_id != order): # filter unmatching tx = False if tx and payment_token and tx.payment_token_id and payment_token != tx.payment_token_id: # new or distinct token tx = False # still draft tx, no more info -> rewrite on tx or create a new one depending on parameter if tx and tx.state == 'draft': if reset_draft: tx.write(dict( self.on_change_partner_id(order.partner_id.id).get('value', {}), amount=order.amount_total, type=tx_type) ) else: tx = False if not tx: tx_values = { 'acquirer_id': acquirer.id, 'type': tx_type, 'amount': order.amount_total, 'currency_id': order.pricelist_id.currency_id.id, 'partner_id': order.partner_id.id, 'partner_country_id': order.partner_id.country_id.id, 'reference': self._get_next_reference(order.name, acquirer=acquirer), 'sale_order_id': order.id, } if add_tx_values: tx_values.update(add_tx_values) if payment_token and payment_token.sudo().partner_id == order.partner_id: tx_values['payment_token_id'] = payment_token.id tx = self.create(tx_values) # update quotation order.write({ 'payment_tx_id': tx.id, }) return tx def render_sale_button(self, order, return_url, submit_txt=None, render_values=None): values = { 'return_url': return_url, 'partner_id': order.partner_shipping_id.id or order.partner_invoice_id.id, 'billing_partner_id': order.partner_invoice_id.id, } if render_values: values.update(render_values) return self.acquirer_id.with_context(submit_class='btn btn-primary', submit_txt=submit_txt or _('Pay Now')).sudo().render( self.reference, order.amount_total, order.pricelist_id.currency_id.id, values=values, )