[14.0][ADD] account_wallet - NOBUMP

account_wallet/models/account_move.py

@@ -15,3 +15,7 @@ class AccountMove(models.Model):
         help="Use this field to give coupon to a customer",
         states={"draft": [("readonly", False)]},
     )
+
+    def action_post(self):
+        self.mapped("line_ids").create_or_set_wallet()
+        return super().action_post()

account_wallet/models/account_move_line.py

@@ -3,8 +3,6 @@
 
 from odoo import api, fields, models
 from odoo.exceptions import ValidationError
-from odoo.osv.expression import FALSE_DOMAIN
-from odoo.tools import float_compare
 from odoo.tools.translate import _
 
 
@@ -31,75 +29,45 @@ def _onchange_account_wallet_id(self):
             if partner:
                 line.partner_id = partner
 
-    def _get_wallet_domain(self, values):
-        account_id = values.get("account_id")
-        partner_id = values.get("partner_id")
-        if not account_id or not partner_id:
-            return FALSE_DOMAIN
-        domain = [
-            ("wallet_type_id.account_id", "=", values["account_id"]),
-            ("partner_id", "=", values["partner_id"]),
-        ]
-        return domain
+    def create_or_set_wallet(self):
+        AccountWallet = self.env["account.wallet"]
+        AccountWalletType = self.env["account.wallet.type"]
 
-    def _get_account_wallet_type(self, values):
-        if values.get("account_id"):
-            wallet_type = self.env["account.wallet.type"].search(
-                [("account_id", "=", values["account_id"])]
+        # check if account/partner is linked to a wallet and assign it
+        # if it the case
+        for move_line in self.filtered(lambda x: x.partner_id):
+            wallet_domain = [
+                ("wallet_type_id.account_id", "=", move_line.account_id.id),
+                ("partner_id", "=", move_line.partner_id.id),
+            ]
+            wallet = AccountWallet.search(wallet_domain)
+            if wallet:
+                move_line.write({"account_wallet_id": wallet.id})
+
+        # Create new wallet, if the account match with the account if a wallet_type
+        for move_line in self.filtered(lambda x: not x.account_wallet_id):
+            wallet_type = AccountWalletType.search(
+                [("account_id", "=", move_line.account_id.id)]
+            )
+            if not wallet_type:
+                continue
+            move_line.account_wallet_id = AccountWallet.create(
+                move_line._prepare_account_wallet_values(wallet_type)
             )
-            return wallet_type
 
-    def _prepare_account_wallet_values(self, values):
-        wallet_type = self._get_account_wallet_type(values)
-        vals = {}
-        if wallet_type:
+    def _prepare_account_wallet_values(self, wallet_type):
+        self.ensure_one()
+        vals = {
+            "wallet_type_id": wallet_type.id,
+        }
+        if wallet_type.no_anonymous:
             vals.update(
                 {
-                    "wallet_type_id": wallet_type.id,
+                    "partner_id": self.partner_id.id,
                 }
             )
-            if wallet_type.no_anonymous:
-                vals.update(
-                    {
-                        "partner_id": values["partner_id"],
-                    }
-                )
         return vals
 
-    def wallet_value(self, vals_list):
-        """If used account is on a wallet type,
-        create a wallet
-        """
-        wallet_obj = self.env["account.wallet"]
-        for values in vals_list:
-            if not values.get("account_wallet_id"):
-                #  check if account/partner is linked to a wallet and assign it
-                # if it the case
-                wallet = wallet_obj.search(self._get_wallet_domain(values))
-                if wallet:
-                    values["account_wallet_id"] = wallet.id
-                else:
-                    # If we try to feed the wallet and none is found, create it
-                    comp = float_compare(
-                        values.get("credit", 0.0), 0.0, precision_digits=2
-                    )
-                    if values.get("account_id") and comp != 0:
-                        wallet_values = self._prepare_account_wallet_values(values)
-                        if wallet_values:
-                            # create wallet
-                            values["account_wallet_id"] = (
-                                self.env["account.wallet"].create(wallet_values).id
-                            )
-            elif values.get("account_wallet_id"):
-                wallet = wallet_obj.browse(values["account_wallet_id"])
-                values["partner_id"] = wallet.partner_id.id or values.get("partner_id")
-        return vals_list
-
-    @api.model_create_multi
-    def create(self, vals_list):
-        vals_list = self.wallet_value(vals_list)
-        return super().create(vals_list)
-
     @api.constrains("account_wallet_id", "account_id")
     def _check_wallet_account(self):
         """Account must correspond to wallet account"""

account_wallet/models/account_wallet.py

@@ -88,13 +88,19 @@ def name_get(self):
 
     @api.model
     def _get_compute_balance_fields(self):
-        return ["account_move_line_ids.debit", "account_move_line_ids.credit"]
+        return [
+            "account_move_line_ids.parent_state",
+            "account_move_line_ids.debit",
+            "account_move_line_ids.credit",
+        ]
 
     @api.depends(lambda self: self._get_compute_balance_fields())
     def _compute_balance(self):
         for wallet in self:
             balance = 0
-            for move_line in wallet.account_move_line_ids:
+            for move_line in wallet.account_move_line_ids.filtered(
+                lambda x: x.parent_state == "posted"
+            ):
                 balance += move_line.credit - move_line.debit
             wallet.balance = balance
 

account_wallet/tests/common.py

@@ -78,7 +78,7 @@ def _create_invoice_credit_wallet(self, credit_amount):
         return invoice, wallet
 
     def _create_payment_move_wallet(self, debit_amount, wallet):
-
+        old_wallet_balance = wallet.balance
         payment_move = self.env["account.move"].create(
             {
                 "journal_id": wallet.wallet_type_id.journal_id.id,
@@ -106,4 +106,7 @@ def _create_payment_move_wallet(self, debit_amount, wallet):
                 ],
             }
         )
+        self.assertEqual(wallet.balance, old_wallet_balance)
         payment_move.action_post()
+        self.assertEqual(wallet.balance, old_wallet_balance - debit_amount)
+        return payment_move