Skip to content

Commit

Permalink
Py3 fix for make-minimal-datafiles
Browse files Browse the repository at this point in the history
  • Loading branch information
mperrin committed Nov 21, 2015
1 parent ccc5806 commit a9854d3
Showing 1 changed file with 8 additions and 8 deletions.
16 changes: 8 additions & 8 deletions dev_utils/make-minimal-datafiles.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,35 +20,35 @@
subprocess.call("mkdir "+WORKING_DIR, shell=True)


print "#### Expanding full tar file into temp directory ####"
print("#### Expanding full tar file into temp directory ####")
os.chdir(WORKING_DIR)
subprocess.call("tar xvzf "+inputfile, shell=True)

print "#### Trimming to only one OPD file per instrument ####"
print("#### Trimming to only one OPD file per instrument ####")
for instr in insts:
files = glob.glob(os.path.join(WORKING_DIR, 'webbpsf-data', instr, "OPD", "*.fits"))
files.sort()
print instr, files
print(instr, files)

# just save the lowest alphabetically of each of them
for file_to_delete in files[1:]:
print "Deleting "+file_to_delete
print("Deleting "+file_to_delete)
os.remove(file_to_delete)

print "Trimming to only 1 datacube slice: "+files[0]
print("Trimming to only 1 datacube slice: "+files[0])

f0 = fits.open(files[0], mode='update')
f0[0].data = f0[0].data[0]
f0.flush()
f0.close()

print "#### Removing extra optional pupil files ####"
print("#### Removing extra optional pupil files ####")
os.remove(os.path.join(WORKING_DIR, 'webbpsf-data','pupil_RevT.fits'))
os.remove(os.path.join(WORKING_DIR, 'webbpsf-data','tricontagon.fits'))

print "#### Creating tar file ####"
print("#### Creating tar file ####")
os.chdir(WORKING_DIR)
subprocess.call('tar cvzf minimal-webbpsf-data.tar.gz webbpsf-data', shell=True)
print "===> {0}/minimal-webbpsf-data.tar.gz ".format(WORKING_DIR)
print("===> {0}/minimal-webbpsf-data.tar.gz ".format(WORKING_DIR))


0 comments on commit a9854d3

Please sign in to comment.