Updated command_line.py for specifying explicit skills file for brute forcing

Author: OmkarPathak

pyresparser/command_line.py

@@ -3,10 +3,12 @@
 import os
 import argparse
 from pprint import pprint
-from pyresparser import ResumeParser
+import io
+import csv
 import multiprocessing as mp
 from urllib.request import Request, urlopen
-import io
+from pyresparser import ResumeParser
+from itertools import product
 
 def print_cyan(text):
     print("\033[96m {}\033[00m" .format(text))
@@ -17,39 +19,49 @@ def __init__(self):
         self.__parser.add_argument('-f', '--file', help="resume file to be extracted")
         self.__parser.add_argument('-d', '--directory', help="directory containing all the resumes to be extracted")
         self.__parser.add_argument('-r', '--remotefile', help="remote path for resume file to be extracted")
-        return
+        self.__parser.add_argument('-sf', '--skillsfile', help="custom skills CSV file against which skills are searched for")
 
     def extract_resume_data(self):
         args = self.__parser.parse_args()
 
         if args.remotefile:
             return self.__extract_from_remote_file(args.remotefile)
         if args.file and not args.directory:
-            return self.__extract_from_file(args.file)
+            if args.skillsfile:
+                return self.__extract_from_file(args.file, args.skillsfile)
+            else:
+                return self.__extract_from_file(args.file)
         elif args.directory and not args.file:
-            return self.__extract_from_directory(args.directory)
+            if args.skillsfile:
+                return self.__extract_from_directory(args.directory, args.skillsfile)
+            else:
+                return self.__extract_from_directory(args.directory)
         else:
-            return 'Invalid option. Please provide a valid option.'
+            self.__parser.print_help()
 
-    def __extract_from_file(self, file):
+    def __extract_from_file(self, file, skills_file=None):
         if os.path.exists(file):
             print_cyan('Extracting data from: {}'.format(file))
-            resume_parser = ResumeParser(file)
+            if skills_file:
+                resume_parser = ResumeParser(file, skills_file)
+            else:
+                resume_parser = ResumeParser(file)
             return [resume_parser.get_extracted_data()]
         else:
             return 'File not found. Please provide a valid file name.'
 
-    def __extract_from_directory(self, directory):
+    def __extract_from_directory(self, directory, skills_file=None):
         if os.path.exists(directory):
             pool = mp.Pool(mp.cpu_count())
 
             resumes = []
-            data = []
-            for root, directories, filenames in os.walk(directory):
+            for root, _, filenames in os.walk(directory):
                 for filename in filenames:
                     file = os.path.join(root, filename)
-                    resumes.append(file)
-
+                    if skills_file:
+                        resumes.append([file, skills_file])
+                    else:
+                        resumes.append(file)
             results = pool.map(resume_result_wrapper, resumes)
             pool.close()
             pool.join()
@@ -67,11 +79,15 @@ def __extract_from_remote_file(self, remote_file):
         resume_parser = ResumeParser(_file)
         return [resume_parser.get_extracted_data()]
 
-def resume_result_wrapper(resume):
-    print_cyan('Extracting data from: {}'.format(resume))
-    parser = ResumeParser(resume)
+def resume_result_wrapper(args):
+    if len(args) == 2:
+        print_cyan('Extracting data from: {}'.format(args[0]))
+        parser = ResumeParser(args[0], args[1])
+    else:
+        print_cyan('Extracting data from: {}'.format(args))
+        parser = ResumeParser(args)
     return parser.get_extracted_data()
 
 def main():
     cli_obj = ResumeParserCli()
-    pprint(cli_obj.extract_resume_data())
+    pprint(cli_obj.extract_resume_data())