From b805452db8025cecef1d97d89c021a4371ad58c9 Mon Sep 17 00:00:00 2001
From: Cedric Collins <cedric@hibou.io>
Date: Wed, 13 Jul 2022 20:59:55 -0500
Subject: [PATCH] [IMP] sale_payment_web: migrate from 13.0 to 14.0

---
 sale_payment_web/__manifest__.py              |   5 +-
 sale_payment_web/models/sale.py               |   1 -
 sale_payment_web/tests/test_sale_payment.py   |  39 ++--
 sale_payment_web/views/payment_views.xml      |  15 ++
 sale_payment_web/views/sale_views.xml         |   5 +-
 .../wizard/account_payment_register.py        | 195 +++++++++---------
 .../wizard/account_payment_register_views.xml |  27 ++-
 7 files changed, 153 insertions(+), 134 deletions(-)
 create mode 100644 sale_payment_web/views/payment_views.xml

diff --git a/sale_payment_web/__manifest__.py b/sale_payment_web/__manifest__.py
index b60ca6a1..87a26bd9 100755
--- a/sale_payment_web/__manifest__.py
+++ b/sale_payment_web/__manifest__.py
@@ -2,7 +2,7 @@
     'name': 'Sale Payment Web',
     'author': 'Hibou Corp. <hello@hibou.io>',
     'category': 'Sales',
