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

def register():
    Pool.register(
        address.Address,
        party.Party,
        product.Template,
        product.Product,
        purchase.Purchase,
        purchase.Line,
        sale.SaleLine,
        sale.CreateSubscriptionStart,
        equipment.OpticalEquipment,
        configuration_equipment.Configuration,
        maintenance.Maintenance,
        subscription.Subscription,
        subscription.SubscriptionEquipment,
        module='optical_equipment', type_='model')
    Pool.register(
        sale.CreateSubscription,
        module='optical_equipment', type_='wizard')