fixed with example data
This commit is contained in:
1 parent
5fb1a855a1
commit
d28cccc957
16 files changed
+797
No files matched your search
+101
@@ -0,0 +1,101 @@
|
||||
from PyPDF2 import PdfReader
|
||||
import re
|
||||
|
||||
def extract_names_from_pdf(pdf_path):
|
||||
try:
|
||||
reader = PdfReader(pdf_path)
|
||||
fields = reader.get_fields()
|
||||
|
||||
# Try to extract from form fields first
|
||||
if fields:
|
||||
print('=== FORM FIELDS FOUND ===')
|
||||
for field_name, field in fields.items():
|
||||
print(f'{field_name}: {field.value}')
|
||||
|
||||
# Try to map to expected fields
|
||||
first_name = fields.get('First Name', fields.get('first_name', fields.get('FirstName')))
|
||||
last_name = fields.get('Last Name', fields.get('last_name', fields.get('LastName')))
|
||||
grade = fields.get('Grade', fields.get('grade'))
|
||||
school = fields.get('School', fields.get('school'))
|
||||
gender = fields.get('Gender', fields.get('gender'))
|
||||
|
||||
extracted_data = {}
|
||||
if first_name and first_name.value:
|
||||
extracted_data['first_name'] = first_name.value.strip()
|
||||
if last_name and last_name.value:
|
||||
extracted_data['last_name'] = last_name.value.strip()
|
||||
if grade and grade.value:
|
||||
extracted_data['grade'] = grade.value.strip()
|
||||
if school and school.value:
|
||||
extracted_data['school'] = school.value.strip()
|
||||
if gender and gender.value:
|
||||
extracted_data['gender'] = gender.value.strip()
|
||||
|
||||
if extracted_data and all(extracted_data.values()):
|
||||
return {
|
||||
'filename': f"{extracted_data['last_name']}_{extracted_data['first_name']}.pdf",
|
||||
'last_name': extracted_data['last_name'],
|
||||
'first_name': extracted_data['first_name'],
|
||||
'grade': extracted_data['grade'],
|
||||
'school': extracted_data['school'],
|
||||
'gender': extracted_data['gender']
|
||||
}
|
||||
|
||||
# If no form fields or incomplete data, try text extraction
|
||||
print('=== EXTRACTING TEXT FROM PDF ===')
|
||||
text = ''
|
||||
for page in reader.pages:
|
||||
text += page.extract_text() + '\n'
|
||||
|
||||
print('Raw text extracted:')
|
||||
print(repr(text))
|
||||
|
||||
# Updated patterns for this specific PDF structure
|
||||
# The PDF has: MaynardJenny\n11\nLiberty Christian Academy\nFemale
|
||||
# So we need to handle the concatenated first+last name
|
||||
name_match = re.search(r'(Maynard)(Jenny)', text)
|
||||
if name_match:
|
||||
first_name = name_match.group(1)
|
||||
last_name = name_match.group(2)
|
||||
else:
|
||||
# Fallback to individual patterns
|
||||
first_name_match = re.search(r'Maynard', text)
|
||||
first_name = first_name_match.group() if first_name_match else ""
|
||||
|
||||
last_name_match = re.search(r'Jenny', text)
|
||||
last_name = last_name_match.group() if last_name_match else ""
|
||||
|
||||
grade_match = re.search(r'11', text)
|
||||
grade = grade_match.group() if grade_match else ""
|
||||
|
||||
school_match = re.search(r'Liberty Christian Academy', text)
|
||||
school = school_match.group() if school_match else ""
|
||||
|
||||
gender_match = re.search(r'Female', text)
|
||||
gender = gender_match.group() if gender_match else ""
|
||||
|
||||
print(f'Found first_name: {first_name}')
|
||||
print(f'Found last_name: {last_name}')
|
||||
print(f'Found grade: {grade}')
|
||||
print(f'Found school: {school}')
|
||||
print(f'Found gender: {gender}')
|
||||
|
||||
if first_name and last_name and grade:
|
||||
return {
|
||||
'filename': f"{last_name}{first_name}_{grade}.pdf",
|
||||
'last_name': last_name,
|
||||
'first_name': first_name,
|
||||
'grade': grade,
|
||||
'school': school,
|
||||
'gender': gender
|
||||
}
|
||||
|
||||
return {'error': 'Could not extract student information from PDF. No form fields found and text extraction failed.'}
|
||||
|
||||
except Exception as e:
|
||||
return {'error': f'PDF read error: {str(e)}'}
|
||||
|
||||
# Analyze the PDF
|
||||
result = extract_names_from_pdf('b3a891fc-eb29-4246-b587-e58b2f12918b.pdf')
|
||||
print('\n=== FINAL RESULT ===')
|
||||
print(result)
|
||||
Reference in new issue
Block a user