You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

122 lines
4.6 KiB

  1. # -*- coding: utf-8 -*-
  2. ##############################################################################
  3. #
  4. # OpenERP, Open Source Management Solution
  5. # This module copyright (C) 2014 Therp BV (<http://therp.nl>).
  6. #
  7. # This program is free software: you can redistribute it and/or modify
  8. # it under the terms of the GNU Affero General Public License as
  9. # published by the Free Software Foundation, either version 3 of the
  10. # License, or (at your option) any later version.
  11. #
  12. # This program is distributed in the hope that it will be useful,
  13. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  15. # GNU Affero General Public License for more details.
  16. #
  17. # You should have received a copy of the GNU Affero General Public License
  18. # along with this program. If not, see <http://www.gnu.org/licenses/>.
  19. #
  20. ##############################################################################
  21. from openerp.osv import orm, fields
  22. from openerp.tools.translate import _
  23. from openerp.addons.base.ir.ir_model import MODULE_UNINSTALL_FLAG
  24. class IrModel(orm.Model):
  25. _inherit = 'ir.model'
  26. def _drop_table(self, cr, uid, ids, context=None):
  27. # Allow to skip this step during model unlink
  28. # The super method crashes if the model cannot be instantiated
  29. if context and context.get('no_drop_table'):
  30. return True
  31. return super(IrModel, self)._drop_table(cr, uid, ids, context=context)
  32. class CleanupPurgeLineModel(orm.TransientModel):
  33. _inherit = 'cleanup.purge.line'
  34. _name = 'cleanup.purge.line.model'
  35. _columns = {
  36. 'wizard_id': fields.many2one(
  37. 'cleanup.purge.wizard.model', 'Purge Wizard', readonly=True),
  38. }
  39. def purge(self, cr, uid, ids, context=None):
  40. """
  41. Unlink models upon manual confirmation.
  42. """
  43. model_pool = self.pool['ir.model']
  44. attachment_pool = self.pool['ir.attachment']
  45. constraint_pool = self.pool['ir.model.constraint']
  46. fields_pool = self.pool['ir.model.fields']
  47. local_context=(context or {}).copy()
  48. local_context.update({
  49. MODULE_UNINSTALL_FLAG: True,
  50. 'no_drop_table': True,
  51. })
  52. for line in self.browse(cr, uid, ids, context=context):
  53. cr.execute(
  54. "SELECT id, model from ir_model WHERE model = %s",
  55. (line.name,))
  56. row = cr.fetchone()
  57. if row:
  58. self.logger.info('Purging model %s', row[1])
  59. attachment_ids = attachment_pool.search(
  60. cr, uid, [('res_model', '=', line.name)], context=context)
  61. if attachment_ids:
  62. cr.execute(
  63. "UPDATE ir_attachment SET res_model = FALSE "
  64. "WHERE id in %s",
  65. (tuple(attachment_ids), ))
  66. constraint_ids = constraint_pool.search(
  67. cr, uid, [('model', '=', line.name)], context=context)
  68. if constraint_ids:
  69. constraint_pool.unlink(
  70. cr, uid, constraint_ids, context=context)
  71. relation_ids = fields_pool.search(
  72. cr, uid, [('relation', '=', row[1])], context=context)
  73. if relation_ids:
  74. fields_pool.unlink(cr, uid, relation_ids,
  75. context=local_context)
  76. model_pool.unlink(cr, uid, [row[0]], context=local_context)
  77. line.write({'purged': True})
  78. cr.commit()
  79. return True
  80. class CleanupPurgeWizardModel(orm.TransientModel):
  81. _inherit = 'cleanup.purge.wizard'
  82. _name = 'cleanup.purge.wizard.model'
  83. def default_get(self, cr, uid, fields, context=None):
  84. res = super(CleanupPurgeWizardModel, self).default_get(
  85. cr, uid, fields, context=context)
  86. if 'name' in fields:
  87. res['name'] = _('Purge models')
  88. return res
  89. def find(self, cr, uid, context=None):
  90. """
  91. Search for models that cannot be instantiated.
  92. """
  93. res = []
  94. cr.execute("SELECT model from ir_model")
  95. for (model,) in cr.fetchall():
  96. if not self.pool.get(model):
  97. res.append((0, 0, {'name': model}))
  98. if not res:
  99. raise orm.except_orm(
  100. _('Nothing to do'),
  101. _('No orphaned models found'))
  102. return res
  103. _columns = {
  104. 'purge_line_ids': fields.one2many(
  105. 'cleanup.purge.line.model',
  106. 'wizard_id', 'Models to purge'),
  107. }