SeniorityCalculator.py 1.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. #!/usr/bin/env python3
  2. from parse import parse_mstrsen, parse_vacsen, mergesen
  3. import pandas as pd
  4. import sys
  5. import os
  6. ########################
  7. ## ARGUMENTS ##
  8. if not len(sys.argv)==3:
  9. print("Need Vacancy Seniority.pdf and Master Seniority.html files as inputs!")
  10. sys.exit("Incorrent Arguments. " + sys.argv[0] + " aborted.")
  11. elif not sys.argv[1].lower().endswith('.pdf'):
  12. sys.exit("Vacancy Seniority file is not a PDF file." + sys.argv[0] + " aborted.")
  13. elif not sys.argv[2].lower().endswith('.html'):
  14. sys.exit("Master Seniority file is not an HTML file." + sys.argv[0] + " aborted.")
  15. else:
  16. if os.path.exists(sys.argv[1]):
  17. VacancyFile=sys.argv[1]
  18. else:
  19. sys.exit("Vacancy Seniority file not found. " + sys.argv[0] + " aborted.")
  20. if os.path.exists(sys.argv[2]):
  21. MasterSenFile=sys.argv[2]
  22. else:
  23. sys.exit("Master Seniority file not found. " + sys.argv[0] + " aborted.")
  24. ########################
  25. print("Starting " + sys.argv[0])
  26. df_mastersen = parse_mstrsen(MasterSenFile)
  27. df_domsen = parse_vacsen(VacancyFile)
  28. df_mergesen = mergesen(df_domsen, df_mastersen)
  29. print("Parsing Complete!")
  30. print("-----------------")
  31. print("Files are formatted for additional analysis")
  32. print("Try running alalyze.py for processing")