from astropy.io import fits
import csv
import os
with open('CFHT_WITH_PSF','r') as csvfile:
    rows = csv.reader(csvfile, delimiter=',')
    i=0
    for row in rows:
      os.chdir(row[2])
      hdulist1 = fits.open('psf_A.fits')
      dat = hdulist1[0].data
      print sum(sum(dat))
      dat = dat / sum(sum(dat))
      hdu = fits.PrimaryHDU(data=dat)
      hdulist = fits.HDUList([hdu])
      hdulist.writeto('psf_A.fits', clobber=True)
      hdulist1.close()
      os.chdir('../../../')