openerp/addons/base/ir/ir_model_relation.py

   1 import logging
   2
   3 import openerp
   4 from openerp import SUPERUSER_ID
   5 from openerp.osv import fields
   6 from openerp.osv.orm import Model
   7
   8 _logger = logging.getLogger(__name__)
   9
  10 class ir_model_relation(Model):
  11     """
  12     This model tracks PostgreSQL tables used to implement OpenERP many2many
  13     relations.
  14     """
  15     _name = 'ir.model.relation'
  16     _columns = {
  17         'name': fields.char('Relation Name', required=True, size=128, select=1,
  18             help="PostgreSQL table name implementing a many2many relation."),
  19         'model': fields.many2one('ir.model', string='Model',
  20             required=True, select=1),
  21         'module': fields.many2one('ir.module.module', string='Module',
  22             required=True, select=1),
  23         'date_update': fields.datetime('Update Date'),
  24         'date_init': fields.datetime('Initialization Date')
  25     }
  26
  27     def _module_data_uninstall(self, cr, uid, ids, context=None):
  28         """
  29         Delete PostgreSQL many2many relations tracked by this model.
  30         """
  31
  32         if uid != SUPERUSER_ID and not self.pool.get('ir.model.access').check_groups(cr, uid, "base.group_system"):
  33             raise except_orm(_('Permission Denied'), (_('Administrator access is required to uninstall a module')))
  34
  35         ids_set = set(ids)
  36         to_drop_table = []
  37         ids.sort()
  38         ids.reverse()
  39         for data in self.browse(cr, uid, ids, context):
  40             model = data.model
  41             model_obj = self.pool.get(model)
  42             name = openerp.tools.ustr(data.name)
  43
  44             # double-check we are really going to delete all the owners of this schema element
  45             cr.execute("""SELECT id from ir_model_relation where name = %s""", (data.name,))
  46             external_ids = [x[0] for x in cr.fetchall()]
  47             if (set(external_ids)-ids_set):
  48                 # as installed modules have defined this element we must not delete it!
  49                 continue
  50
  51             cr.execute("SELECT 1 FROM information_schema.tables WHERE table_name=%s", (name,))
  52             if cr.fetchone() and not name in to_drop_table:
  53                 to_drop_table.append(name)
  54
  55         self.unlink(cr, uid, ids, context)
  56
  57         # drop m2m relation tables
  58         for table in to_drop_table:
  59             cr.execute('DROP TABLE %s CASCADE'% (table),)
  60             _logger.info('Dropped table %s', table)
  61
  62         cr.commit()