-    'version': '13.0.1.0.0',
+    'version': '14.0.1.0.0',
     'description':
         """
 Sale Payment Web
@@ -21,7 +21,8 @@ Electronic payments will create transactions and automatically reconcile on the
     'data': [
         'security/sale_security.xml',
         'security/ir.model.access.csv',
-        'wizard/account_payment_register_views.xml',
+        'views/payment_views.xml',
         'views/sale_views.xml',
+        'wizard/account_payment_register_views.xml',
     ],
 }
diff --git a/sale_payment_web/models/sale.py b/sale_payment_web/models/sale.py
index eaa7b9fc..dd002d90 100644
--- a/sale_payment_web/models/sale.py
+++ b/sale_payment_web/models/sale.py
@@ -29,7 +29,6 @@ class SaleOrder(models.Model):
             'name': _('Register Payment'),
             'res_model': 'account.payment.register',
             'view_mode': 'form',
-            'view_id': self.env.ref('account.view_account_payment_form_multi').id,
             'context': {'active_ids': self.ids, 'active_model': 'sale.order'},
             'target': 'new',
             'type': 'ir.actions.act_window',
diff --git a/sale_payment_web/tests/test_sale_payment.py b/sale_payment_web/tests/test_sale_payment.py
index 38bd2edf..50d180cf 100644
--- a/sale_payment_web/tests/test_sale_payment.py
+++ b/sale_payment_web/tests/test_sale_payment.py
@@ -1,42 +1,49 @@
 from odoo.addons.sale.tests.test_sale_to_invoice import TestSaleToInvoice
 from odoo.exceptions import UserError
+from odoo.tests import tagged
 
 
+@tagged('post_install', '-at_install')
 class TestSalePayment(TestSaleToInvoice):
 
-    def setUp(self):
-        super(TestSalePayment, self).setUp()
-        self.context = {
+    def _sale_context(self):
+        return {
             'active_model': 'sale.order',
             'active_ids': [self.sale_order.id],
             'active_id': self.sale_order.id,
         }
 
-    def test_payment(self):
+    def test_00_payment(self):
         self.sale_order.action_confirm()
 
-        payment_wizard = self.env['account.payment.register'].with_context(self.context).create({'amount': -15})
+        payment_wizard = self.env['account.payment.register'].with_context(self._sale_context()).create({'amount': -15})
         self.assertTrue(payment_wizard.journal_id)
         with self.assertRaises(UserError):
-            payment_wizard.create_payments()
+            payment_wizard._create_payments()
 
-        payment_wizard = self.env['account.payment.register'].with_context(self.context).create({'amount': 0})
+        payment_wizard = self.env['account.payment.register'].with_context(self._sale_context()).create({'amount': 0})
         self.assertTrue(payment_wizard.journal_id)
         with self.assertRaises(UserError):
-            payment_wizard.create_payments()
+            payment_wizard._create_payments()
 
-        payment_wizard = self.env['account.payment.register'].with_context(self.context).create({})
+        payment_wizard = self.env['account.payment.register'].with_context(self._sale_context()).create({})
         self.assertTrue(payment_wizard.journal_id)
 
-        payment_action = payment_wizard.create_payments()
-        self.assertTrue(isinstance(payment_action, dict))
-        payment = self.env[payment_action['res_model']].browse(payment_action['res_id'])
-        self.assertTrue(payment.exists())
+        payment = payment_wizard._create_payments()
         self.assertEqual(payment.amount, self.sale_order.amount_total)
-
         self.assertEqual(payment.sale_order_id, self.sale_order)
+        self.assertEqual(self.sale_order.manual_amount_remaining, 0.0)
 
-        payment_wizard = self.env['account.payment.register'].with_context(self.context).create({'amount': 15})
+        payment_wizard = self.env['account.payment.register'].with_context(self._sale_context()).create({'amount': 15})
         self.assertTrue(payment_wizard.journal_id)
         with self.assertRaises(UserError):
-            payment_wizard.create_payments()
+            payment_wizard._create_payments()
+            
+    def test_10_partial_payment(self):
+        payment_wizard = self.env['account.payment.register'].with_context(self._sale_context()).create({'amount': 50})
+        self.assertTrue(payment_wizard.journal_id)
+
+        payment = payment_wizard._create_payments()
+        self.assertEqual(payment.amount, 50)
+        self.assertEqual(payment.sale_order_id, self.sale_order)
+        self.assertEqual(self.sale_order.manual_amount_remaining, self.sale_order.amount_total - 50)
diff --git a/sale_payment_web/views/payment_views.xml b/sale_payment_web/views/payment_views.xml
new file mode 100644
index 00000000..ae63961a
--- /dev/null
+++ b/sale_payment_web/views/payment_views.xml
@@ -0,0 +1,15 @@
+<?xml version="1.0" encoding="UTF-8" ?>
+<odoo>
+
+    <record id="view_account_payment_form_inherit" model="ir.ui.view">
+        <field name="name">account.payment.form.inherit</field>
+        <field name="model">account.payment</field>
+        <field name="inherit_id" ref="account.view_account_payment_form"/>
+        <field name="arch" type="xml">
+            <xpath expr="//field[@name='move_id']" position="after">
+                <field name="sale_order_id" attrs="{'invisible': [('state', '!=', 'draft'), ('sale_order_id', '=', False)], 'readonly': [('state', '!=', 'draft')]}" groups="account.group_account_readonly"/>
+            </xpath>
+        </field>
+    </record>
+
+</odoo>
diff --git a/sale_payment_web/views/sale_views.xml b/sale_payment_web/views/sale_views.xml
index f736b90b..cf5a4869 100644
--- a/sale_payment_web/views/sale_views.xml
+++ b/sale_payment_web/views/sale_views.xml
@@ -8,7 +8,10 @@
         <field name="groups_id" eval="[(4, ref('group_payment_web'))]"/>
         <field name="arch" type="xml">
             <xpath expr="//button[@name='action_draft']" position="after">
-                <button name="action_payment_register" type="object" string="Register Payment"/>
+                <button name="action_payment_register" 
+                        type="object" 
+                        context="{'dont_redirect_to_payments': True}"
+                        string="Register Payment"/>
             </xpath>
             <xpath expr="//field[@name='amount_total']" position="after">
                 <field name="manual_amount_remaining" widget="monetary"/>
diff --git a/sale_payment_web/wizard/account_payment_register.py b/sale_payment_web/wizard/account_payment_register.py
index f669f9eb..91c28506 100644
--- a/sale_payment_web/wizard/account_payment_register.py
+++ b/sale_payment_web/wizard/account_payment_register.py
@@ -6,129 +6,124 @@ class AccountPaymentRegister(models.TransientModel):
     _inherit = 'account.payment.register'
 
     sale_order_id = fields.Many2one('sale.order', string='Sale Order')
-    payment_token_id = fields.Many2one('payment.token', string="Saved payment token",
-                                       domain="[('acquirer_id.capture_manually', '=', False), ('company_id', '=', company_id)]",
-                                       help="Note that tokens from acquirers set to only authorize transactions "
-                                            "(instead of capturing the amount) are not available.")
-    company_id = fields.Many2one('res.company')  # technical requirement for acquirer domain
-    partner_id = fields.Many2one('res.partner')  # technical payment mode/token domain
-
-    currency_id = fields.Many2one(related='journal_id.currency_id')
-    amount = fields.Monetary('Amount')
-    so_amount_registered_payment = fields.Monetary(related='sale_order_id.manual_amount_registered_payment')
     so_amount_remaining = fields.Monetary(related='sale_order_id.manual_amount_remaining')
     so_amount_over = fields.Boolean()
 
     @api.onchange('amount')
     def _compute_amount_remaining(self):
-        self.so_amount_over = self.amount > self.so_amount_remaining
+        self.so_amount_over = self.sale_order_id and self.amount > self.so_amount_remaining
 
     @api.model
     def default_get(self, fields):
         # super implementation checks active_ids, but not active_model
         active_ids = self._context.get('active_ids')
-        if self._context.get('active_model') != 'sale.order' or not active_ids:
-            return super(AccountPaymentRegister, self).default_get(fields)
-
-        rec = super(AccountPaymentRegister, self.with_context(active_ids=None)).default_get(fields)
+        if self._context.get('active_model') == 'sale.order' and 'line_ids' in fields:
+            fields.remove('line_ids')
+        rec = super(AccountPaymentRegister, self).default_get(fields)
+        if self._context.get('active_model') != 'sale.order':
+            return rec
         sale_order = self.env['sale.order'].browse(active_ids)
         if len(sale_order) != 1:
             raise UserError('Register payment must be called on a sale order.')
 
         if 'sale_order_id' not in rec:
             rec['sale_order_id'] = sale_order.id
-            rec['so_amount_registered_payment'] = sale_order.manual_amount_registered_payment
-            rec['so_amount_remaining'] = sale_order.manual_amount_remaining
-            if 'amount' not in rec:
-                rec['amount'] = sale_order.manual_amount_remaining
-        if 'company_id' not in rec:
-            rec['company_id'] = sale_order.company_id.id
-        if 'partner_id' not in rec:
-            rec['partner_id'] = sale_order.partner_id.id
+        return rec
+    
+    def _compute_from_lines(self):
+        wizards = self.filtered('sale_order_id')
+        super(AccountPaymentRegister, self - wizards)._compute_from_lines()
+        for wizard in wizards:
+            sale_order = wizard.sale_order_id
             account_receivable = sale_order.partner_id.property_account_receivable_id
             if not account_receivable:
                 raise UserError('Your partner must have an Account Receivable setup.')
-        if 'journal_id' not in rec:
-            rec['journal_id'] = self.env['account.journal'].search([('company_id', '=', sale_order.company_id.id), ('type', 'in', ('bank', 'cash'))], limit=1).id
-        if 'payment_method_id' not in rec:
-            domain = [('payment_type', '=', 'inbound')]
-            rec['payment_method_id'] = self.env['account.payment.method'].search(domain, limit=1).id
-        return rec
-
-    @api.onchange('partner_id', 'payment_method_id', 'journal_id')
-    def _onchange_set_payment_token_id(self):
-        res = {}
-
-        if not self.payment_method_id.code == 'electronic' or not self.partner_id or not self.journal_id:
-            self.payment_token_id = False
-            return res
-
-        partners = self.partner_id | self.partner_id.commercial_partner_id | self.partner_id.commercial_partner_id.child_ids
-        domain = [('partner_id', 'in', partners.ids), ('acquirer_id.journal_id', '=', self.journal_id.id)]
-        self.payment_token_id = self.env['payment.token'].search(domain, limit=1)
-
-        res['domain'] = {'payment_token_id': domain}
-        return res
-
-    @api.onchange('journal_id', 'invoice_ids', 'sale_order_id')
-    def _onchange_journal(self):
-        active_ids = self._context.get('active_ids')
-        if self._context.get('active_model') != 'sale.order' or not active_ids:
-            return super(AccountPaymentRegister, self)._onchange_journal()
-        journal = self.journal_id
-        if journal:
-            domain_payment = [('payment_type', '=', 'inbound'), ('id', 'in', journal.inbound_payment_method_ids.ids)]
-        return {'domain': {'payment_method_id': domain_payment}}
-
-    def _prepare_communication_sale_orders(self, sale_order):
-        return " ".join(o.reference or o.name for o in sale_order)
-
-    def _prepare_payment_vals_sale_orders(self, sale_order):
-        '''Create the payment values.
-
-        :param sale_order: The sale orders to pay. In case of multiple
-            documents.
-        :return: The payment values as a dictionary.
-        '''
+            company = sale_order.company_id
+            amount_currency = sale_order.manual_amount_remaining
+            if sale_order.currency_id == company.currency_id:
+                amount = amount_currency
+            else:
+                amount = sale_order.currency_id._convert(amount_currency, company.currency_id, company, fields.Date.context_today(self))
+            wizard.update({
+                'company_id': company.id,
+                'partner_id': sale_order.partner_id.id,
+                'partner_type': 'customer',
+                'payment_type': 'inbound',
+                'source_currency_id': sale_order.currency_id.id,
+                'source_amount': amount,
+                'source_amount_currency': amount_currency,
+                'can_edit_wizard': True,
+                'can_group_payments': False,
+            })
+    
+    def _compute_communication(self):
+        # The communication can't be computed in '_compute_from_lines' because
+        # it's a compute editable field and then, should be computed in a separated method.    
+        wizards = self.filtered('sale_order_id')
+        super(AccountPaymentRegister, self - wizards)._compute_communication()
+        for wizard in wizards:
+            so = wizard.sale_order_id
+            wizards.communication = so.reference or so.name
+        
+    def _compute_group_payment(self):
+        wizards = self.filtered('sale_order_id')
+        super(AccountPaymentRegister, self - wizards)._compute_group_payment()
+        wizards.write({'group_payment': False})
+        
+    def _compute_journal_id(self):
+        wizards = self.filtered('sale_order_id')
+        super(AccountPaymentRegister, self - wizards)._compute_journal_id()
+        for wizard in wizards:
+            wizard.journal_id = self.env['account.journal'].search([
+                    ('type', 'in', ('bank', 'cash')),
+                    ('company_id', '=', wizard.company_id.id),
+                ], limit=1)
+        
+    def _compute_available_partner_bank_ids(self):
+        wizards = self.filtered('sale_order_id')
+        super(AccountPaymentRegister, self - wizards)._compute_available_partner_bank_ids()
+        for wizard in wizards:
+            wizard.available_partner_bank_ids = wizard.journal_id.bank_account_id
+    
+    def _compute_partner_bank_id(self):
+        wizards = self.filtered('sale_order_id')
+        super(AccountPaymentRegister, self - wizards)._compute_partner_bank_id()
+        wizards.write({'partner_bank_id': None})
+    
+    def _compute_suitable_payment_token_partner_ids(self):
+        """ Override from `payment_fix_register_token` """
+        wizards = self.filtered('sale_order_id')
+        super(AccountPaymentRegister, self - wizards)._compute_suitable_payment_token_partner_ids()
+        for wizard in wizards:
+            partners = wizard.sale_order_id.partner_id
+            commercial_partners = partners.commercial_partner_id
+            children_partners = commercial_partners.child_ids
+            wizard.suitable_payment_token_partner_ids = (partners + commercial_partners + children_partners)._origin
+            
+    def _create_payment_vals_from_sale_order(self):
         if self.amount <= 0:
             raise UserError("You must enter a positive amount.")
         elif self.amount > self.so_amount_remaining:
             raise UserError("You cannot make a payment for more than the difference of the total amount and existing "
                             "payments: %.2f" % self.so_amount_remaining)
-        values = {
-            'journal_id': self.journal_id.id,
-            'payment_method_id': self.payment_method_id.id,
-            'payment_date': self.payment_date,
-            'communication': self._prepare_communication_sale_orders(sale_order),
-            # TODO sale orders need to get to transactions somehow
-            # 'invoice_ids': [(6, 0, invoices.ids)],
-            'payment_type': 'inbound',  # if amount can be negative we need to allow this to be outbound
+        return {
+            'date': self.payment_date,
             'amount': self.amount,
-            'currency_id': sale_order.currency_id.id,
-            'partner_id': sale_order.partner_id.id,
-            'partner_type': 'customer',
-            'payment_token_id': self.payment_token_id.id,
+            'payment_type': self.payment_type,
+            'partner_type': self.partner_type,
+            'ref': self.communication,
+            'journal_id': self.journal_id.id,
+            'currency_id': self.currency_id.id,
+            'partner_id': self.partner_id.id,
+            'partner_bank_id': self.partner_bank_id.id,
+            'payment_method_id': self.payment_method_id.id,
+            'sale_order_id': self.sale_order_id.id,
         }
-        return values
-
-    def get_payments_vals(self):
-        if self.sale_order_id:
-            return [self._prepare_payment_vals_sale_orders(self.sale_order_id)]
-        return super(AccountPaymentRegister, self).get_payments_vals()
-
-    def create_payments(self):
-        if self.sale_order_id:
-            # user may not be able to create payment
-            res = super(AccountPaymentRegister, self.sudo()).create_payments()
-        else:
-            res = super(AccountPaymentRegister, self).create_payments()
-        if res and 'res_id' in res and self.sale_order_id:
-            payment = self.env['account.payment'].browse(res['res_id'])
-            self.sale_order_id.manual_payment_ids += payment
-            if payment.name:  # if we don't have a name, then it started a transaction and that will be in chatter
-                payment_link = payment._get_payment_chatter_link()
-                for order in self.sale_order_id:
-                    order.message_post(body=_('A %s payment has been registered: %s') % (payment.payment_method_code, payment_link))
-            # return a 'dummy' action like object for tests
-            return {'res_id': payment.id, 'res_model': payment._name}
-        return res
+            
+    def _create_payments(self):
+        self.ensure_one()
+        if not self.sale_order_id:
+            return super(AccountPaymentRegister, self)._create_payments()
+        payments = self.env['account.payment'].create(self._create_payment_vals_from_sale_order())
+        payments.action_post()
+        return payments
diff --git a/sale_payment_web/wizard/account_payment_register_views.xml b/sale_payment_web/wizard/account_payment_register_views.xml
index 02eb97f3..b5e3671d 100644
--- a/sale_payment_web/wizard/account_payment_register_views.xml
+++ b/sale_payment_web/wizard/account_payment_register_views.xml
@@ -1,29 +1,28 @@
 <?xml version="1.0" encoding="UTF-8" ?>
 <odoo>
 
-    <record id="view_account_payment_form_multi_inherit" model="ir.ui.view">
+    <record id="view_account_payment_register_form_inherit" model="ir.ui.view">
         <field name="name">account.payment.form.multi.inherit</field>
         <field name="model">account.payment.register</field>
-        <field name="inherit_id" ref="account.view_account_payment_form_multi"/>
+        <field name="inherit_id" ref="account.view_account_payment_register_form"/>
         <field name="arch" type="xml">
-            <xpath expr="//group/group[1]" position="inside">
-                <field name="company_id" invisible="1" />
-                <field name="partner_id" invisible="1" />
-                <field name="payment_token_id" nocreate="1" />
-            </xpath>
-            <xpath expr="//field[@name='payment_date']" position="after">
-                <field name="currency_id" invisible="True"/>
-                <field name="amount"/>
-                <field name="so_amount_remaining" invisible="1"/>
+            <xpath expr="//field[@name='company_currency_id']" position="after">
+                <field name="sale_order_id" invisible="1"/>
                 <field name="so_amount_over" invisible="1"/>
-                <div attrs="{'invisible':[('so_amount_over', '=', False)]}" class="text-danger" colspan="4">
+                <field name="so_amount_remaining" invisible="1"/>
+            </xpath>
+            <xpath expr="//field[@name='payment_difference_handling']" position="attributes">
+                <attribute name="attrs">{'invisible': [('sale_order_id', '!=', False)]}</attribute>
+            </xpath>
+            <xpath expr="//group[@name='group3']" position="after">
+                <group name="group_so_amount_over" attrs="{'invisible':[('so_amount_over', '=', False)]}" class="text-danger" colspan="4">
                     <p>
                         You cannot make a payment for more than the difference of the total amount and existing payments:
                         <field name="so_amount_remaining"/>
                     </p>
-                </div>
+                </group>
             </xpath>
         </field>
     </record>
 
-</odoo>
\ No newline at end of file
+</odoo>