import os

import numpy

from gavo import api
from gavo.utils import fitstools, imgtools

class PreviewMaker(api.PreviewMaker):
	def getPreviewData(self, srcName):
		with open(api.getConfig("inputsDir") / srcName, "rb") as inFile:
			pixels = numpy.array([row
				for row in fitstools.iterScaledRows(inFile,
					destSize=200)])
		return imgtools.jpegFromNumpyArray(pixels)
		

if __name__=="__main__":
  api.procmain(PreviewMaker, "bgds/q", "import")