diff options
Diffstat (limited to 'stwcs/gui/updatewcs.py')
| -rw-r--r-- | stwcs/gui/updatewcs.py | 38 |
1 files changed, 20 insertions, 18 deletions
diff --git a/stwcs/gui/updatewcs.py b/stwcs/gui/updatewcs.py index 3dacb67..b5dc700 100644 --- a/stwcs/gui/updatewcs.py +++ b/stwcs/gui/updatewcs.py @@ -1,19 +1,22 @@ -from __future__ import print_function +from __future__ import absolute_import, division, print_function + import os from astropy.io import fits from stsci.tools import parseinput from stsci.tools import fileutil from stsci.tools import teal -import stwcs -from stwcs import updatewcs -from stwcs.wcsutil import convertwcs +from .. import __version__ +from .. import updatewcs + + +allowed_corr_dict = {'vacorr': 'VACorr', 'tddcorr': 'TDDCorr', 'npolcorr': 'NPOLCorr', + 'd2imcorr': 'DET2IMCorr'} -allowed_corr_dict = {'vacorr':'VACorr','tddcorr':'TDDCorr','npolcorr':'NPOLCorr','d2imcorr':'DET2IMCorr'} __taskname__ = __name__.split('.')[-1] # needed for help string __package__ = updatewcs.__name__ -__version__ = stwcs.__version__ +#__version__ = stwcs.__version__ # #### Interfaces used by TEAL @@ -23,24 +26,24 @@ def getHelpAsString(docstring=False): return useful help from a file in the script directory called __taskname__.help """ install_dir = os.path.dirname(__file__) - htmlfile = os.path.join(install_dir,'htmlhelp',__taskname__+'.html') - helpfile = os.path.join(install_dir,__taskname__+'.help') + htmlfile = os.path.join(install_dir, 'htmlhelp', __taskname__ + '.html') + helpfile = os.path.join(install_dir, __taskname__ + '.help') if docstring or (not docstring and not os.path.exists(htmlfile)): - helpString = __taskname__+' Version '+__version__+'\n\n' + helpString = __taskname__ + ' Version ' + __version__ + '\n\n' if os.path.exists(helpfile): - helpString += teal.getHelpFileAsString(__taskname__,__file__) + helpString += teal.getHelpFileAsString(__taskname__, __file__) else: - helpString += eval('.'.join([__package__,__taskname__,'__doc__'])) + helpString += eval('.'.join([__package__, __taskname__, '__doc__'])) else: - helpString = 'file://'+htmlfile + helpString = 'file://' + htmlfile return helpString + def run(configObj=None): # Interpret primary parameters from configObj instance - extname = configObj['extname'] input = configObj['input'] # create dictionary of remaining parameters, deleting extraneous ones @@ -53,7 +56,7 @@ def run(configObj=None): del cdict['extname'] # parse input - input,altfiles = parseinput.parseinput(configObj['input']) + input, altfiles = parseinput.parseinput(configObj['input']) # Insure that all input files have a correctly archived # set of OPUS WCS keywords @@ -69,7 +72,7 @@ def run(configObj=None): # Check to insure that there is a valid reference file to be used idctab = fits.getval(file, 'idctab') if not os.path.exists(fileutil.osfn(idctab)): - print('No valid distortion reference file ',idctab,' found in ',file,'!') + print('No valid distortion reference file ', idctab, ' found in ', file, '!') raise ValueError # Re-define 'cdict' to only have switches for steps supported by that instrument @@ -78,7 +81,7 @@ def run(configObj=None): # for file in input: # get instrument name from input file - instr = fits.getval(file,'INSTRUME') + instr = fits.getval(file, 'INSTRUME') # make copy of input parameters dict for this file fdict = cdict.copy() # Remove any parameter that is not part of this instrument's allowed corrections @@ -86,5 +89,4 @@ def run(configObj=None): if allowed_corr_dict[step] not in updatewcs.apply_corrections.allowed_corrections[instr]: fdict[step] # Call 'updatewcs' on correctly archived file - updatewcs.updatewcs(file,**fdict) - + updatewcs.updatewcs(file, **fdict) |