import os
import csv
from make_assoc import *
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]


  hdu2 = fits.PrimaryHDU(data=psf_0)
  hdulist2 = fits.HDUList([hdu2])
  hdulist2.writeto('psf_A.fits', clobber=True)






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