import os
import csv
from astropy.io import fits


def make_psf(file_name):
  hdulist = fits.open(file_name)
  x = 1
  y = 1
  q = hdulist[1].data
  a = q['PSF_MASK']
  psf_0 = a[0][0]

  psf_0 = psf_0/sum(map(sum, psf_0))
  hdu2 = fits.PrimaryHDU(data=psf_0)
  hdulist2 = fits.HDUList([hdu2])
  hdulist2.writeto('psf.fits', clobber=True)






f_psf_data = open('psf_data_list.txt', 'w')
with open('list_fits_files_CM.csv','r') as csvfile:
    rows = csv.reader(csvfile, delimiter=',')
    i=0
    for row in rows:
      os.chdir(row[2])
      try:
        print row
        make_psf('res_catalog_A.psf')
        hdulist = fits.open('res_catalog_A.psf')
        hdr = hdulist[1].header
        fwhm=hdr["PSF_FWHM"]
        f_psf_data.write(row[2]+','+str(fwhm)+ '\n')
      except IOError:
          print 'no file'
      os.chdir('../../../')
    f_psf_data.close()