pms/hotel/models/inherited_account_payment.py

# Copyright 2017  Dario Lodeiros
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
from odoo.exceptions import except_orm
from odoo import models, fields, api, _

class AccountPayment(models.Model):
    _inherit = 'account.payment'

    folio_id = fields.Many2one('hotel.folio', string='Folio')
    amount_total_folio = fields.Float(
        compute="_compute_folio_amount", store=True,
        string="Total amount in folio",
    )
    save_amount = fields.Monetary(string='onchange_amount')
    save_date = fields.Date()
    save_journal_id = fields.Integer()

    @api.onchange('amount','payment_date','journal_id')
    def onchange_amount(self):
        if self._origin:
            self.save_amount = self._origin.amount
            self.save_journal_id = self._origin.journal_id.id
            self.save_date = self._origin.payment_date

    """WIP"""
    @api.multi
    def return_payment_folio(self):
        journal = self.journal_id
        partner = self.partner_id
        amount = self.amount
        reference = self.communication
        account_move_lines = self.move_line_ids.filtered(lambda x: (
            x.account_id.internal_type == 'receivable'))
        return_line_vals = {
            'move_line_ids': [(6, False, [x.id for x in account_move_lines])],
            'partner_id': partner.id,
            'amount': amount,
            'reference': reference,
            }
        return_vals = {
            'journal_id': journal.id,
            'line_ids': [(0, 0, return_line_vals)],
            }
        return_pay = self.env['payment.return'].create(return_vals)
        if self.save_amount:
            self.amount = self.save_amount
        if self.save_date:
            self.payment_date = self.save_date
        if self.save_journal_id:
            self.journal_id = self.env['account.journal'].browse(self.save_journal_id)
        return_pay.action_confirm()

    @api.multi
    def modify(self):
        self.cancel()
        vals = {
            'journal_id': self.journal_id,
            'partner_id': self.partner_id,
            'amount': self.amount,
            'payment_date': self.payment_date,
            'communication': self.communication,
            'state': 'draft'}
        self.update(vals)
        self.with_context({'ignore_notification_post': True}).post()
        if self.folio_id:
            msg = _("Payment %s modified: \n") % (self.communication)
            if self.save_amount and self.save_amount != self.amount:
                msg += _("Amount from %s to %s %s \n") % (self.save_amount, self.amount, self.currency_id.symbol)
            if self.save_date and self.save_date != self.payment_date:
                msg += _("Date from %s to %s \n") % (self.save_date, self.payment_date)
            if self.save_journal_id and self.save_journal_id != self.journal_id.id:
                msg += _("Journal from %s to %s") % (self.env['account.journal'].browse(self.save_journal_id).name, self.journal_id.name)
            self.folio_id.message_post(subject=_('Payment'), body=msg)

    @api.multi
    def delete(self):
        msg = False
        if self.folio_id:
            msg = _("Deleted payment: %s %s ") % (self.amount, self.currency_id.symbol)
        self.cancel()
        self.move_name = ''
        self.unlink()
        if msg:
            self.folio_id.message_post(subject=_('Payment Deleted'), body=msg)

    @api.multi
    @api.depends('state')
    def _compute_folio_amount(self):
        # FIXME: Finalize method
        res = []
        fol = ()
        for payment in self:
            if payment.folio_id:
                fol = payment.env['hotel.folio'].search([
                    ('id', '=', payment.folio_id.id)
                ])
            else:
                return
            if not any(fol):
                return
            if len(fol) > 1:
                raise except_orm(_('Warning'), _('This pay is related with \
                                                more than one Reservation.'))
            else:
                fol.compute_amount()
            return res

    @api.multi
    def post(self):
        rec = super(AccountPayment, self).post()
        if rec and not self._context.get("ignore_notification_post", False):
            for pay in self:
                if pay.folio_id:
                    msg = _("Payment of %s %s registered from %s using %s payment method") % (pay.amount, pay.currency_id.symbol, pay.communication, pay.journal_id.name)
                    pay.folio_id.message_post(subject=_('Payment'), body=msg)

    @api.multi
    def modify_payment(self):
        self.ensure_one()
        view_form_id = self.env.ref('hotel.account_payment_view_form_folio').id
        # invoices = self.mapped('invoice_ids.id')
        return{
            'name': _('Payment'),
            'view_type': 'form',
            'views': [(view_form_id, 'form')],
            'view_mode': 'tree,form',
            'res_model': 'account.payment',
            'target': 'new',
            'init_mode': 'edit',
            'type': 'ir.actions.act_window',
            'res_id': self.id,
        }