126 lines
3.9 KiB
Python
126 lines
3.9 KiB
Python
import argparse
|
|
import math
|
|
import shutil
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import pandas as pd
|
|
from PIL import Image, ImageDraw, ImageFont
|
|
|
|
from config import FINAL_SCORE_FONT_PATH, FINAL_SCORE_ODS_PATH, FINAL_SCORE_OUTPUT_DIR
|
|
|
|
# Configuration constants
|
|
ODS_PATH = Path(FINAL_SCORE_ODS_PATH).expanduser()
|
|
OUTPUT_DIR = Path(FINAL_SCORE_OUTPUT_DIR).expanduser()
|
|
|
|
|
|
def score_font(size):
|
|
candidates = [FINAL_SCORE_FONT_PATH, "DejaVuSans-Bold.ttf", "arial.ttf"]
|
|
for candidate in candidates:
|
|
if not candidate:
|
|
continue
|
|
try:
|
|
return ImageFont.truetype(str(candidate), size)
|
|
except OSError:
|
|
continue
|
|
return ImageFont.load_default()
|
|
|
|
def get_rounded_score(score):
|
|
"""Round score to one decimal place below (floor)."""
|
|
try:
|
|
val = float(score)
|
|
return math.floor(val * 10) / 10
|
|
except (ValueError, TypeError):
|
|
return None
|
|
|
|
def process_images(base_dir, output_dir):
|
|
# 1. Load Data
|
|
try:
|
|
# header=None assumes the file starts directly with data.
|
|
# If row 0 is a header, change to header=0
|
|
df = pd.read_excel(ODS_PATH, engine="odf", header=None)
|
|
# Create a lookup dictionary: {Name: Score}
|
|
score_db = dict(zip(df[0], df[1]))
|
|
except Exception as e:
|
|
print(f"CRITICAL ERROR: Could not read ODS file.\n{e}")
|
|
sys.exit(1)
|
|
|
|
# 2. Prepare Output Directory
|
|
output_dir.mkdir(parents=True, exist_ok=True)
|
|
|
|
# 3. Iterate Files
|
|
# Structure: Dir/A Rendre/{name}/{name}.jpg
|
|
search_path = base_dir / "A Rendre"
|
|
|
|
if not search_path.exists():
|
|
print(f"Error: Directory '{search_path}' not found.")
|
|
sys.exit(1)
|
|
|
|
for img_path in sorted(search_path.glob("*/*.jpg")):
|
|
student_name = img_path.stem # Filename without extension
|
|
|
|
# 4. Find Score
|
|
if student_name not in score_db:
|
|
print(f"Error: Student '{student_name}' not found in ODS file.")
|
|
continue
|
|
|
|
raw_score = score_db[student_name]
|
|
score = get_rounded_score(raw_score)
|
|
|
|
if score is None:
|
|
print(f"Error: Invalid score '{raw_score}' for '{student_name}'.")
|
|
continue
|
|
|
|
# 5. Process Image
|
|
try:
|
|
with Image.open(img_path) as img:
|
|
img = img.convert("RGB")
|
|
draw = ImageDraw.Draw(img)
|
|
width, height = img.size
|
|
|
|
# Dynamic font size (15% of image height)
|
|
font_size = int(width * 0.08)
|
|
|
|
font = score_font(font_size)
|
|
|
|
text = str(score)
|
|
|
|
# Calculate text size and position (Top Right)
|
|
bbox = draw.textbbox((0, 0), text, font=font)
|
|
text_w = bbox[2] - bbox[0]
|
|
text_h = bbox[3] - bbox[1]
|
|
|
|
# 30px padding
|
|
x = width - text_w - 30
|
|
y = 30
|
|
|
|
# Draw Text (Red)
|
|
draw.text((x, y), text, fill=(255, 0, 0), font=font)
|
|
|
|
# Save
|
|
save_path = output_dir / f"{student_name}.jpg"
|
|
img.save(save_path)
|
|
print(f"Processed: {student_name} -> {score}")
|
|
|
|
except Exception as e:
|
|
print(f"Error processing image for '{student_name}': {e}")
|
|
|
|
for pdf_path in sorted(search_path.glob("*/*.pdf")):
|
|
student_name = pdf_path.stem # Filename without extension
|
|
save_path = output_dir / f"{student_name}.pdf"
|
|
|
|
shutil.copy(str(pdf_path), str(save_path))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="Stamp scores on exam copies.")
|
|
parser.add_argument("dir", type=Path, help="Root directory containing 'A Rendre' folder")
|
|
|
|
|
|
args = parser.parse_args()
|
|
|
|
base_dir = args.dir.expanduser().resolve()
|
|
output_dir = OUTPUT_DIR / base_dir.name
|
|
output_dir.mkdir(parents=True, exist_ok=True)
|
|
process_images(base_dir, output_dir)
|