from trytond.pool import Pool
from . import (address, diary, party, product, purchase, sale,
               equipment, configuration_equipment, contract, maintenance, move, exceptions)

def register():
    Pool.register(
        address.Address,
        diary.Diary,
        party.Party,
        product.Template,
        product.Product,
        product.Pattern,
        purchase.Purchase,
        purchase.Line,
        equipment.OpticalEquipment,
        equipment.EquipmentMaintenance,
        equipment.EquipmentParty,
        equipment.ChangePropietary,
        configuration_equipment.Configuration,
        maintenance.Calibration,
        maintenance.CreateContractInitial,
        maintenance.CalibrationSample,
        maintenance.Maintenance,
        maintenance.MaintenanceService,
        maintenance.MaintenanceActivity,
        maintenance.MaintenanceLine,
        maintenance.AgendedInitial,
        maintenance.ReAgended,
        maintenance.ServiceMaintenanceAgended,
        move.Move,
        move.ShipmentOut,
        move.ShipmentInternal,
        sale.Sale,
        sale.SaleLine,
        contract.Contract,
        contract.ContractEquipment,
        contract.ContractMaintenanceServices,
        module='optical_equipment', type_='model')
    Pool.register(
        maintenance.AssingAgended,
        maintenance.ReAssingAgended,
        maintenance.CreateContract,
        equipment.NewPropietary,
        module='optical_equipment', type_='wizard')