[FIX+IMP+MIG] contract: Several refinements:

* Remove incorrect oldname attributes.
* Add filter on partners for running contracts (+ a support o2m field for that).
* Cover more tables in model renaming + cleaner code using a loop.
* Don't copy contract lines, but rename table + copy contract records on pre.
* Contract code is now populated to "Reference/Description" field in invoice.
* Order on new contract model has been restored to the same as old analytic accounts.

contract/migrations/12.0.4.0.0/post-migration.py

@@ -8,164 +8,6 @@ from odoo.tools import parse_version
 
 _logger = logging.getLogger(__name__)
 
-CONTRACT_FIELDS = [
-    'id',
-    'name',
-    'partner_id',
-    'pricelist_id',
-    'contract_type',
-    'journal_id',
-    'company_id',
-    'active',
-    'code',
-    'group_id',
-    'contract_template_id',
-    'user_id',
-    'recurring_next_date',
-    'date_end',
-    'message_main_attachment_id',
-    'create_uid',
-    'create_date',
-    'write_uid',
-    'write_date',
-    'payment_term_id',
-    'fiscal_position_id',
-    'invoice_partner_id',
-]
-
-CONTRACT_LINE_FIELDS = [
-    'id',
-    'product_id',
-    'name',
-    'quantity',
-    'uom_id',
-    'automatic_price',
-    'specific_price',
-    'discount',
-    'recurring_rule_type',
-    'recurring_invoicing_type',
-    'recurring_interval',
-    'sequence',
-    'date_start',
-    'date_end',
-    'recurring_next_date',
-    'last_date_invoiced',
-    'termination_notice_date',
-    'successor_contract_line_id',
-    'predecessor_contract_line_id',
-    'manual_renew_needed',
-    'create_uid',
-    'create_date',
-    'write_uid',
-    'write_date',
-]
-
-
-def _get_contract_field_name(cr):
-    """
-    Contract field changed the name from analytic_account_id to contract_id
-    in 12.0.2.0.0. This method used to get the contract field name in
-    account_analytic_invoice_line"""
-    return (
-        'contract_id'
-        if openupgrade.column_exists(
-            cr, 'account_analytic_invoice_line', 'contract_id'
-        )
-        else 'analytic_account_id'
-    )
-
-
-def _copy_contract_table(cr):
-    contract_fields = []
-    for field in CONTRACT_FIELDS:
-        if openupgrade.column_exists(cr, 'account_analytic_account', field):
-            contract_fields.append(field)
-    contract_field_name = _get_contract_field_name(cr)
-    openupgrade.logged_query(
-        cr,
-        "INSERT INTO contract_contract ("
-        + ', '.join(contract_fields)
-        + ") "
-        + "SELECT "
-        + ', '.join(contract_fields)
-        + " FROM account_analytic_account "
-        + "WHERE id in ( SELECT DISTINCT "
-        + contract_field_name
-        + " FROM "
-        + "account_analytic_invoice_line)",
-    )
-    if openupgrade.column_exists(cr, 'account_invoice', 'old_contract_id_tmp'):
-        openupgrade.logged_query(
-            cr,
-            """
-            UPDATE account_invoice
-            SET old_contract_id = old_contract_id_tmp
-            """,
-        )
-
-    # Move contract attachments
-    openupgrade.logged_query(
-        cr,
-        "UPDATE ir_attachment SET res_model='contract.contract'"
-        + "WHERE res_model='account.analytic.account' "
-        + "AND res_id IN ( SELECT DISTINCT "
-        + contract_field_name
-        + " FROM account_analytic_invoice_line)",
-    )
-    # Move contract messages
-    openupgrade.logged_query(
-        cr,
-        "UPDATE mail_message SET model='contract.contract'"
-        + "WHERE model='account.analytic.account' "
-        + "AND res_id IN ( SELECT DISTINCT "
-        + contract_field_name
-        + " FROM account_analytic_invoice_line)",
-    )
-    # Move contract followers
-    openupgrade.logged_query(
-        cr,
-        "UPDATE mail_followers SET res_model='contract.contract'"
-        + "WHERE res_model='account.analytic.account' "
-        + "AND res_id IN ( SELECT DISTINCT "
-        + contract_field_name
-        + " FROM account_analytic_invoice_line)",
-    )
-
-
-def _copy_contract_line_table(cr):
-    contract_line_fields = []
-    contract_field_name = _get_contract_field_name(cr)
-    for field in CONTRACT_LINE_FIELDS:
-        if openupgrade.column_exists(
-            cr, 'account_analytic_invoice_line', field
-        ):
-            contract_line_fields.append(field)
-    account_analytic_invoice_line_fields = contract_line_fields.copy()
-    contract_line_fields.append('contract_id')
-    account_analytic_invoice_line_fields.append(contract_field_name)
-    contract_line_fields.append('analytic_account_id')
-    account_analytic_invoice_line_fields.append(contract_field_name)
-    contract_line_fields.append('active')
-    account_analytic_invoice_line_fields.append('true')
-
-    openupgrade.logged_query(
-        cr,
-        "INSERT INTO contract_line ("
-        + ', '.join(contract_line_fields)
-        + ") "
-        + "SELECT "
-        + ', '.join(account_analytic_invoice_line_fields)
-        + " FROM account_analytic_invoice_line ",
-    )
-
-    openupgrade.logged_query(
-        cr,
-        """
-        UPDATE account_invoice_line
-        SET contract_line_id = contract_line_id_tmp
-        """,
-    )
-
 
 def _update_no_update_ir_cron(env):
     # Update ir.cron
