Spaces:
Running
Running
File size: 2,978 Bytes
c412b90 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 |
import PyPDF2
import docx
import io
import datetime
from components.candidate_data_schema import candidate
import re
def read_pdf_text(resume_file):
"""
Extracts text from a PDF file.
Args:
resume_file (file-like object): The PDF file to be read.
Returns:
str: The extracted text from the PDF.
"""
pdf_reader = PyPDF2.PdfReader(io.BytesIO(resume_file.read()))
text = ""
for page_num in range(len(pdf_reader.pages)):
page = pdf_reader.pages[page_num]
text += page.extract_text().strip()
return text
def read_docx_text(word_file):
"""
Extracts text from a DOCX file.
Args:
word_file (file-like object): The DOCX file to be read.
Returns:
str: The extracted text from the DOCX file.
"""
doc = docx.Document(io.BytesIO(word_file.read()))
text = ""
for paragraph in doc.paragraphs:
text += paragraph.text.strip() + "\n"
return text
def extract_resume_text(resume_file):
"""
Extracts text from a resume file, either PDF or DOCX.
Args:
resume_file (file-like object): The resume file to be read.
Returns:
str: The extracted text from the resume file.
"""
file_type = resume_file.name.split(".")[-1]
if file_type == "pdf":
return read_pdf_text(resume_file)
elif file_type == "docx":
return read_docx_text(resume_file)
def date_to_datetime(input):
"""
Converts a dictionary representing a date to a datetime.date object.
Args:
input (dict): Dictionary with keys 'year', 'month', 'day'.
Returns:
datetime.date or None: The corresponding datetime.date object or None if input is invalid.
"""
for _, value in input.items():
if value is None:
return None
return datetime.date(**input)
def convert_dates_to_datetime(candidate_data: candidate):
"""
Returns the model_dump() dictionary of a "candidate" pydantic class after converting dates to datetime.date objects.
Args:
candidate_data (candidate): The candidate object containing date fields.
Returns:
dict: The candidate model_dump dictionary with date fields converted to datetime.date objects.
"""
candidate_dict = candidate_data.model_dump()
if "degrees" in candidate_dict.keys():
for degree in candidate_dict["degrees"]:
if degree["graduation_date"]:
degree["graduation_date"] = date_to_datetime(degree["graduation_date"])
if "jobs" in candidate_dict.keys():
for job in candidate_dict["jobs"]:
if job["started_at"]:
job["started_at"] = date_to_datetime(job["started_at"])
if job["ended_at"]:
job["ended_at"] = date_to_datetime(job["ended_at"])
return candidate_dict
def is_valid_email(email):
pattern = r'^[\w\.-]+@[\w\.-]+\.\w+$'
return re.match(pattern, email) is not None |