# The COPYRIGHT file at the top level of
# this repository contains the full copyright notices and license terms.
from trytond.pool import PoolMeta
from trytond.model import fields

__all__ = ['Party', 'PartyIdentifier', 'Address']


class Party(metaclass=PoolMeta):
    __name__ = 'party.party'

    edi_template_path = fields.Char('Template Path')
    edi_add_header = fields.Boolean('Add Header')
    edi_release_number = fields.Selection([
        (None, ' '),
        ('96A', '96A'),
        ('01B', '01B'),
        ], 'Release Number')
    edi_assigned_code = fields.Selection([
        (None, ' '),
        ('EAN001', 'EAN001'),
        ('EAN002', 'EAN002'),
        ('EAN003', 'EAN003'),
        ('EAN004', 'EAN004'),
        ('EAN005', 'EAN005'),
        ('EAN006', 'EAN006'),
        ('EAN007', 'EAN007'),
        ('EAN008', 'EAN008'),
        ('EAN009', 'EAN009'),
        ('EAN010', 'EAN010'),
        ], 'Assigned Code')


class PartyIdentifier(metaclass=PoolMeta):
    __name__ = 'party.identifier'

    @classmethod
    def __setup__(cls):
        super(PartyIdentifier, cls).__setup__()
        cls.type.selection.extend(
            [('EDI_sender', 'EDI Sender'),
            ('EDI_receiver', 'EDI Receiver'),
            ('EDI_supplier', 'EDI Supplier')])


class Address(metaclass=PoolMeta):
    __name__ = 'party.address'

    EDI_code = fields.Char('EDI Code')