from rest_framework import viewsets
from imagedb.serializers import InstrumentSerializer, NispDetectorSerializer, NispRawFrameSerializer
from imagedb.models import Instrument, NispDetector, NispRawFrame
from url_filter.filtersets import ModelFilterSet


class InstrumentViewSet(viewsets.ReadOnlyModelViewSet):
  queryset = Instrument.objects.all()
  serializer_class = InstrumentSerializer
  
class NispDetectorViewSet(viewsets.ReadOnlyModelViewSet):
  queryset = NispDetector.objects.all()
  serializer_class = NispDetectorSerializer



class NispRawFrameFilterSet(ModelFilterSet):
  class Meta:
    model = NispRawFrame
    fields = [f.name for f in NispRawFrame._meta.get_fields() 
              if hasattr(f, 'serialize') and f.serialize]
    
  
class NispRawFrameViewSet(viewsets.ReadOnlyModelViewSet):
  queryset = NispRawFrame.objects.all()
  serializer_class = NispRawFrameSerializer
  filter_class = NispRawFrameFilterSet