Fixed problems with python 2 to 3 conversion

git-svn-id: http://svn.stsci.edu/svn/ssb/stsci_python/stwcs/trunk@40220 fe389314-cf27-0410-b35b-8c050e845b92

1 files changed, 3 insertions, 3 deletions

diff --git a/lib/stwcs/wcsutil/altwcs.py b/lib/stwcs/wcsutil/altwcs.py
index 15123b7..293f22d 100644
--- a/lib/stwcs/wcsutil/altwcs.py
+++ b/lib/stwcs/wcsutil/altwcs.py
@@ -80,7 +80,7 @@ def archiveWCS(fname, ext, wcskey=" ", wcsname=" ", reusekey=False):
         Alternatively choose another wcskey with altwcs.available_wcskeys()." %wcskey)
     elif wcskey == " ":
         # wcsname exists, overwrite it if reuse is True or get the next key
-        if wcsname.strip() in wcsnames(f[wcsext].header):
+        if wcsname.strip() in wcsnames(f[wcsext].header).values():
             if reusekey:
                 # try getting the key from an existing WCS with WCSNAME
                 wkey = getKeyFromName(f[wcsext].header, wcsname)
@@ -133,7 +133,7 @@ def archiveWCS(fname, ext, wcskey=" ", wcsname=" ", reusekey=False):
         except:
             pass
         
-        for k in hwcs:
+        for k in hwcs.keys():
             key = k[:7] + wkey
             f[e].header[key] = hwcs[k]
     closefobj(fname, f)
@@ -419,7 +419,7 @@ def _restore(fobj, ukey, fromextnum,
     if hwcs is None:
         return
 
-    for k in hwcs:
+    for k in hwcs.keys():
         key = k[:-1]
         if key in fobj[toextension].header:
             #fobj[toextension].header.update(key=key, value = hwcs[k])