# 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 fields, models


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