diff options
| author | dencheva <dencheva@stsci.edu> | 2010-10-06 16:09:54 -0400 |
|---|---|---|
| committer | dencheva <dencheva@stsci.edu> | 2010-10-06 16:09:54 -0400 |
| commit | 5aff646c4ffbb860ef9e95e501b1725dd880abbd (patch) | |
| tree | 7c0223b6f9e93c2a6741a6a003a65004c6a46edf /wcsutil/altwcs.py | |
| parent | 02ba60cfb8db0288977724ebe73a13ee68544c31 (diff) | |
| download | stwcs_hcf-5aff646c4ffbb860ef9e95e501b1725dd880abbd.tar.gz | |
Corrected a problem with ext parameter
git-svn-id: http://svn.stsci.edu/svn/ssb/stsci_python/stsci_python/trunk/stwcs@10478 fe389314-cf27-0410-b35b-8c050e845b92
Diffstat (limited to 'wcsutil/altwcs.py')
| -rw-r--r-- | wcsutil/altwcs.py | 13 |
1 files changed, 6 insertions, 7 deletions
diff --git a/wcsutil/altwcs.py b/wcsutil/altwcs.py index 42a2d67..76bfc76 100644 --- a/wcsutil/altwcs.py +++ b/wcsutil/altwcs.py @@ -53,7 +53,7 @@ def archiveWCS(fname, ext, wcskey=" ", wcsname=" ", clobber=False): return if isinstance(ext, int) or isinstance(ext, tuple): - exts = [ext] + ext = [ext] if wcskey == " ": # try getting the key from WCSNAME @@ -74,7 +74,7 @@ def archiveWCS(fname, ext, wcskey=" ", wcsname=" ", clobber=False): wkey = wcskey wname = wcsname - for e in exts: + for e in ext: w = pywcs.WCS(f[e].header, fobj=f) hwcs = w.to_header() wcsnamekey = 'WCSNAME' + wkey @@ -134,7 +134,7 @@ def restoreWCS(f, ext, wcskey=" ", wcsname=" ", clobber=False): return if isinstance(ext, int) or isinstance(ext, tuple): - exts = [ext] + ext = [ext] if not clobber: name = (fobj.filename().split('.fits')[0] + '_%s_' + '.fits') %wcskey @@ -155,13 +155,12 @@ def restoreWCS(f, ext, wcskey=" ", wcsname=" ", clobber=False): return wkey = wcskey - for e in exts: + for e in ext: try: extname = fobj[e].header['EXTNAME'].lower() except KeyError: continue #Restore always from a 'SCI' extension but write it out to 'ERR' and 'DQ' - print 'extname', extname if extname == 'sci': sciver = fobj[e].header['extver'] try: @@ -227,7 +226,7 @@ def deleteWCS(fname, ext, wcskey=" ", wcsname=" "): return if isinstance(ext, int) or isinstance(ext, tuple): - exts = [ext] + ext = [ext] # Do not allow deleting the original WCS. if wcskey == 'O': @@ -255,7 +254,7 @@ def deleteWCS(fname, ext, wcskey=" ", wcsname=" "): wkey = wcskey prexts = [] - for i in exts: + for i in ext: hdr = fobj[i].header try: w = pywcs.WCS(hdr, fobj, key=wkey) |