enhanced cli

Author: anemele

filetype/__main__.py

@@ -1,4 +1,6 @@
-import sys
+import glob
+from itertools import chain
+from os.path import isfile
 
 import filetype
 
@@ -17,20 +19,22 @@ def main():
     parser = argparse.ArgumentParser(
         prog='filetype', description='Determine type of FILEs.'
     )
-    parser.add_argument('-f', '--file', nargs='+')
+    parser.add_argument(
+        'file', nargs='+',
+        help='files, wildcard is supported'
+    )
     parser.add_argument(
         '-v', '--version', action='version',
-        version='%(prog)s ' + filetype.version,
+        version=f'%(prog)s {filetype.version}',
         help='output version information and exit'
     )
 
     args = parser.parse_args()
-    if len(sys.argv) < 2:
-        parser.print_help()
-        sys.exit(1)
+    items = chain.from_iterable(map(glob.iglob, args.file))
+    files = filter(isfile, items)
 
-    for i in args.file:
-        guess(i)
+    for file in files:
+        guess(file)
 
 
 if __name__ == '__main__':