@@ -196,12 +38,7 @@ def _init_invoicing_partner_id_on_contracts(env):
 
 @openupgrade.migrate()
 def migrate(env, version):
-    cr = env.cr
-
-    _copy_contract_table(cr)
-    _copy_contract_line_table(cr)
     _update_no_update_ir_cron(env)
-
     if parse_version(version) < parse_version('12.0.2.0.0'):
         # We check the version here as this post-migration script was in
         # 12.0.2.0.0 and already done for those who used the module when

contract/migrations/12.0.4.0.0/pre-migration.py

@@ -1,19 +1,15 @@
 # Copyright 2019 ACSONE SA/NV
+# Copyright 2019 Tecnativa 2019 - Pedro M. Baeza
 # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
 
 import logging
 
 from openupgradelib import openupgrade
+from psycopg2 import sql
 
 _logger = logging.getLogger(__name__)
 
-
-@openupgrade.migrate()
-def migrate(env, version):
-    _logger.info(">> Pre-Migration 12.0.4.0.0")
-    cr = env.cr
-
-    models_to_rename = [
+models_to_rename = [
     # Contract Line Wizard
     ('account.analytic.invoice.line.wizard', 'contract.line.wizard'),
     # Abstract Contract
@@ -29,14 +25,21 @@ def migrate(env, version):
     ('account.analytic.contract', 'contract.template'),
     # Contract Template Line
     ('account.analytic.contract.line', 'contract.template.line'),
-    ]
-    tables_to_rename = [
+]
+tables_to_rename = [
+    # Contract Line
+    ('account_analytic_invoice_line', 'contract_line'),
     # Contract Template
     ('account_analytic_contract', 'contract_template'),
     # Contract Template Line
     ('account_analytic_contract_line', 'contract_template_line'),
-    ]
-    xmlids_to_rename = [
+]
+columns_to_copy = {
+    'contract_line': [
+        ('analytic_account_id', 'contract_id', None),
+    ],
+}
+xmlids_to_rename = [
     (
         'contract.account_analytic_cron_for_invoice',
         'contract.contract_cron_for_invoice',
@@ -65,49 +68,46 @@ def migrate(env, version):
         'contract.account_analytic_contract_line_user',
         'contract.contract_template_line_user',
     ),
-    ]
-    openupgrade.rename_models(cr, models_to_rename)
-    openupgrade.rename_tables(cr, tables_to_rename)
-    openupgrade.rename_xmlids(cr, xmlids_to_rename)
-    # A temporary column is needed to avoid breaking the foreign key constraint
-    # The temporary column is dropped in the post-migration script
-    openupgrade.logged_query(
-        cr,
+]
+
+
+def _get_contract_field_name(cr):
     """
-        ALTER TABLE account_invoice_line
-        ADD COLUMN contract_line_id_tmp INTEGER
-        """,
-    )
+    Contract field changed the name from analytic_account_id to contract_id
+    in 12.0.2.0.0. This method used to get the contract field name in
+    account_analytic_invoice_line"""
+    return (
+        'contract_id'
         if openupgrade.column_exists(
-        cr, 'account_invoice_line', 'contract_line_id'
-    ):
-        openupgrade.logged_query(
-            cr,
-            """
-            UPDATE account_invoice_line
-            SET contract_line_id_tmp = contract_line_id
-            """,
+            cr, 'account_analytic_invoice_line', 'contract_id'
         )
-        openupgrade.logged_query(
-            cr,
-            """
-            UPDATE account_invoice_line SET contract_line_id = NULL
-            """,
+        else 'analytic_account_id'
     )
-    if not openupgrade.column_exists(
-        cr, 'account_invoice', 'old_contract_id'
-    ):
+
+
+def create_contract_records(cr):
+    contract_field_name = _get_contract_field_name(cr)
     openupgrade.logged_query(
-            cr,
-            """
-            ALTER TABLE account_invoice
-            ADD COLUMN old_contract_id_tmp INTEGER
-            """,
+        cr, """
+        CREATE TABLE contract_contract
+        (LIKE account_analytic_account INCLUDING ALL)""",
     )
     openupgrade.logged_query(
-            cr,
-            """
-            UPDATE account_invoice
-            SET old_contract_id_tmp = contract_id
-            """,
+        cr, sql.SQL("""
+        INSERT INTO contract_contract
+        SELECT * FROM account_analytic_account
+        WHERE id IN (SELECT DISTINCT {} FROM contract_line)
+        """).format(
+            sql.Identifier(contract_field_name),
+        ),
     )
+
+
+@openupgrade.migrate()
+def migrate(env, version):
+    cr = env.cr
+    openupgrade.rename_models(cr, models_to_rename)
+    openupgrade.rename_tables(cr, tables_to_rename)
+    openupgrade.rename_xmlids(cr, xmlids_to_rename)
+    openupgrade.copy_columns(cr, columns_to_copy)
+    create_contract_records(cr)

contract/models/abstract_contract_line.py

@@ -113,7 +113,6 @@ class ContractAbstractContractLine(models.AbstractModel):
         comodel_name='contract.abstract.contract',
         required=True,
         ondelete='cascade',
-        oldname='analytic_account_id',
     )
 
     @api.depends(

contract/models/contract.py

@@ -14,6 +14,7 @@ from odoo.tools.translate import _
 class ContractContract(models.Model):
     _name = 'contract.contract'
     _description = "Contract"
+    _order = 'code, name asc'
     _inherit = [
         'mail.thread',
         'mail.activity.mixin',
@@ -44,7 +45,6 @@ class ContractContract(models.Model):
         comodel_name='contract.line',
         inverse_name='contract_id',
         copy=True,
-        oldnae='contract_line_ids',
     )
 
     user_id = fields.Many2one(
@@ -273,7 +273,7 @@ class ContractContract(models.Model):
         if self.contract_type == 'purchase':
             invoice_type = 'in_invoice'
         return {
-            'reference': self.code,
+            'name': self.code,
             'type': invoice_type,
             'partner_id': self.invoice_partner_id.id,
             'currency_id': currency.id,

contract/models/contract_line.py

@@ -25,7 +25,6 @@ class ContractLine(models.Model):
         required=True,
         index=True,
         ondelete='cascade',
-        oldname='analytic_account_id',
     )
     analytic_account_id = fields.Many2one(
         string="Analytic account",

contract/models/contract_template.py

@@ -19,5 +19,4 @@ class ContractTemplate(models.Model):
         inverse_name='contract_id',
         copy=True,
         string='Contract template lines',
-        oldname='contract_line_ids',
     )

contract/models/res_partner.py

@@ -15,6 +15,11 @@ class ResPartner(models.Model):
         string='Purchase Contracts',
         compute='_compute_contract_count',
     )
+    contract_ids = fields.One2many(
+        comodel_name='contract.contract',
+        inverse='partner_id',
+        string="Contracts",
+    )
 
     def _compute_contract_count(self):
         contract_model = self.env['contract.contract']

contract/views/res_partner_view.xml

@@ -3,7 +3,7 @@
      License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl). -->
 <odoo>
 
-<record id="view_partner_form" model="ir.ui.view">
+    <record id="view_partner_form" model="ir.ui.view">
         <field name="inherit_id" ref="base.view_partner_form" />
         <field name="model">res.partner</field>
         <field type="xml" name="arch">
@@ -22,6 +22,21 @@
             </button>
         </xpath>
         </field>
-</record>
+    </record>
+
+    <record id="view_res_partner_filter" model="ir.ui.view">
+        <field name="inherit_id" ref="base.view_res_partner_filter" />
+        <field name="model">res.partner</field>
+        <field type="xml" name="arch">
+            <filter name="inactive" position="after">
+                <separator/>
+                <filter
+                    name="filter_running_contract"
+                    string="With running contracts"
+                    domain="['|', ('contract_ids.date_end', '&gt;=', context_today().strftime('%Y-%m-%d')), ('contract_ids.date_end', '=', False)]"
+                />
+            </filter>
+        </field>
+    </record>
 
 </odoo>