flectra/addons/digest/wizard/digest_custom_remove.py

76 lines
4.0 KiB
Python

# -*- coding: utf-8 -*-
# Part of Flectra. See LICENSE file for full copyright and licensing details.
from flectra import api, fields, models, _
from flectra.exceptions import ValidationError, UserError
from lxml import etree
from flectra.tools.safe_eval import test_python_expr
import xml.etree.ElementTree as ET
from flectra.osv import expression
class DigestCustomRemove(models.TransientModel):
_name = 'digest.custom.remove'
remove_type = fields.Selection([('group', 'Group'), ('field', 'Field')], string='Remove Type')
field_id = fields.Many2one('ir.model.fields', 'Field', domain=[('model', '=', 'digest.digest'), ('required', '=', False), ('ttype', '=', 'boolean'), ('name', 'ilike', 'x_kpi_')])
available_group_name = fields.Selection('_get_group_name', string='Available Group')
def _get_group_name(self):
digest_view_id = self.env.ref('digest.digest_digest_view_form').id
view_ids = self.env['ir.ui.view'].search([('inherit_id', 'child_of', digest_view_id)])
group_value = {}
for view_id in view_ids:
root = ET.fromstring(view_id.arch_base)
for group_name in root.iter('group'):
if group_name.attrib.get('name', False) and group_name.attrib.get('string', False) and group_name.attrib['name'].startswith('x_kpi_'):
group_key = str(view_id.id) + '_' + str(group_name.attrib['name'])
group_value.update({group_key : group_name.attrib['string']})
return [(x) for x in group_value.items()]
@api.multi
def action_customize_digest_remove(self):
ir_model_fields_obj = self.env['ir.model.fields']
ir_ui_view_obj = self.env['ir.ui.view']
if self.remove_type == 'group':
find_view_id = self.available_group_name and self.available_group_name.split('_', 1)[0] or False
print("===find_view_id==", find_view_id)
view_ids = ir_ui_view_obj.search([('inherit_id', 'child_of', int(find_view_id))], order="id desc")
field_list = []
for view_id in view_ids:
print("===view_id========", view_id)
root = ET.fromstring(view_id.arch_base)
print("==========root=====", root)
for child in root.iter('group'):
name = child.find('field')
if name.attrib and name.attrib.get('name', False):
field_list.append(name.attrib.get('name', False))
field_ids = ir_model_fields_obj.search([('name', 'in', field_list)])
print("====field_ids====", field_ids, view_ids.ids)
view_ids.unlink()
for field_id in field_ids:
ir_model_fields_obj.search([('depends', '=', field_id.name)]).unlink()
field_ids.unlink()
else:
domain = expression.OR([('arch_db', 'like', record.name)] for record in self.field_id)
print("===domain======", domain)
view_ids = ir_ui_view_obj.search(domain)
print("==========>>>>>>>>.", view_ids)
for view_id in view_ids:
# print("=====view_id.arch_base======before========", view_id.arch_base)
root = ET.fromstring(view_id.arch_base)
# result = len(root.getchildren())
# count = sum(1 for root in root.iter("field"))
# print("===============result==============>", result, count)
for child in root.iter('field'):
# print("===========>>>>",child.text, child.attrib, child.tag)
if child.attrib and child.attrib.get('name', False) == self.field_id.name:
view_id.unlink()
# root.remove(child)
# view_id.write({'arch_base': ET.tostring(root)})
ir_model_fields_obj.search([('depends', '=', self.field_id.name)]).unlink()
self.field_id.unlink()
return {
'type': 'ir.actions.client',
'tag': 'reload',
}