"""
Upstream data is missing the CUNIT1 header.
"""

from gavo import api
from gavo.utils import pyfits


class HeaderFixer(api.FileProcessor):
# We need to manipulate multiple extensions, so the normal header processor
# doesn't really help
	def process(self, srcName):
		hdus = pyfits.open(srcName, mode='update')
		hdus[0].header.set("CUNIT1", "Angstrom")
		hdus[0].header.set("BUNIT", "erg/cm**2/s/Angstrom")
		hdus[0].header.set("CTYPE1", "WAVELENGTH")
		hdus[1].header.set("CUNIT1", "Angstrom")
		hdus[1].header.set("BUNIT", "erg/cm**2/s/Angstrom")
		hdus[1].header.set("CTYPE1", "WAVELENGTH")
		hdus.flush()



if __name__=="__main__":
	api.procmain(HeaderFixer, "pcslg/q", "import")