# -*- coding: utf-8 -*-

from collections import defaultdict
from dateutil.relativedelta import relativedelta

from odoo import models, _


class AccountMove(models.Model):
    _inherit = 'account.move'

    def _post(self, soft=True):
        posted_moves = super()._post(soft=soft)
        automatic_invoice = self.env.context.get('recurring_automatic')
        all_subscription_ids = []
        post_hook_subscription_ids = []
        for move in posted_moves:
            if not move.invoice_line_ids.subscription_id:
                continue
            if move.move_type != 'out_invoice':
                if move.move_type == 'out_refund':
                    body = _("The following refund %s has been made on this contract. Please check the next invoice date if necessary.", move._get_html_link())
                    for so in move.invoice_line_ids.subscription_id:
                        # Normally, only one subscription_id per move, but we handle multiple contracts as a precaution
                        so.message_post(body=body)
                continue
            aml_by_subscription = defaultdict(lambda: self.env['account.move.line'])
            for aml in move.invoice_line_ids:
                if not aml.subscription_id:
                    continue
                aml_by_subscription[aml.subscription_id] |= aml
            for subscription, aml in aml_by_subscription.items():
                sale_order = aml.sale_line_ids.order_id
                if subscription != sale_order:
                    # we are invoicing an upsell
                    continue
                # Normally, only one period_end should exist
                end_dates = [ed for ed in aml.mapped('deferred_end_date') if ed]
                if end_dates and max(end_dates) > subscription.next_invoice_date:
                    subscription.next_invoice_date = max(end_dates) + relativedelta(days=1)
                    all_subscription_ids.append(subscription.id)
                if not automatic_invoice:
                    post_hook_subscription_ids.append(subscription.id)
                subscription.pending_transaction = False
            if all_subscription_ids:
                # update the renewal quotes to start at the next invoice date values
                renewal_quotes = self.env['sale.order'].search([
                    ('subscription_id', 'in', all_subscription_ids),
                    ('subscription_state', '=', '2_renewal'),
                    ('state', 'in', ['draft', 'sent'])
                ])
                for quote in renewal_quotes:
                    next_invoice_date = quote.subscription_id.next_invoice_date
                    if not quote.start_date or quote.start_date < next_invoice_date:
                        quote.update({
                            'next_invoice_date': next_invoice_date,
                            'start_date': next_invoice_date,
                        })
            if post_hook_subscription_ids:
                self.env['sale.order'].browse(post_hook_subscription_ids)._post_invoice_hook()

        return posted_moves

    def _message_auto_subscribe_followers(self, updated_values, subtype_ids):
        """ Assignment emails are sent when an account.move is created with a
        different user_id than the current one. For subscriptions it can send
        thousands of email depending on the database size.
        This override prevent the assignment emails to be sent to the
        salesperson.
        """
        res = super()._message_auto_subscribe_followers(updated_values, subtype_ids)
        user_id = updated_values.get('user_id')
        if user_id and self.invoice_line_ids.subscription_id:
            for move in self:
                salesperson = move.invoice_line_ids.subscription_id.user_id
                if salesperson and user_id in salesperson.ids and user_id != self.env.user.id:
                    partner_ids = salesperson.partner_id.ids
                    res = [(v[0], v[1], False) for v in res if v[0] in partner_ids and v[2] == 'mail.message_user_assigned']
        return res