import os
import re

from gavo import api
from gavo.helpers import anet
from gavo.helpers import processing


class APOCalibrator(anet.AnetHeaderProcessor):
	sp_indices = ["/data/gavo/inputs/maidanak/customindex/custom.idx",
		"/data/gavo/inputs/maidanak/customindex/custom1.idx",
		"/data/gavo/inputs/maidanak/customindex/custom2.idx",]
	sp_endob = 15
	sexControl = ""

	def _isProcessed(self, srcName):
		return self.getPrimaryHeader(srcName).has_key("A_2_0")

	def classify(self, srcName):
		if self._isProcessed(srcName):
			return "solved"
		return "raw"

	def _getHeader(self, srcName):
		hdr = anet.AnetHeaderProcessor._getHeader(self, srcName)
		if hdr is not None:
			hdr.set("LOCATION", "Unknown")
		return hdr

if __name__=="__main__":
	processing.procmain(APOCalibrator, "apo/res/apo", "import")