mirror of
https://github.com/OCA/reporting-engine.git
synced 2025-02-16 16:30:38 +02:00
aa39365841
Improve usability about variable creation and choice by hiding base wizard base fields and display simplified view Add a description on paramter fields to help the user to use parameters If not parameter are used for the query, display the wizard with the generated file directly
100 lines
4.0 KiB
Python
100 lines
4.0 KiB
Python
# Copyright (C) 2015 Akretion (<http://www.akretion.com>)
|
|
# @author: Florian da Costa
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
|
|
|
|
import json
|
|
from datetime import datetime
|
|
|
|
from lxml import etree
|
|
|
|
from odoo import api, fields, models
|
|
from odoo.tools import DEFAULT_SERVER_DATETIME_FORMAT
|
|
|
|
|
|
class SqlFileWizard(models.TransientModel):
|
|
_name = "sql.file.wizard"
|
|
_description = "Allow the user to save the file with sql request's data"
|
|
|
|
binary_file = fields.Binary("File", readonly=True)
|
|
file_name = fields.Char("File Name", readonly=True)
|
|
sql_export_id = fields.Many2one(comodel_name="sql.export", required=True)
|
|
|
|
@api.model
|
|
def fields_view_get(
|
|
self, view_id=None, view_type="form", toolbar=False, submenu=False
|
|
):
|
|
"""
|
|
Display dynamically parameter fields depending on the sql_export.
|
|
"""
|
|
res = super(SqlFileWizard, self).fields_view_get(
|
|
view_id=view_id, view_type=view_type, toolbar=toolbar, submenu=submenu
|
|
)
|
|
export_obj = self.env["sql.export"]
|
|
if view_type == "form":
|
|
sql_export = export_obj.browse(self.env.context.get("active_id"))
|
|
if sql_export.field_ids:
|
|
eview = etree.fromstring(res["arch"])
|
|
group = etree.Element("group", name="variables_group", colspan="4")
|
|
toupdate_fields = []
|
|
for field in sql_export.field_ids:
|
|
toupdate_fields.append(field.name)
|
|
attrib = {"name": field.name, "required": "0", "readonly": "0"}
|
|
view_field = etree.SubElement(group, "field", attrib=attrib)
|
|
modifiers = json.loads(view_field.get("modifiers", "{}"))
|
|
if field.required:
|
|
modifiers["required"] = True
|
|
view_field.set("modifiers", json.dumps(modifiers))
|
|
|
|
res["fields"].update(self.fields_get(toupdate_fields))
|
|
placeholder = eview.xpath(
|
|
"//separator[@string='variables_placeholder']"
|
|
)[0]
|
|
placeholder.getparent().replace(placeholder, group)
|
|
res["arch"] = etree.tostring(eview, pretty_print=True)
|
|
return res
|
|
|
|
def export_sql(self):
|
|
self.ensure_one()
|
|
sql_export = self.sql_export_id
|
|
|
|
# Manage Params
|
|
variable_dict = {}
|
|
now_tz = fields.Datetime.context_timestamp(sql_export, datetime.now())
|
|
date = now_tz.strftime(DEFAULT_SERVER_DATETIME_FORMAT)
|
|
if sql_export.field_ids:
|
|
for field in sql_export.field_ids:
|
|
if field.ttype == "many2one":
|
|
variable_dict[field.name] = self[field.name].id
|
|
elif field.ttype == "many2many":
|
|
variable_dict[field.name] = tuple(self[field.name].ids)
|
|
else:
|
|
variable_dict[field.name] = self[field.name]
|
|
if "%(company_id)s" in sql_export.query:
|
|
company_id = self.env.company.id
|
|
variable_dict["company_id"] = company_id
|
|
if "%(user_id)s" in sql_export.query:
|
|
user_id = self.env.user.id
|
|
variable_dict["user_id"] = user_id
|
|
|
|
# Call different method depending on file_type since the logic will be
|
|
# different
|
|
method_name = "%s_get_data_from_query" % sql_export.file_format
|
|
data = getattr(sql_export, method_name)(variable_dict)
|
|
extension = sql_export._get_file_extension()
|
|
self.write(
|
|
{
|
|
"binary_file": data,
|
|
"file_name": "%(name)s_%(date)s.%(extension)s"
|
|
% {"name": sql_export.name, "date": date, "extension": extension},
|
|
}
|
|
)
|
|
return {
|
|
"view_mode": "form",
|
|
"res_model": "sql.file.wizard",
|
|
"res_id": self.id,
|
|
"type": "ir.actions.act_window",
|
|
"target": "new",
|
|
"context": self.env.context,
|
|
"nodestroy": True,
|
|
}
|