mirror of
https://gitlab.com/flectra-community/reporting-engine.git
synced 2024-11-15 02:32:05 +00:00
78 lines
2.7 KiB
Python
78 lines
2.7 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).
|
||
|
|
|
||
|
|
from datetime import datetime
|
||
|
|
|
||
|
|
from flectra import _, fields, models
|
||
|
|
from flectra.exceptions import UserError
|
||
|
|
from flectra.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(readonly=True)
|
||
|
|
sql_export_id = fields.Many2one(comodel_name="sql.export", required=True)
|
||
|
|
query_properties = fields.Properties(
|
||
|
|
string="Properties",
|
||
|
|
definition="sql_export_id.query_properties_definition",
|
||
|
|
copy=False,
|
||
|
|
)
|
||
|
|
|
||
|
|
def export_sql(self):
|
||
|
|
self.ensure_one()
|
||
|
|
|
||
|
|
properties = self.read(["query_properties"])[0]["query_properties"]
|
||
|
|
|
||
|
|
# Check properties
|
||
|
|
bad_props = [x for x in properties if not x["value"]]
|
||
|
|
if bad_props:
|
||
|
|
raise UserError(
|
||
|
|
_("Please enter a values for the following properties : %s")
|
||
|
|
% (",".join([x["string"] for x in bad_props]))
|
||
|
|
)
|
||
|
|
|
||
|
|
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)
|
||
|
|
for prop in properties:
|
||
|
|
if prop["type"] == "many2many":
|
||
|
|
m2m_ids = []
|
||
|
|
for m2m_id in prop["value"]:
|
||
|
|
m2m_ids.append(m2m_id[0])
|
||
|
|
variable_dict[prop["string"]] = tuple(m2m_ids)
|
||
|
|
else:
|
||
|
|
variable_dict[prop["string"]] = prop["value"]
|
||
|
|
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": f"{sql_export.name}_{date}.{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,
|
||
|
|
}
|