You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
70 lines
2.1 KiB
70 lines
2.1 KiB
# 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 odoo import models, fields
|
|
|
|
|
|
class SqlExport(models.Model):
|
|
_name = "sql.export"
|
|
_inherit = ['sql.request.mixin']
|
|
_description = "SQL export"
|
|
|
|
_sql_request_groups_relation = 'groups_sqlquery_rel'
|
|
|
|
_sql_request_users_relation = 'users_sqlquery_rel'
|
|
|
|
_check_execution_enabled = False
|
|
|
|
copy_options = fields.Char(
|
|
string='Copy Options', required=False,
|
|
default="CSV HEADER DELIMITER ';'")
|
|
|
|
file_format = fields.Selection(
|
|
[('csv', 'CSV')],
|
|
default='csv',
|
|
required=True)
|
|
|
|
field_ids = fields.Many2many(
|
|
'ir.model.fields',
|
|
'fields_sqlquery_rel',
|
|
'sql_id',
|
|
'field_id',
|
|
'Parameters',
|
|
domain=[('model', '=', 'sql.file.wizard')])
|
|
|
|
encoding = fields.Selection(
|
|
[('utf-8', 'utf-8'), ('utf-16', 'utf-16'),
|
|
('windows-1252', 'windows-1252'), ('latin1', 'latin1'),
|
|
('latin2', 'latin2'), ('big5', 'big5'), ('gb18030', 'gb18030'),
|
|
('shift_jis', 'shift_jis'), ('windows-1251', 'windows-1251'),
|
|
('koir8_r', 'koir8_r')], string='Encoding', required=True,
|
|
default='utf-8')
|
|
|
|
def export_sql_query(self):
|
|
self.ensure_one()
|
|
wiz = self.env['sql.file.wizard'].create({
|
|
'sql_export_id': self.id})
|
|
return {
|
|
'view_mode': 'form',
|
|
'res_model': 'sql.file.wizard',
|
|
'res_id': wiz.id,
|
|
'type': 'ir.actions.act_window',
|
|
'target': 'new',
|
|
'context': self.env.context,
|
|
'nodestroy': True,
|
|
}
|
|
|
|
def _get_file_extension(self):
|
|
self.ensure_one()
|
|
if self.file_format == 'csv':
|
|
return 'csv'
|
|
|
|
def csv_get_data_from_query(self, variable_dict):
|
|
self.ensure_one()
|
|
# Execute Request
|
|
res = self._execute_sql_request(
|
|
params=variable_dict, mode='stdout',
|
|
copy_options=self.copy_options)
|
|
if self.encoding:
|
|
res = res.decode(self.encoding)
|
|
return res
|