fix --ext default value

titusz · Jul 21, 2015 · e425ece · e425ece
1 parent 2f5feb7
commit e425ece
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/onixcheck/__main__.py b/src/onixcheck/__main__.py
@@ -14,7 +14,7 @@
 @click.command()
 @click.argument('infile', type=click.File('rb'), required=False)
 @click.option('--path', '-p', type=click.Path(exists=True, resolve_path=True), help='Validate all files in path')
-@click.option('--ext', '-e', multiple=True, default={'xml'}, help='Extension to check (default: xml)')
+@click.option('--ext', '-e', multiple=True, help='Extension to check (default: xml, onx, onix)')
 @click.option('--recursive', '-r', default=False, is_flag=True, help='Scan subdirectories in path')
 @click.option('--debug', '-d', is_flag=True, help='Show debug information')
 @click.version_option(prog_name='Onixcheck')

diff --git a/src/onixcheck/utils.py b/src/onixcheck/utils.py
@@ -4,7 +4,7 @@
 from os.path import splitext, join
 from scandir import scandir, walk
 
-DEFAULT_EXTENSIONS = {'xml', 'onx', 'onix'}
+DEFAULT_EXTENSIONS = ('xml', 'onx', 'onix')
 
 
 def iter_files(root, exts=None, recursive=False):
@@ -20,7 +20,7 @@ def iter_files(root, exts=None, recursive=False):
     if exts is None:
         exts = DEFAULT_EXTENSIONS
 
-    exts = {x.lower() for x in exts}
+    exts = set((x.lower() for x in exts))
 
     if recursive is False:
         for entry in scandir(root):