# HG changeset patch # User greg # Date 1449676870 18000 # Node ID cda30ae2e9d74819112d7303e956247cbc49712e # Parent cb3e79d1e85e0f6826e76838fc075d7edc202a13 Uploaded diff -r cb3e79d1e85e -r cda30ae2e9d7 icqsol_utils.py --- a/icqsol_utils.py Sat Dec 05 10:28:42 2015 -0500 +++ b/icqsol_utils.py Wed Dec 09 11:01:10 2015 -0500 @@ -2,7 +2,9 @@ import sys import tempfile +PLY = 'ply' POLYDATA = 'POLYDATA' +VTK = 'vtk' def asbool(val): @@ -14,9 +16,9 @@ format = None datatype = None if galaxy_ext in ['vtkascii', 'vtkbinary']: - format = 'vtk' + format = VTK elif galaxy_ext in ['plyascii', 'plybinary']: - format = 'ply' + format = PLY if galaxy_ext in ['vtkascii', 'plyascii']: datatype = 'ascii' elif galaxy_ext in ['vtkbinary', 'plybinary']: @@ -48,7 +50,13 @@ """ Return a temporary file name. """ - fd, name = tempfile.mkstemp(suffix=suffix, dir=dir) + if suffix is None: + s = None + elif suffix.startswith('.'): + s = suffix + else: + s = '.%s' % suffix + fd, name = tempfile.mkstemp(suffix=s, dir=dir) os.close(fd) return name