59 lines
2.1 KiB
Python
59 lines
2.1 KiB
Python
# coding: utf-8
|
|
|
|
import datetime
|
|
|
|
from odoo import _, api, fields, models
|
|
from odoo.exceptions import ValidationError
|
|
|
|
|
|
class AccountPayment(models.Model):
|
|
_inherit = 'account.payment'
|
|
|
|
payment_transaction_id = fields.Many2one('payment.transaction', string="Payment Transaction")
|
|
payment_token_id = fields.Many2one('payment.token', string="Saved payment token")
|
|
payment_type = fields.Selection(selection_add=[('electronic', 'Electronically receive money')])
|
|
payment_method_id_code = fields.Char(related='payment_method_id.code')
|
|
|
|
@api.onchange('partner_id')
|
|
def _onchange_partner_id(self):
|
|
res = {}
|
|
if self.partner_id:
|
|
res['domain'] = {'payment_token_id': [('partner_id', '=', self.partner_id.id)]}
|
|
|
|
return res
|
|
|
|
@api.onchange('payment_method_id', 'journal_id')
|
|
def _onchange_payment_method(self):
|
|
if self.payment_method_id.code == 'electronic':
|
|
self.payment_token_id = self.env['payment.token'].search([('partner_id', '=', self.partner_id.id)], limit=1)
|
|
else:
|
|
self.payment_token_id = False
|
|
|
|
@api.model
|
|
def create(self, vals):
|
|
account_payment = super(AccountPayment, self).create(vals)
|
|
|
|
if account_payment.payment_token_id:
|
|
account_payment._do_payment()
|
|
return account_payment
|
|
|
|
def _do_payment(self):
|
|
reference = "PAYMENT-%s-%s" % (self.id, datetime.datetime.now().strftime('%y%m%d_%H%M%S'))
|
|
tx = self.env['payment.transaction'].create({
|
|
'amount': self.amount,
|
|
'acquirer_id': self.payment_token_id.acquirer_id.id,
|
|
'type': 'server2server',
|
|
'currency_id': self.currency_id.id,
|
|
'reference': reference,
|
|
'payment_token_id': self.payment_token_id.id,
|
|
'partner_id': self.partner_id.id,
|
|
'partner_country_id': self.partner_id.country_id.id,
|
|
})
|
|
|
|
s2s_result = tx.s2s_do_transaction()
|
|
|
|
if not s2s_result or tx.state != 'done':
|
|
raise ValidationError(_("Payment transaction failed (%s)") % tx.state_message)
|
|
|
|
self.payment_transaction_id = tx
|