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

def register():
    Pool.register(
        address.Address,
        diary.Diary,
        party.Party,
        product.Template,
        product.Product,
        purchase.Purchase,
        purchase.Line,
        equipment.OpticalEquipment,
        equipment.EquipmentMaintenance,
        configuration_equipment.Configuration,
        maintenance.Maintenance,
        maintenance.MaintenanceActivity,
        maintenance.MaintenanceLine,
        move.Move,
        move.ShipmentOut,
        subscription.Contract,
        subscription.Subscription,
        subscription.CreateInitialContract,
        subscription.CreateNextProrogue,
        subscription.ContractSupscription,
        subscription.ContractEquipment,
        subscription.SubscriptionEquipment,
        module='optical_equipment', type_='model')
    Pool.register(
        subscription.CreateContract,
        subscription.CreateProrogue,
        subscription.CreateSubscriptionInvoice,        
        module='optical_equipment', type_='wizard')