[IMP] partner_financial_risk: New fields in view. Improve code.

partner_financial_risk/models/account_invoice.py

@@ -10,7 +10,8 @@ class AccountInvoice(models.Model):
 
     @api.multi
     def action_invoice_open(self):
-        if not self.env.context.get('bypass_risk', False):
+        if self.env.context.get('bypass_risk', False):
+            return super(AccountInvoice, self).action_invoice_open()
         for invoice in self:
             partner = invoice.partner_id.commercial_partner_id
             exception_msg = ""
@@ -21,6 +22,7 @@ class AccountInvoice(models.Model):
                     partner.risk_invoice_open_limit):
                 exception_msg = _(
                     "This invoice exceeds the open invoices risk.\n")
+            # If risk_invoice_draft_include this invoice included in risk_total
             elif not partner.risk_invoice_draft_include and (
                     partner.risk_invoice_open_include and
                     (partner.risk_total + invoice.amount_total) >

partner_financial_risk/models/res_partner.py

@@ -114,7 +114,7 @@ class ResPartner(models.Model):
 
     @api.multi
     @api.depends(
-        'invoice_ids', 'invoice_ids.state',
+        'customer', 'invoice_ids', 'invoice_ids.state',
         'invoice_ids.amount_total',
         'child_ids.invoice_ids', 'child_ids.invoice_ids.state',
         'child_ids.invoice_ids.amount_total')
@@ -122,11 +122,13 @@ class ResPartner(models.Model):
         all_partners_and_children = {}
         all_partner_ids = []
         for partner in self.filtered('customer'):
+            if not partner.id:
+                continue
             all_partners_and_children[partner] = self.with_context(
                 active_test=False).search([('id', 'child_of', partner.id)]).ids
             all_partner_ids += all_partners_and_children[partner]
         if not all_partner_ids:
-            return  # pragma: no cover
+            return
         total_group = self.env['account.invoice'].sudo().read_group(
             [('type', 'in', ['out_invoice', 'out_refund']),
              ('state', 'in', ['draft', 'proforma', 'proforma2']),

partner_financial_risk/tests/test_partner_financial_risk.py

@@ -70,6 +70,7 @@ class TestPartnerFinancialRisk(SavepointCase):
                 'invoice_line_tax_ids': [(6, 0, [cls.tax.id])],
             })],
         })
+        cls.env.user.lang = False
 
     def test_invoices(self):
         self.partner.risk_invoice_draft_include = True
@@ -142,3 +143,8 @@ class TestPartnerFinancialRisk(SavepointCase):
         line.date_maturity = '2017-01-01'
         self.assertAlmostEqual(self.partner.risk_account_amount, 0.0)
         self.assertAlmostEqual(self.partner.risk_account_amount_unpaid, 100.0)
+
+    def test_recompute_newid(self):
+        """Computing risk shouldn't fail if record is a NewId."""
+        new = self.env['res.partner'].new({'customer': True})
+        new._compute_risk_invoice()

partner_financial_risk/views/res_partner_view.xml

@@ -28,10 +28,13 @@
                                 <field name="risk_account_amount_include"
                                        attrs="{'readonly': [('risk_allow_edit', '=', False)]}"/>
                                 <field name="risk_account_amount" nolabel="1"/>
+                                <field name="risk_account_amount_unpaid_include"
+                                       attrs="{'readonly': [('risk_allow_edit', '=', False)]}"/>
+                                <field name="risk_account_amount_unpaid" nolabel="1"/>
                                 <field name="risk_total" colspan="3" class="oe_subtotal_footer_separator"/>
                             </group>
                         </group>
-                        <group string="Specific Limits" name="risk_limits" col="1">
+                        <group string="Specific Limits" name="risk_limits" col="2">
                             <group class="oe_subtotal_footer">
                                 <field name="risk_invoice_draft_limit"
                                        attrs="{'readonly': [('risk_allow_edit', '=', False)]}"/>
@@ -41,6 +44,8 @@
                                        attrs="{'readonly': [('risk_allow_edit', '=', False)]}"/>
                                 <field name="risk_account_amount_limit"
                                        attrs="{'readonly': [('risk_allow_edit', '=', False)]}"/>
+                                <field name="risk_account_amount_unpaid_limit"
+                                       attrs="{'readonly': [('risk_allow_edit', '=', False)]}"/>
                                 <field name="risk_allow_edit" invisible="1"/>
                             </group>
                         </group>