DTCLOUD之树形列表批量更改

# -*- coding: utf-8 -*-

# Part of dtcloud. See LICENSE file for full copyright and licensing details.

 

from dtcloud import models, tools, _, api

import xmltodict

import json

 

 

class BaseModel(models.AbstractModel):

    _inherit = 'base'

 

    @api.model

    def load_views(self, views, options=None):

        """ Returns the fields_views of given views, along with the fields of

            the current model, and optionally its filters for the given action.

 

        :param views: list of [view_id, view_type]

        :param options['toolbar']: True to include contextual actions when loading fields_views

        :param options['load_filters']: True to return the model's filters

        :param options['action_id']: id of the action to get the filters

        :return: dictionary with fields_views, fields and optionally filters

        """

        options = options or {}

        result = {}

        toolbar = options.get('toolbar')

        result['fields_views'] = {

            v_type: self.fields_view_get(v_id, v_type if v_type != 'list' else 'tree',

                                         toolbar=toolbar if v_type != 'search' else False)

            for [v_id, v_type] in views

        }

        # TEL TODO 树形列表视图批量更改

        xml_str = result.get('fields_views').get('list').get('arch')

        new_xml_str = self.update_arch_result(xml_str)

        result.get('fields_views').get('list').update({'arch': new_xml_str})

 

        result['fields'] = self.fields_get()

        if options.get('load_filters'):

            result['filters'] = self.env['ir.filters'].get_filters(self._name, options.get('action_id'))

        return result

 

    def update_arch_result(self, xml_str):

        """

        :param xml_str: str树形视图

        :return: 新的树形视图结构

        """

        # 返回一个OrderedDict类型的对象

        new_dict_obj = xmltodict.parse(xml_str, encoding='UTF-8')

        # 使用内置的json模块转换成json字符串

        json_str = json.dumps(new_dict_obj, ensure_ascii=False)

        # 转换Json格式

        jsons = json.loads(json_str)

        # Json格式

        tree_json = jsons.get('tree')

        export_xlsx_attr = tree_json.get('@export_xlsx', False)

        # 不存在export_xlsx属性则更新

        if not export_xlsx_attr:

            tree_json.update({'@export_xlsx': 0})

        # 将以更改好Json数据重新解析为字符串

        new_xml_str = xmltodict.unparse(jsons)

        # 替换固定表头

        new_xml_str = new_xml_str.replace('<?xml version="1.0" encoding="utf-8"?>', '')

        return new_xml_str