"""
Adds links to wedges and previews to existing files.

[TODO: add this to bin/fixIncoming so there's no separate step necessary
when  re-running that]
"""

import os

from gavo import api

class LinkAdder(api.ImmediateHeaderProcessor):
	def _createAuxiliaries(self, dd):
		self.staticBase = dd.rd.getById("dl").getURL("static")

	def _isProcessed(self, srcName, hdr):
		return hdr.get("FN-PRE", "").startswith("http")
	
	def _changeHeader(self, srcName, hdr):
		baseName = os.path.splitext(os.path.basename(srcName))[0]
		hdr.set("FN-WEDGE",
			str("%s/wedges/%sw.fits"%(self.staticBase, baseName)),
			after="FILENAME")
		hdr.set("FN-PRE",
			str("%s/jpegs/%s.jpg"%(self.staticBase, baseName)),
			after="FN-WEDGE")


if __name__=="__main__":
    res = api.procmain(LinkAdder, "kapteyn/q", "import")