from odoo import api, fields, models, tools
from psycopg2.extensions import AsIs


class MisCashFlow(models.Model):

    _inherit = 'mis.cash_flow'
    
    active = fields.Boolean(default=True)
    analytic_account_id = fields.Many2one(
        comodel_name='account.analytic.account',
        string='Analytic Account',
        auto_join=True,
        index=True,
        readonly=True,
    )
    

    @api.model_cr
    def init(self):
        account_type_receivable = self.env.ref(
            'account.data_account_type_receivable')
        query = """
            SELECT
                -- we use negative id to avoid duplicates and we don't use
                -- ROW_NUMBER() because the performance was very poor
                -aml.id as id,
                CAST('move_line' AS varchar) as line_type,
                aml.id as move_line_id,
                aml.account_id as account_id,
                aml.analytic_account_id as analytic_account_id,
                aml.debit as debit,
                aml.credit as credit,
                aml.reconciled as reconciled,
                aml.full_reconcile_id as full_reconcile_id,
                aml.partner_id as partner_id,
                aml.company_id as company_id,
                aml.user_type_id as user_type_id,
                aml.name as name,
                aml.date_maturity as date,
                CAST('true' AS BOOLEAN) as active
            FROM account_move_line as aml
            UNION ALL
            SELECT
                fl.id as id,
                CAST('forecast_line' AS varchar) as line_type,
                Null as move_line_id,
                fl.account_id as account_id,
                fl.analytic_account_id as analytic_account_id,   
                CASE
                    WHEN fl.balance > 0
                    THEN fl.balance
                    ELSE 0.0
                END AS debit,
                CASE
                    WHEN fl.balance < 0
                    THEN -fl.balance
                    ELSE 0.0
                END AS credit,
                Null as reconciled,
                Null as full_reconcile_id,
                fl.partner_id as partner_id,
                fl.company_id as company_id,
                %i as user_type_id,
                fl.name as name,
                fl.date as date,
                fl.active as active
            FROM mis_cash_flow_forecast_line as fl
        """ % account_type_receivable.id
        tools.drop_view_if_exists(self.env.cr, self._table)
        self._cr.execute(
            'CREATE OR REPLACE VIEW %s AS %s',
            (AsIs(self._table), AsIs(query))
        )

    @api.multi
    def action_open_related_line(self):
        self.ensure_one()
        if self.line_type == 'move_line':
            return self.move_line_id.get_formview_action()
        else:
            return self.env['mis.cash_flow.forecast_line'].browse(
                self.id).get_formview_action()