data-protection/privacy_partner_report/report/privacy_partner_xlsx.py

# Copyright 2018 Eficent Business and IT Consulting Services S.L.
# License AGPL-3 - See http://www.gnu.org/licenses/agpl-3.0.html

import logging

from odoo import models
from odoo.tools.translate import _

_logger = logging.getLogger(__name__)


class ReportPartnerXlsx(models.AbstractModel):
    _name = "report.privacy_partner_report.report_partner_xlsx"
    _inherit = "report.report_xlsx.abstract"

    def _search_longest_row(self, tables):
        res = 0
        for table in tables:
            for model in tables[table]:
                if len(tables[table][model]) > 0:
                    if len(tables[table][model][0]) > res:
                        res = len(tables[table][model][0])
        return res

    def generate_xlsx_report(self, workbook, data, objects):
        for o in objects:
            report_data = o.compute_data_for_report(data)
            partner = report_data["form"].get("partner_id", False)
            partner = self.env["res.partner"].sudo().browse(partner[0])
            workbook.set_properties(
                {"comments": "Created with Python and XlsxWriter from Odoo"}
            )
            sheet = workbook.add_worksheet(_("Partner Data"))
            sheet.set_landscape()
            sheet.fit_to_pages(1, 0)
            sheet.set_zoom(75)
            sheet.set_column(0, self._search_longest_row(report_data["tables"]), 25)
            title_style = workbook.add_format(
                {"bold": True, "bg_color": "#FFFFCC", "border": 2}
            )
            sheet.set_row(0, None, None, {"collapsed": 1})
            sheet.write_row(1, 0, ["Partner: " + partner.display_name], title_style)
            i = 3
            first_row = i + 2
            for table in sorted(report_data["tables"].keys()):
                for model in sorted(report_data["tables"][table].keys()):
                    rows = len(report_data["tables"][table][model])
                    if rows:
                        style = workbook.add_format()
                        style.set_bold(True)
                        style.set_border(2)
                        sheet.write_row(i, 0, [model], style)
                        i += 1
                        j = 0
                        for column in report_data["tables"][table][model][0]:
                            style = workbook.add_format()
                            style.set_bold(True)
                            if j == 0:
                                style.set_left(1)
                            if j == len(report_data["tables"][table][model][0]) - 1:
                                style.set_right(1)
                            style.set_top(1)
                            style.set_bottom(1)
                            sheet.write_row(i, j, [column], style)
                            j += 1
                        for row in report_data["tables"][table][model]:
                            i += 1
                            j = 0
                            for column in row:
                                style = workbook.add_format()
                                if j == 0:
                                    style.set_left(1)
                                if j == len(row) - 1:
                                    style.set_right(1)
                                if i == rows + first_row - 1:
                                    style.set_bottom(1)
                                if row[column]:
                                    sheet.write_row(i, j, [row[column]], style)
                                else:
                                    sheet.write_row(i, j, [""], style)
                                j += 1
                    i += 2
                    first_row = i + 2