#!/usr/bin/env python
import os
import csv
from all_plots import *
import argparse

parser = argparse.ArgumentParser()
parser.add_argument("-p", "--path", type=str)
args = parser.parse_args()


with open('list_fits_files_sdss.csv','r') as csvfile:
    rows = csv.reader(csvfile, delimiter=',')
    i=0
    for row in rows:
      if args.path:
	print args.path
	print "PARAMS"
	if args.path in row[2]:
	  print row
	  os.chdir(row[2])
	  if os.path.exists('output_Ba_psf.fits'):
	    print_plots('output_Ba_psf.fits')
	  os.chdir('../../../')
      else:
	  print "NO PARAMS"
	  os.chdir(row[2])
	  if os.path.exists('output_Ba_psf.fits'):
	    print_plots('output_Ba_psf.fits')
	  os.chdir('../../../')