from ishtar_common.rest import SearchAPIView, FacetAPIView
from archaeological_operations import models, forms


class SearchOperationAPI(SearchAPIView):
    model = models.Operation


class FacetOperationAPIView(FacetAPIView):
    models = [models.Operation, models.ArchaeologicalSite]
    select_forms = [forms.OperationSelect, forms.SiteSelect]