Merge b27cee0 into d9d7672

weaverba137 · Jan 8, 2016 · 407dd89 · 407dd89
2 parents d9d7672 + b27cee0
commit 407dd89
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/pydl/pydlspec2d/spec1d/template_input.py b/pydl/pydlspec2d/spec1d/template_input.py
@@ -496,20 +496,23 @@ def template_input_main():  # pragma: no cover
     import os
     import sys
     from astropy.utils.compat import argparse
+
+    # Get home directory in platform-independent way
+    home_dir = os.path.expanduser('~')
     #
     # Get Options
     #
     parser = argparse.ArgumentParser(description="Compute spectral templates.",
                                      prog=os.path.basename(sys.argv[0]))
     parser.add_argument('-d', '--dump', action='store', dest='dump',
                         metavar='FILE',
-                        default=os.path.join(os.getenv('HOME'), 'scratch', 'templates', 'compute_templates.dump'),
+                        default=os.path.join(home_dir, 'scratch', 'templates', 'compute_templates.dump'),
                         help='Dump data to a pickle file.')
     parser.add_argument('-F', '--flux', action='store_true', dest='flux',
                         help='Plot input spectra.')
     parser.add_argument('-f', '--file', action='store', dest='inputfile',
                         metavar='FILE',
-                        default=os.path.join(os.getenv('HOME'), 'scratch', 'templates', 'compute_templates.par'),
+                        default=os.path.join(home_dir, 'scratch', 'templates', 'compute_templates.par'),
                         help='Read input spectra and redshifts from FILE.')
     parser.add_argument('-v', '--verbose', action='store_true', dest='verbose',
                         help='Print lots of extra information.')