-r Recursive option

pyt/__main__.py

@@ -30,6 +30,24 @@
 )
 
 
+def discover_files(targets, excluded_files, recursive=False):
+    included_files = list()
+    excluded_list = excluded_files.split(",")
+
+    for target in targets:
+        if os.path.isdir(target):
+            if recursive:
+                for root, dirs, files in os.walk(target):
+                    for f in files:
+                        fullpath = os.path.join(root, f)
+                        if os.path.splitext(fullpath)[1] == '.py' and fullpath.split("/")[-1] not in excluded_list:
+                            included_files.append(fullpath)
+        else:
+            if targets not in excluded_list:
+                included_files.append(targets[0])
+    return(included_files)
+
+
 def main(command_line_args=sys.argv[1:]):
     args = parse_args(command_line_args)
 
@@ -39,6 +57,14 @@ def main(command_line_args=sys.argv[1:]):
     elif args.trim_reassigned_in:
         ui_mode = UImode.TRIM
 
+
+
+    targets = args.targets
+    excluded_files = args.excluded_paths
+    recursive = args.recursive
+    test = discover_files(targets, excluded_files, recursive) #just for see files in directory
+    print(test)
+
     path = os.path.normpath(args.filepath)
 
     if args.ignore_nosec:

pyt/usage.py

@@ -29,11 +29,11 @@ def valid_date(s):
 
 def _add_required_group(parser):
     required_group = parser.add_argument_group('required arguments')
-    required_group.add_argument(
+    '''required_group.add_argument(
         '-f', '--filepath',
         help='Path to the file that should be analysed.',
         type=str
-    )
+    )'''
 
 
 def _add_optional_group(parser):
@@ -91,7 +91,21 @@ def _add_optional_group(parser):
         action='store_true',
         help='do not skip lines with # nosec comments'
     )
-
+    optional_group.add_argument(
+        '-r', '--recursive', dest='recursive',
+        action='store_true', help='find and process files in subdirectories'
+    )
+    optional_group.add_argument(
+        '-x', '--exclude',
+        dest='excluded_paths',
+        action='store',
+        default='',
+        help='Separate files with commas'
+    )
+    optional_group.add_argument(
+        'targets', metavar='targets', type=str, nargs='*',
+        help='source file(s) or directory(s) to be tested'
+    )
 
 def _add_print_group(parser):
     print_group = parser.add_argument_group('print arguments')
@@ -110,8 +124,8 @@ def _add_print_group(parser):
 
 
 def _check_required_and_mutually_exclusive_args(parser, args):
-    if args.filepath is None:
-        parser.error('The -f/--filepath argument is required')
+    if args.targets is None:
+        parser.error('The target argument is required')
 
 
 def parse_args(args):