Extracting Java methods: if there are no subdirectories, extract from the root dir

Author: urialon

JavaExtractor/extract.py

@@ -92,5 +92,9 @@ def ExtractFeaturesForDirsList(args, dirs):
  os.system(command)
  elif args.dir is not None:
  subdirs = get_immediate_subdirectories(args.dir)
- ExtractFeaturesForDirsList(args, subdirs)
- 
+ to_extract = subdirs
+ if len(subdirs) == 0:
+ to_extract = [sys.argv[2]]
+ ExtractFeaturesForDirsList(args, to_extract)
+
+