"""
remove all header cards but the minimum necessary (what came from LSW).
"""

import sys

import pyfits

from gavo import utils
from gavo.utils import fitstools

def process(name):
	cardDict = dict((card.key, card)
		for card in utils.readPrimaryHeaderQuick(open(name)).cards)
	newCards = []
	for key in ["SIMPLE", "BITPIX", "NAXIS", "NAXIS1", "NAXIS2", "EXTEND",
			"BZERO", "BSCALE"]:
		newCards.append(cardDict[key])
	newHdr = pyfits.Header(newCards)
	fitstools.replacePrimaryHeaderInPlace(name, newHdr)


if __name__=="__main__":
	for name in sys.argv[1:]:
		process(name)