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