Photo-Composition-Designer 0.0.7__py3-none-any.whl

Photo_Composition_Designer/image/CalendarRenderer.py

@@ -0,0 +1,319 @@
+from __future__ import annotations
+
+import calendar
+import locale
+import logging
+from datetime import datetime, timedelta
+from pathlib import Path
+
+import holidays
+import pytz
+from astral import LocationInfo
+from astral.sun import sun
+from PIL import Image, ImageDraw
+
+from Photo_Composition_Designer.common.Anniversaries import Anniversaries
+from Photo_Composition_Designer.common.MoonPhase import MoonPhase
+from Photo_Composition_Designer.config.config import ConfigParameterManager
+from Photo_Composition_Designer.tools.Helpers import load_font
+
+
+class CalendarRenderer:
+    """Responsible for rendering a weekly calendar strip with holidays,
+    sun times and anniversaries.
+    """
+
+    def __init__(
+        self,
+        backgroundColor: tuple[int],
+        textColor1: tuple[int],
+        textColor2: tuple[int],
+        holidayColor: tuple[int],
+        language: str,
+        startDate: datetime,
+        holidayCountries: list[str],
+        fontSizeSmall: float,
+        fontSizeLarge: float,
+        fontSizeAnniversaries: float,
+        useShortDayNames: bool,
+        useShortMonthNames: bool,
+        marginSides: float,
+        anniversaries: Anniversaries | None = None,
+    ) -> None:
+        self.anniversaries = anniversaries or Anniversaries()
+
+        # Convert colors to 0–255 int tuples used by Pillow
+        self.backgroundColor = backgroundColor
+        self.textColor1 = textColor1
+        self.textColor2 = textColor2
+        self.holidayColor = holidayColor
+
+        self.fontSizeSmall = fontSizeSmall
+        self.fontSizeLarge = fontSizeLarge
+        self.fontSizeAnniversaries = fontSizeAnniversaries
+
+        self.language = language
+        self.startDate = startDate
+        self.holidayCountries = holidayCountries
+        self.useShortDayNames = useShortDayNames
+        self.useShortMonthNames = useShortMonthNames
+        self.marginSides = marginSides
+
+        # Extract country code from locale ("de_DE" → "DE")
+        country_code = language.split("_")[1].upper()
+
+        self.localHolidays = self.get_combined_holidays(
+            startDate.year,
+            country_code,
+            holidayCountries,
+        )
+
+    # -------------------------------------------------------------------------
+    # Rendering
+    # -------------------------------------------------------------------------
+
+    def generate(self, d: datetime, width: int | float, height: int | float) -> Image.Image:
+        """Render full weekly calendar image."""
+        width = int(width)
+        height = int(height)
+        week_dates = [d + timedelta(days=i) for i in range(7)]
+
+        img = Image.new("RGB", (width, height), self.backgroundColor)
+        draw = ImageDraw.Draw(img)
+
+        font_large = load_font("DejaVuSans.ttf", int(self.fontSizeLarge))
+        font_small = load_font("DejaVuSans.ttf", int(self.fontSizeSmall))
+        font_ann = load_font("DejaVuSans.ttf", int(self.fontSizeAnniversaries))
+
+        # Header (month + year)
+        month_name = self.get_month_name(
+            week_dates[0].month,
+            locale_name=self.language,
+            abbreviation=self.useShortMonthNames,
+        )
+        header_text = f"{month_name} {str(d.year)[-2:]}"
+        draw.text(
+            (0, height - self.fontSizeAnniversaries),
+            header_text,
+            font=font_large,
+            fill=self.textColor2,
+            anchor="ld",
+        )
+
+        # Sun times for Europe/Berlin
+        location = LocationInfo("Dresden", "Germany", "Europe/Berlin", 51.0504, 13.7373)
+        tz = pytz.timezone("Europe/Berlin")
+        sun_times = sun(location.observer, date=d)
+
+        sunrise = sun_times["sunrise"].astimezone(tz).strftime("%H:%M")
+        sunset = sun_times["sunset"].astimezone(tz).strftime("%H:%M")
+        week_no = d.isocalendar().week
+
+        sun_string = f"KW {week_no}  ● ↑ {sunrise}  ○ ↓ {sunset}"
+        draw.text(
+            (0, height),
+            sun_string,
+            font=font_ann,
+            fill=self.textColor2,
+            anchor="ld",
+        )
+
+        # Day columns
+        month_cols, col_width = self.get_cols_property(width)
+
+        for idx, day_date in enumerate(week_dates):
+            x = self.marginSides + (idx + month_cols + 0.5) * col_width
+
+            date_key = (day_date.day, day_date.month)
+            holiday_name = self.localHolidays.get(day_date)
+
+            is_weekend = day_date.weekday() >= 5
+            is_holiday = day_date in self.localHolidays
+
+            color_day = self.holidayColor if (is_holiday or is_weekend) else self.textColor1
+
+            # Day name
+            day_name = self.get_day_name(day_date.weekday(), self.language)
+            if self.useShortDayNames:
+                day_name = day_name[:2]
+
+            moon_symbol = MoonPhase.get_moon_phase_symbol_dark(day_date)
+            if moon_symbol:
+                day_name = f"{day_name} {moon_symbol}"
+
+            draw.text(
+                (x, height - self.fontSizeAnniversaries - self.fontSizeLarge * 1.15),
+                day_name,
+                font=font_small,
+                fill=self.textColor2,
+                anchor="md",
+            )
+
+            draw.text(
+                (x, height - self.fontSizeAnniversaries),
+                str(day_date.day),
+                font=font_large,
+                fill=color_day,
+                anchor="md",
+            )
+
+            # Anniversaries + holidays
+            label = None
+            if date_key in self.anniversaries:
+                label = self.anniversaries[date_key]
+                if holiday_name:
+                    label += f", {holiday_name}"
+                draw.fill = self.textColor1
+            elif holiday_name:
+                label = holiday_name
+
+            if label:
+                draw.text(
+                    (x, height),
+                    label,
+                    font=font_ann,
+                    fill=self.holidayColor,
+                    anchor="md",
+                )
+
+        return img
+
+    def generateTitle(self, title: str, width: int | float, height: int | float) -> Image.Image:
+        width = int(width)
+        height = int(height)
+        img = Image.new("RGB", (width, height), self.backgroundColor)
+        draw = ImageDraw.Draw(img)
+        font_large = load_font("DejaVuSans.ttf", int(self.fontSizeLarge))
+
+        draw.text(
+            (width // 2, height - self.fontSizeAnniversaries),
+            title,
+            font=font_large,
+            fill=self.textColor1,
+            anchor="md",
+        )
+        return img
+
+    # -------------------------------------------------------------------------
+    # Helpers
+    # -------------------------------------------------------------------------
+
+    def get_cols_property(self, width: int) -> tuple[float, float]:
+        month_cols = 1.5 if self.useShortMonthNames else 4.0
+        col_width = (width - 3 * self.marginSides) / (7.0 + month_cols)
+        return month_cols, col_width
+
+    @staticmethod
+    def get_month_name(month: int, locale_name: str, abbreviation: bool = False) -> str:
+        try:
+            locale.setlocale(locale.LC_TIME, locale_name)
+            return calendar.month_abbr[month] if abbreviation else calendar.month_name[month]
+        except locale.Error:
+            return calendar.month_abbr[month] if abbreviation else calendar.month_name[month]
+        finally:
+            locale.setlocale(locale.LC_TIME, "")
+
+    @staticmethod
+    def get_day_name(day: int, locale_name: str) -> str:
+        try:
+            locale.setlocale(locale.LC_TIME, locale_name)
+            return calendar.day_name[day]
+        except locale.Error:
+            return calendar.day_name[day]
+        finally:
+            locale.setlocale(locale.LC_TIME, "")
+
+    @staticmethod
+    def get_combined_holidays(year: int, country: str, subdivs: list[str]) -> holidays.HolidayBase:
+        years = (year, year + 1)
+        combined = holidays.HolidayBase()
+        combined.update(holidays.country_holidays(country, years=years))
+        try:
+            for sub in subdivs:
+                combined.update(holidays.country_holidays(country, years=years, subdiv=sub))
+        except Exception as e:
+            logging.warning(f"Unable to load holiday subdivision {sub}: {e}")
+
+        return combined
+
+
+# -----------------------------------------------------------------------------
+# Main helpers for production use
+# -----------------------------------------------------------------------------
+
+
+def from_config(config: ConfigParameterManager) -> CalendarRenderer:
+    """Factory function to create CalendarGenerator using the config manager."""
+    return CalendarRenderer(
+        backgroundColor=config.colors.backgroundColor.value.to_pil(),
+        textColor1=config.colors.textColor1.value.to_pil(),
+        textColor2=config.colors.textColor2.value.to_pil(),
+        holidayColor=config.colors.holidayColor.value.to_pil(),
+        language=config.calendar.language.value,
+        startDate=config.calendar.startDate.value,
+        holidayCountries=[s.strip() for s in config.calendar.holidayCountries.value.split(",")],
+        fontSizeSmall=int(
+            config.layout.fontSizeSmall.value
+            * config.size.calendarHeight.value
+            * config.size.dpi.value
+            / 25.4
+        ),
+        fontSizeLarge=int(
+            config.layout.fontSizeLarge.value
+            * config.size.calendarHeight.value
+            * config.size.dpi.value
+            / 25.4
+        ),
+        fontSizeAnniversaries=int(
+            config.layout.fontSizeAnniversaries.value
+            * config.size.calendarHeight.value
+            * config.size.dpi.value
+            / 25.4
+        ),
+        useShortDayNames=config.layout.useShortDayNames.value,
+        useShortMonthNames=config.layout.useShortMonthNames.value,
+        marginSides=int(config.layout.marginSides.value * config.size.dpi.value / 25.4),
+        anniversaries=None,  # use default
+    )
+
+
+def main() -> None:
+    import os
+
+    from Photo_Composition_Designer.config.config import ConfigParameterManager
+
+    config = ConfigParameterManager()
+    cg = from_config(config)
+
+    # Create temp directory
+    project_root = Path(__file__).resolve().parents[3]
+    temp_dir = project_root / "temp"
+    temp_dir.mkdir(exist_ok=True)
+
+    title = config.general.compositionTitle.value + " " + str(config.calendar.startDate.value.year)
+    first = True
+
+    for w in range(config.calendar.collagesToGenerate.value):
+        dt = config.calendar.startDate.value + timedelta(weeks=w - 2)
+
+        if first:
+            img = cg.generateTitle(
+                title,
+                width=int(config.size.width.value * config.size.dpi.value / 25.4),
+                height=int(config.size.calendarHeight.value * config.size.dpi.value / 25.4),
+            )
+            first = False
+        else:
+            img = cg.generate(
+                dt,
+                width=int(config.size.width.value * config.size.dpi.value / 25.4),
+                height=config.size.calendarHeight.value * config.size.dpi.value / 25.4,
+            )
+
+        path = os.path.join(temp_dir, f"calendar_{dt.year}-{dt.month:02d}-{dt.day:02d}.jpg")
+        img.save(path)
+        print("Generated:", path)
+
+
+if __name__ == "__main__":
+    main()