Florian da Costa
7 years ago
4 changed files with 6 additions and 141 deletions
-
13sql_export_mail/README.rst
-
1sql_export_mail/wizard/__init__.py
-
106sql_export_mail/wizard/wizard_file.py
-
27sql_export_mail/wizard/wizard_file_view.xml
@ -1 +0,0 @@ |
|||||
from . import wizard_file |
|
||||
@ -1,106 +0,0 @@ |
|||||
# -*- coding: utf-8 -*- |
|
||||
############################################################################## |
|
||||
# |
|
||||
# OpenERP, Open Source Management Solution |
|
||||
# Copyright (C) 2015 Akretion (<http://www.akretion.com>). |
|
||||
# |
|
||||
# This program is free software: you can redistribute it and/or modify |
|
||||
# it under the terms of the GNU Affero General Public License as |
|
||||
# published by the Free Software Foundation, either version 3 of the |
|
||||
# License, or (at your option) any later version. |
|
||||
# |
|
||||
# This program is distributed in the hope that it will be useful, |
|
||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
||||
# GNU Affero General Public License for more details. |
|
||||
# |
|
||||
# You should have received a copy of the GNU Affero General Public License |
|
||||
# along with this program. If not, see <http://www.gnu.org/licenses/>. |
|
||||
# |
|
||||
############################################################################## |
|
||||
|
|
||||
import datetime |
|
||||
from lxml import etree |
|
||||
|
|
||||
from openerp import models, fields, api, osv |
|
||||
from openerp.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 dinamicaly 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._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: |
|
||||
kwargs = {'name': "%s" % field.name} |
|
||||
toupdate_fields.append(field.name) |
|
||||
view_field = etree.SubElement(group, 'field', **kwargs) |
|
||||
osv.orm.setup_modifiers( |
|
||||
view_field, self.fields_get(field.name)) |
|
||||
|
|
||||
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 |
|
||||
|
|
||||
@api.multi |
|
||||
def export_sql(self): |
|
||||
self.ensure_one() |
|
||||
sql_export = self.sql_export_id |
|
||||
|
|
||||
# Manage Params |
|
||||
variable_dict = {} |
|
||||
today = datetime.datetime.now() |
|
||||
today_tz = fields.Datetime.context_timestamp( |
|
||||
sql_export, today) |
|
||||
date = today_tz.strftime(DEFAULT_SERVER_DATETIME_FORMAT) |
|
||||
if sql_export.field_ids: |
|
||||
for field in sql_export.field_ids: |
|
||||
variable_dict[field.name] = self[field.name] |
|
||||
if "%(company_id)s" in sql_export.query: |
|
||||
variable_dict['company_id'] = self.env.user.company_id.id |
|
||||
if "%(user_id)s" in sql_export.query: |
|
||||
variable_dict['user_id'] = self._uid |
|
||||
|
|
||||
# Execute Request |
|
||||
res = sql_export._execute_sql_request( |
|
||||
params=variable_dict, mode='stdout', |
|
||||
copy_options=sql_export.copy_options) |
|
||||
|
|
||||
self.write({ |
|
||||
'binary_file': res, |
|
||||
'file_name': sql_export.name + '_' + date + '.csv' |
|
||||
}) |
|
||||
return { |
|
||||
'view_type': 'form', |
|
||||
'view_mode': 'form', |
|
||||
'res_model': 'sql.file.wizard', |
|
||||
'res_id': self.id, |
|
||||
'type': 'ir.actions.act_window', |
|
||||
'target': 'new', |
|
||||
'context': self._context, |
|
||||
'nodestroy': True, |
|
||||
} |
|
||||
@ -1,27 +0,0 @@ |
|||||
<?xml version="1.0" encoding="utf-8"?> |
|
||||
<openerp> |
|
||||
<data> |
|
||||
|
|
||||
<record id="sql_file_wizard_view_form" model="ir.ui.view"> |
|
||||
<field name="name">sql.file.wizard.view.form</field> |
|
||||
<field name="model">sql.file.wizard</field> |
|
||||
<field name="arch" type="xml"> |
|
||||
<form string="Csv File"> |
|
||||
<separator string="variables_placeholder" colspan="4" invisible="1"/> |
|
||||
<separator string="Export file" colspan="4" |
|
||||
attrs="{'invisible': [('binary_file', '=', False)]}"/> |
|
||||
<field name="binary_file" filename="file_name"/> |
|
||||
<field name="file_name" invisible="1"/> |
|
||||
<footer> |
|
||||
<button name="export_sql" string="Export" type="object" |
|
||||
icon="gtk-apply" /> |
|
||||
or |
|
||||
<button special="cancel" string="Cancel" type="object" |
|
||||
icon="gtk-cancel" /> |
|
||||
</footer> |
|
||||
</form> |
|
||||
</field> |
|
||||
</record> |
|
||||
|
|
||||
</data> |
|
||||
</openerp> |
|
||||
Write
Preview
Loading…
Cancel
Save
Reference in new issue