|
|
|
@ -18,12 +18,98 @@ |
|
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>. |
|
|
|
# |
|
|
|
############################################################################## |
|
|
|
from openerp import models, fields |
|
|
|
from openerp import models, fields, api |
|
|
|
from openerp.osv.orm import setup_modifiers |
|
|
|
import StringIO |
|
|
|
import base64 |
|
|
|
import datetime |
|
|
|
from lxml import etree |
|
|
|
from openerp.tools import DEFAULT_SERVER_DATETIME_FORMAT |
|
|
|
import uuid |
|
|
|
|
|
|
|
|
|
|
|
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', required=True, readonly=True) |
|
|
|
binary_file = fields.Binary('File', readonly=True) |
|
|
|
file_name = fields.Char('File Name', readonly=True) |
|
|
|
valid = fields.Boolean() |
|
|
|
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) |
|
|
|
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 |
|
|
|
today = datetime.datetime.now() |
|
|
|
today_tz = fields.Datetime.context_timestamp( |
|
|
|
sql_export, today) |
|
|
|
date = today_tz.strftime(DEFAULT_SERVER_DATETIME_FORMAT) |
|
|
|
output = StringIO.StringIO() |
|
|
|
variable_dict = {} |
|
|
|
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 |
|
|
|
format_query = self.env.cr.mogrify(sql_export.query, variable_dict).decode('utf-8') |
|
|
|
query = "COPY (" + format_query + ") TO STDOUT WITH " + \ |
|
|
|
sql_export.copy_options |
|
|
|
name = 'export_query_%s' % uuid.uuid1().hex |
|
|
|
self.env.cr.execute("SAVEPOINT %s" % name) |
|
|
|
try: |
|
|
|
self.env.cr.copy_expert(query, output) |
|
|
|
output.getvalue() |
|
|
|
new_output = base64.b64encode(output.getvalue()) |
|
|
|
output.close() |
|
|
|
finally: |
|
|
|
self.env.cr.execute("ROLLBACK TO SAVEPOINT %s" % name) |
|
|
|
self.write({ |
|
|
|
'binary_file': new_output, |
|
|
|
'file_name': sql_export.name + '_' + date + '.csv' |
|
|
|
}) |
|
|
|
if not sql_export.valid: |
|
|
|
sql_export.sudo().valid = True |
|
|
|
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, |
|
|
|
} |
Benoit
9 years ago
David Beal