#This file is part of Tryton.  The COPYRIGHT file at the top level of
#this repository contains the full copyright notices and license terms.
from trytond.model import fields
from trytond.pool import Pool, PoolMeta
from trytond.pyson import Bool, Eval, Not, Id

MODULE_NAME = "farm_prescription"


class Specie(metaclass=PoolMeta):
    __name__ = 'farm.specie'

    prescription_enabled = fields.Boolean('Prescriptions Enabled',
        help="This specie uses prescriptions.")
    prescription_sequence = fields.Many2One('ir.sequence.strict',
        'Prescription Reference Sequence', domain=[
            ('sequence_type', '=', Id('farm_prescription',
                    'sequence_type_prescription')),
        ], states={
            'readonly': Not(Bool(Eval('prescription_enabled'))),
            'required': Bool(Eval('prescription_enabled')),
        }, help='Sequence used for prescriptions.')

    @staticmethod
    def default_prescription_enabled():
        return True

    def _create_additional_menus(self, specie_menu, specie_submenu_seq,
            current_menus, current_actions, current_wizards):
        pool = Pool()
        Menu = pool.get('ir.ui.menu')
        ModelData = pool.get('ir.model.data')

        specie_submenu_seq = super(Specie,
            self)._create_additional_menus(specie_menu, specie_submenu_seq,
            current_menus, current_actions, current_wizards)

        if not self.prescription_enabled:
            return specie_submenu_seq

        new_domain = [
            ('specie', '=', self.id),
            ]
        new_context = {
            'specie': self.id,
            }
        for suffix in ['feed', 'medical']:
            prescriptions_menu = Menu(ModelData.get_id('farm_prescription',
                    'menu_farm_%s_prescriptions' % suffix))
            prescription_templates_menu = Menu(
                ModelData.get_id('farm_prescription',
                    'menu_farm_%s_prescription_templates' % suffix))
            menu = self._duplicate_menu(prescriptions_menu, specie_menu,
                specie_submenu_seq, current_menus, current_actions,
                current_wizards, new_domain=new_domain,
                new_context=new_context)

            self._duplicate_menu(prescription_templates_menu, menu, 1,
                current_menus, current_actions, current_wizards,
                new_domain=new_domain, new_context=new_context)
            specie_submenu_seq += 1
        return specie_submenu_seq