from netbox.search import SearchIndex, register_search

from .models import Service, ServiceTemplate


@register_search
class ServiceTemplateSearchIndex(SearchIndex):
    model = ServiceTemplate
    fields = (
        ("name", 100),
        ("description", 500),
    )
    display_attrs = ("status",)


@register_search
class ServiceSearchIndex(SearchIndex):
    model = Service
    fields = (
        ("name", 100),
        ("description", 500),
    )
    display_attrs = ("service_template", "status", "tenant")