header_info.py

#header_info.py: Script to print relevant information about the image files.
#Marjorie Decleir
#Updated (to Python 3.6) on 24-10-2018

#Import the necessary packages.
import os
from astropy.io import fits
import configloader


#Load the configuration file.
config = configloader.load_config()

#Specify the galaxy and the path to the raw images.
galaxy = config['galaxy']
path = config['path']  + galaxy + "/Raw_images/"

#Print titles of columns.
print("filename"+ "\t" + "\t" + "\t" + "#frames" + "\t" + "filter"+ "\t" + "date" + "\n")

#For all files in the path directory:
for filename in sorted(os.listdir(path)):

    #If the file is not a raw image file, skip this file and continue with the next file.
    if not filename.endswith("rw.img"): continue

    #Open the image, calculate the number of individual frames in the image. Print relevant header information.
    hdulist = fits.open(path+filename)
    number_of_frames = len(hdulist)-1
    print(filename + "\t"+ "\t" + str(number_of_frames) + "\t" + hdulist[0].header["FILTER"] + "\t" + hdulist[0].header["DATE-OBS"].split('T')[0])

    #For all frames in the image: print the datamode and calculate the total exposure time.
    print("\t" + "frame" + "\t" + "datamode")
    exptime = 0.
    for i in range(1, len(hdulist)):
        exptime += hdulist[i].header["EXPOSURE"]
        print("\t" + str(i) + "\t" + hdulist[i].header["DATAMODE"] + "\t" + "\t" + str(hdulist[i].header["NAXIS1"]) +  "\t" + str(hdulist[i].header["NAXIS2"]))
    print("Total exposure time: " + str(exptime))
    print("\n")