Initial commit of Roster

main.py

@@ -0,0 +1,102 @@
+from modules.adjust_layout import adjust_layout
+from modules.add_logo import add_logo
+from pathlib import Path
+from modules.add_date import add_date
+from datetime import datetime
+from dateutil.relativedelta import relativedelta # type: ignore
+import argparse
+import fitz # type: ignore
+
+def convert_pdf_to_png(pdf_path, output_path, dpi=150):
+    doc = fitz.open(pdf_path)
+    page = doc.load_page(0)
+    pix = page.get_pixmap(dpi=dpi)
+    pix.save(output_path)
+    doc.close()
+
+parser = argparse.ArgumentParser(description="Generate formatted roster")
+parser.add_argument("--month", type=str, help="Override the month/year label (e.g., 'September 2025')")
+args = parser.parse_args()
+
+if args.month:
+    month_year = args.month.upper()
+else:
+    today = datetime.now()
+    target = today if today.day < 15 else today + relativedelta(months=1)
+    month_year = target.strftime("%B %Y").upper()
+
+
+# Define paths
+base = Path(__file__).parent
+input_pdf = base / "input/original.pdf"
+shifted_pdf = base / "output/shifted.pdf"
+
+# Apply layout adjustment
+adjust_layout(input_pdf, shifted_pdf, shift_amount=100)
+
+# Paths
+logo_input = shifted_pdf
+logo_output = base / "output/with_logo.pdf"
+logo_file = base / "assets/logo.png"
+
+add_logo(logo_input, logo_output, logo_file, position=(50, -10), scale=0.27)
+
+
+
+# Determine month/year label
+if args.month:
+    parts = args.month.strip().split()
+    full_month = parts[0].capitalize()
+    year = parts[1] if len(parts) > 1 else str(datetime.now().year)
+
+    # Smart abbreviation
+    if len(full_month) > 5:
+        month = full_month[:3].upper()  # "September" → "SEP"
+    else:
+        month = full_month.upper()      # "June" → "JUNE"
+
+    month_year = f"{month} {year}"
+else:
+    today = datetime.now()
+    target = today if today.day < 15 else today + relativedelta(months=1)
+    full_month = target.strftime("%B")
+    year = target.strftime("%Y")
+
+    if len(full_month) > 5:
+        month = target.strftime("%b").upper()
+    else:
+        month = full_month.upper()
+
+    month_year = f"{month} {year}"
+
+final_pdf_name = f"{month} {year} Roster.pdf"
+final_pdf_path = base / "output" / final_pdf_name
+final_png_path = final_pdf_path.with_suffix(".png")
+date_output = final_pdf_path
+
+# Paths
+date_input = logo_output
+font_file = base / "assets/fonts/Roboto-Variable.ttf"
+
+add_date(
+    input_path=date_input,
+    output_path=date_output,
+    month_year=month_year,
+    font_path=font_file,
+    position=(460, 88),
+    font_size=48
+)
+
+convert_pdf_to_png(final_pdf_path, final_png_path)
+
+final_pdf_name = f"{month} {year} Roster.pdf"
+final_pdf_path = base / "output" / final_pdf_name
+
+final_png_name = final_pdf_name.replace(".pdf", ".png")
+final_png_path = base / "output" / final_png_name
+
+
+# Clean up intermediate PDFs
+for path in [shifted_pdf, logo_output, final_pdf_path]:
+    if path.exists():
+        path.unlink()