windows test "fix"

stl/main.py

@@ -1,26 +1,35 @@
-import sys
-import random
 import argparse
+import pprint
+import random
+import sys
 
 from . import stl
 
 
 def _get_parser(description):
  parser = argparse.ArgumentParser(description=description)
- parser.add_argument('infile', nargs='?', type=argparse.FileType('rb'),
- default=sys.stdin, help='STL file to read')
- parser.add_argument('outfile', nargs='?', type=argparse.FileType('wb'),
- default=sys.stdout, help='STL file to write')
+ parser.add_argument(
+ 'infile', nargs='?', type=argparse.FileType('rb'),
+ default=sys.stdin, help='STL file to read'
+ )
+ parser.add_argument(
+ 'outfile', nargs='?', type=argparse.FileType('wb'),
+ default=sys.stdout, help='STL file to write'
+ )
  parser.add_argument('--name', nargs='?', help='Name of the mesh')
  parser.add_argument(
  '-n', '--use-file-normals', action='store_true',
- help='Read the normals from the file instead of recalculating them')
+ help='Read the normals from the file instead of recalculating them'
+ )
  parser.add_argument(
  '-r', '--remove-empty-areas', action='store_true',
  help='Remove areas with 0 surface areas to prevent errors during '
- 'normal calculation')
- parser.add_argument('-s', '--disable-speedups', action='store_true',
- help='Disable Cython speedups')
+ 'normal calculation'
+ )
+ parser.add_argument(
+ '-s', '--disable-speedups', action='store_true',
+ help='Disable Cython speedups'
+ )
  return parser
 
 
@@ -31,28 +40,39 @@ def _get_name(args):
  getattr(args.infile, 'name', None),
  'numpy-stl-%06d' % random.randint(0, 1_000_000),
  ]
- print('names', names)
- print('args', args)
 
  for name in names: # pragma: no branch
- if name and isinstance(name, str) and not name.startswith('<'):
+ if not isinstance(name, str):
+ continue
+ elif name.startswith('<'):
+ continue
+ elif r'\AppData\Local\Temp' in name:
+ # Windows temp file
+ continue
+ else:
  return name
 
 
 def main():
  parser = _get_parser('Convert STL files from ascii to binary and back')
- parser.add_argument('-a', '--ascii', action='store_true',
- help='Write ASCII file (default is binary)')
- parser.add_argument('-b', '--binary', action='store_true',
- help='Force binary file (for TTYs)')
+ parser.add_argument(
+ '-a', '--ascii', action='store_true',
+ help='Write ASCII file (default is binary)'
+ )
+ parser.add_argument(
+ '-b', '--binary', action='store_true',
+ help='Force binary file (for TTYs)'
+ )
 
  args = parser.parse_args()
  name = _get_name(args)
- stl_file = stl.StlMesh(filename=name,
- fh=args.infile,
- calculate_normals=False,
- remove_empty_areas=args.remove_empty_areas,
- speedups=not args.disable_speedups)
+ stl_file = stl.StlMesh(
+ filename=name,
+ fh=args.infile,
+ calculate_normals=False,
+ remove_empty_areas=args.remove_empty_areas,
+ speedups=not args.disable_speedups
+ )
 
  if args.binary:
  mode = stl.BINARY
@@ -61,30 +81,39 @@ def main():
  else:
  mode = stl.AUTOMATIC
 
- stl_file.save(name, args.outfile, mode=mode,
- update_normals=not args.use_file_normals)
+ stl_file.save(
+ name, args.outfile, mode=mode,
+ update_normals=not args.use_file_normals
+ )
 
 
 def to_ascii():
  parser = _get_parser('Convert STL files to ASCII (text) format')
  args = parser.parse_args()
  name = _get_name(args)
- stl_file = stl.StlMesh(filename=name, fh=args.infile,
- calculate_normals=False,
- remove_empty_areas=args.remove_empty_areas,
- speedups=not args.disable_speedups)
- stl_file.save(name, args.outfile, mode=stl.ASCII,
- update_normals=not args.use_file_normals)
+ stl_file = stl.StlMesh(
+ filename=name, fh=args.infile,
+ calculate_normals=False,
+ remove_empty_areas=args.remove_empty_areas,
+ speedups=not args.disable_speedups
+ )
+ stl_file.save(
+ name, args.outfile, mode=stl.ASCII,
+ update_normals=not args.use_file_normals
+ )
 
 
 def to_binary():
  parser = _get_parser('Convert STL files to binary format')
  args = parser.parse_args()
  name = _get_name(args)
- stl_file = stl.StlMesh(filename=name, fh=args.infile,
- calculate_normals=False,
- remove_empty_areas=args.remove_empty_areas,
- speedups=not args.disable_speedups)
- stl_file.save(name, args.outfile, mode=stl.BINARY,
- update_normals=not args.use_file_normals)
-
+ stl_file = stl.StlMesh(
+ filename=name, fh=args.infile,
+ calculate_normals=False,
+ remove_empty_areas=args.remove_empty_areas,
+ speedups=not args.disable_speedups
+ )
+ stl_file.save(
+ name, args.outfile, mode=stl.BINARY,
+ update_normals=not args.use_file_normals
+ )