server-tools/database_cleanup/models/create_indexes.py

# Copyright 2017 Therp BV <http://therp.nl>
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
# pylint: disable=consider-merging-classes-inherited
from ..identifier_adapter import IdentifierAdapter
from odoo import api, fields, models


class CreateIndexesLine(models.TransientModel):
    _inherit = 'cleanup.purge.line'
    _name = 'cleanup.create_indexes.line'

    purged = fields.Boolean('Created')
    wizard_id = fields.Many2one('cleanup.create_indexes.wizard')
    field_id = fields.Many2one('ir.model.fields', required=True)

    @api.multi
    def purge(self):
        tables = set()
        for field in self.mapped('field_id'):
            model = self.env[field.model]
            name = '%s_%s_index' % (model._table, field.name)
            self.env.cr.execute(
                'create index %s ON %s (%s)',
                (
                    IdentifierAdapter(name, quote=False),
                    IdentifierAdapter(model._table),
                    IdentifierAdapter(field.name),
                ),
            )
            tables.add(model._table)
        for table in tables:
            self.env.cr.execute(
                'analyze %s', (IdentifierAdapter(model._table),)
            )
        self.write({
            'purged': True,
        })


class CreateIndexesWizard(models.TransientModel):
    _inherit = 'cleanup.purge.wizard'
    _name = 'cleanup.create_indexes.wizard'
    _description = 'Create indexes'

    purge_line_ids = fields.One2many(
        'cleanup.create_indexes.line', 'wizard_id',
    )

    @api.multi
    def find(self):
        res = list()
        for field in self.env['ir.model.fields'].search([
                ('index', '=', True),
        ]):
            if field.model not in self.env.registry:
                continue
            model = self.env[field.model]
            name = '%s_%s_index' % (model._table, field.name)
            self.env.cr.execute(
                'select indexname from pg_indexes '
                'where indexname=%s and tablename=%s',
                (name, model._table)
            )
            if self.env.cr.rowcount:
                continue

            self.env.cr.execute(
                'select a.attname '
                'from pg_attribute a '
                'join pg_class c on a.attrelid=c.oid '
                'join pg_tables t on t.tablename=c.relname '
                'where attname=%s and c.relname=%s',
                (field.name, model._table,)
            )
            if not self.env.cr.rowcount:
                continue

            res.append((0, 0, {
                'name': '%s.%s' % (field.model, field.name),
                'field_id': field.id,
            }))
        return res