|
|
edi_formats = self.env['account.edi.format'].search([])
journal_ids = self.ids
if journal_ids:
self.env.cr.execute('''
SELECT
move.journal_id,
ARRAY_AGG(doc.edi_format_id) AS edi_format_ids
FROM account_edi_document doc
JOIN account_move move ON move.id = doc.move_id
WHERE doc.state IN ('to_cancel', 'to_send')
AND move.journal_id IN %s
GROUP BY move.journal_id
''', [tuple(journal_ids)])
protected_edi_formats_per_journal = {r[0]: set(r[1]) for r in self.env.cr.fetchall()}
else:
protected_edi_formats_per_journal = defaultdict(set)
for journal in self:
enabled_edi_formats = edi_formats.filtered(lambda e: e._is_compatible_with_journal(journal) and
(e._is_enabled_by_default_on_journal(journal)
or (e in journal.edi_format_ids)))
# The existing edi formats that are already in use so we can't remove it.
protected_edi_format_ids = protected_edi_formats_per_journal.get(journal.id, set())
protected_edi_formats = journal.edi_format_ids.filtered(lambda e: e.id in protected_edi_format_ids)
journal.edi_format_ids = enabled_edi_formats + protected_edi_formats
|
|