wipe_wcs.py

#!/usr/bin/env python
# -*- coding: UTF-8 -*-

import argparse
import repipy.utilities as utilities
import astropy.io.fits as fits
import sys
import repipy.astroim as astroim

def wipe(args):
    for im_name in args.input:
        im = astroim.Astroim(im_name)
        hdulist = im.HDUList
        if im.HDUList_mask:
            hdulist += im.HDUList_mask

        for hdu in  hdulist:
            hdu.header = utilities.remove_WCS(hdu.header)

        if im.HDUList_mask:
            im.HDUList_mask.writeto(im.mask_name, clobber=True)
        im.write()

# Create parser
parser = argparse.ArgumentParser(description='(Almost) Completely wipe the World Coordinate Sytem from the header of a '
                                             'list of images')
parser.add_argument("input", metavar='input', action='store', help='list of images from which to remove the WCS.',
                    nargs="+", type=str)


def main(arguments = None):
  # Pass arguments to variable args
  if arguments == None:
      arguments = sys.argv[1:]

  args = parser.parse_args(arguments)
  wipe(args)

if __name__ == "__main__":
    main()