ics-query 0.1.1a0__py3-none-any.whl → 0.3.1b0__py3-none-any.whl

ics_query/query.py

@@ -0,0 +1,71 @@
+# ics-query
+# Copyright (C) 2024 Nicco Kunzmann
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <https://www.gnu.org/licenses/>.
+"""This is an adaptation of the CalendarQuery."""
+
+from __future__ import annotations
+
+import datetime
+import zoneinfo
+from typing import TYPE_CHECKING
+
+import x_wr_timezone
+from recurring_ical_events import CalendarQuery, Occurrence
+
+if TYPE_CHECKING:
+    from collections.abc import Sequence
+
+    from icalendar import Calendar
+
+
+class Query(CalendarQuery):
+    def __init__(self, calendar: Calendar, timezone: str, components: Sequence[str]):
+        """Create a new query."""
+        super().__init__(
+            x_wr_timezone.to_standard(calendar),
+            components=components,
+            skip_bad_series=True,
+        )
+        self.timezone = zoneinfo.ZoneInfo(timezone) if timezone else None
+
+    def with_timezone(self, dt: datetime.date | datetime.datetime):
+        """Add the timezone."""
+        if self.timezone is None:
+            return dt
+        if not isinstance(dt, datetime.datetime):
+            return datetime.datetime(
+                year=dt.year, month=dt.month, day=dt.day, tzinfo=self.timezone
+            )
+        if dt.tzinfo is None:
+            return dt.replace(tzinfo=self.timezone)
+        return dt.astimezone(self.timezone)
+
+    def _occurrences_between(
+        self,
+        start: datetime.date | datetime.datetime,
+        end: datetime.date | datetime.datetime,
+    ) -> list[Occurrence]:
+        """Override to adapt timezones."""
+        result = []
+        for occurrence in super()._occurrences_between(
+            self.with_timezone(start), self.with_timezone(end)
+        ):
+            occurrence.start = self.with_timezone(occurrence.start)
+            occurrence.end = self.with_timezone(occurrence.end)
+            result.append(occurrence)
+        return result
+
+
+__all__ = ["Query"]

ics_query/tests/conftest.py

@@ -1,3 +1,18 @@
+# ics-query
+# Copyright (C) 2024 Nicco Kunzmann
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <https://www.gnu.org/licenses/>.
 """Configure the tests."""
 
 from __future__ import annotations

ics_query/tests/runs/all --tz Singapore one-event.ics -.run

@@ -0,0 +1,9 @@
+BEGIN:VEVENT
+SUMMARY:test1
+DTSTART;TZID=Singapore:20190304T150000
+DTEND;TZID=Singapore:20190304T153000
+DTSTAMP:20190303T111937
+UID:UYDQSG9TH4DE0WM3QFL2J
+CREATED:20190303T111937
+LAST-MODIFIED:20190303T111937
+END:VEVENT

ics_query/tests/runs/all three-events.ics -.run

@@ -0,0 +1,33 @@
+BEGIN:VEVENT
+SUMMARY:test4
+DTSTART;TZID=Europe/Berlin:20190304T000000
+DTEND;TZID=Europe/Berlin:20190304T010000
+DTSTAMP:20190303T111937
+UID:UYDQSG9TH4DE0WM3QFL2J
+CLASS:PUBLIC
+CREATED:20190303T111937
+LAST-MODIFIED:20190303T111937
+STATUS:CONFIRMED
+END:VEVENT
+BEGIN:VEVENT
+SUMMARY:test4
+DTSTART;TZID=Europe/Berlin:20190307T000000
+DTEND;TZID=Europe/Berlin:20190307T010000
+DTSTAMP:20190303T111937
+UID:UYDQSG9TH4DE0WM3QFL2J
+CLASS:PUBLIC
+CREATED:20190303T111937
+LAST-MODIFIED:20190303T111937
+STATUS:CONFIRMED
+END:VEVENT
+BEGIN:VEVENT
+SUMMARY:test4
+DTSTART;TZID=Europe/Berlin:20190310T000000
+DTEND;TZID=Europe/Berlin:20190310T010000
+DTSTAMP:20190303T111937
+UID:UYDQSG9TH4DE0WM3QFL2J
+CLASS:PUBLIC
+CREATED:20190303T111937
+LAST-MODIFIED:20190303T111937
+STATUS:CONFIRMED
+END:VEVENT

ics_query/tests/runs/at 2024-08-20 Berlin-Los-Angeles.ics -.run

@@ -0,0 +1,23 @@
+BEGIN:VEVENT
+SUMMARY:6:00-7:00 Europe/Berlin 20th August
+DTSTART;TZID=Europe/Berlin:20240820T060000
+DTEND;TZID=Europe/Berlin:20240820T070000
+DTSTAMP:20240823T130444Z
+UID:b27ea261-f23d-4e03-a7ac-f8cb0d00f07f
+CREATED:20240823T120639Z
+LAST-MODIFIED:20240823T130444Z
+TRANSP:OPAQUE
+X-MOZ-GENERATION:3
+END:VEVENT
+BEGIN:VEVENT
+SUMMARY:6:00-7:00 Amerika/Los Angeles 20th August
+DTSTART;TZID=America/Los_Angeles:20240820T060000
+DTEND;TZID=America/Los_Angeles:20240820T070000
+DTSTAMP:20240823T130711Z
+UID:6d7ff7f3-4bc4-4d89-afa0-771bd690518a
+SEQUENCE:1
+CREATED:20240823T120639Z
+LAST-MODIFIED:20240823T130711Z
+TRANSP:OPAQUE
+X-MOZ-GENERATION:4
+END:VEVENT

ics_query/tests/runs/between 20240823 4d recurring-work-events.ics -.run

@@ -0,0 +1,24 @@
+BEGIN:VEVENT
+SUMMARY:Work
+DTSTART;TZID=Europe/Berlin:20240823T090000
+DTEND;TZID=Europe/Berlin:20240823T170000
+DTSTAMP:20240823T082915Z
+UID:6b85b60c-eb1a-4338-9ece-33541b95bf17
+SEQUENCE:1
+CREATED:20240823T082829Z
+LAST-MODIFIED:20240823T082915Z
+TRANSP:OPAQUE
+X-MOZ-GENERATION:2
+END:VEVENT
+BEGIN:VEVENT
+SUMMARY:Work
+DTSTART;TZID=Europe/Berlin:20240826T090000
+DTEND;TZID=Europe/Berlin:20240826T170000
+DTSTAMP:20240823T082915Z
+UID:6b85b60c-eb1a-4338-9ece-33541b95bf17
+SEQUENCE:1
+CREATED:20240823T082829Z
+LAST-MODIFIED:20240823T082915Z
+TRANSP:OPAQUE
+X-MOZ-GENERATION:2
+END:VEVENT

ics_query/tests/runs/calendars/Berlin-Los-Angeles.ics

@@ -0,0 +1,381 @@
+BEGIN:VCALENDAR
+PRODID:-//Mozilla.org/NONSGML Mozilla Calendar V1.1//EN
+VERSION:2.0
+BEGIN:VTIMEZONE
+TZID:America/Los_Angeles
+X-TZINFO:America/Los_Angeles[2024a]
+BEGIN:STANDARD
+TZOFFSETTO:-080000
+TZOFFSETFROM:-075258
+TZNAME:America/Los_Angeles(STD)
+DTSTART:18831118T120702
+RDATE:18831118T120702
+END:STANDARD
+BEGIN:DAYLIGHT
+TZOFFSETTO:-070000
+TZOFFSETFROM:-080000
+TZNAME:America/Los_Angeles(DST)
+DTSTART:19180331T020000
+RRULE:FREQ=YEARLY;BYMONTH=3;BYDAY=-1SU;UNTIL=19190330T020000
+END:DAYLIGHT
+BEGIN:STANDARD
+TZOFFSETTO:-080000
+TZOFFSETFROM:-070000
+TZNAME:America/Los_Angeles(STD)
+DTSTART:19181027T020000
+RRULE:FREQ=YEARLY;BYMONTH=10;BYDAY=-1SU;UNTIL=19191026T020000
+END:STANDARD
+BEGIN:DAYLIGHT
+TZOFFSETTO:-070000
+TZOFFSETFROM:-080000
+TZNAME:America/Los_Angeles(DST)
+DTSTART:19420209T020000
+RDATE:19420209T020000
+END:DAYLIGHT
+BEGIN:STANDARD
+TZOFFSETTO:-080000
+TZOFFSETFROM:-070000
+TZNAME:America/Los_Angeles(STD)
+DTSTART:19450930T020000
+RDATE:19450930T020000
+END:STANDARD
+BEGIN:DAYLIGHT
+TZOFFSETTO:-070000
+TZOFFSETFROM:-080000
+TZNAME:America/Los_Angeles(DST)
+DTSTART:19480314T020100
+RDATE:19480314T020100
+END:DAYLIGHT
+BEGIN:STANDARD
+TZOFFSETTO:-080000
+TZOFFSETFROM:-070000
+TZNAME:America/Los_Angeles(STD)
+DTSTART:19490101T020000
+RDATE:19490101T020000
+END:STANDARD
+BEGIN:STANDARD
+TZOFFSETTO:-080000
+TZOFFSETFROM:-070000
+TZNAME:America/Los_Angeles(STD)
+DTSTART:19500924T020000
+RRULE:FREQ=YEARLY;BYMONTH=9;BYDAY=-1SU;UNTIL=19610924T020000
+END:STANDARD
+BEGIN:DAYLIGHT
+TZOFFSETTO:-070000
+TZOFFSETFROM:-080000
+TZNAME:America/Los_Angeles(DST)
+DTSTART:19500430T010000
+RRULE:FREQ=YEARLY;BYMONTH=4;BYDAY=-1SU;UNTIL=19660424T010000
+END:DAYLIGHT
+BEGIN:DAYLIGHT
+TZOFFSETTO:-070000
+TZOFFSETFROM:-080000
+TZNAME:America/Los_Angeles(DST)
+DTSTART:19670430T020000
+RRULE:FREQ=YEARLY;BYMONTH=4;BYDAY=-1SU;UNTIL=19730429T020000
+END:DAYLIGHT
+BEGIN:DAYLIGHT
+TZOFFSETTO:-070000
+TZOFFSETFROM:-080000
+TZNAME:America/Los_Angeles(DST)
+DTSTART:19740106T020000
+RDATE:19740106T020000
+END:DAYLIGHT
+BEGIN:DAYLIGHT
+TZOFFSETTO:-070000
+TZOFFSETFROM:-080000
+TZNAME:America/Los_Angeles(DST)
+DTSTART:19750223T020000
+RDATE:19750223T020000
+END:DAYLIGHT
+BEGIN:DAYLIGHT
+TZOFFSETTO:-070000
+TZOFFSETFROM:-080000
+TZNAME:America/Los_Angeles(DST)
+DTSTART:19760425T020000
+RRULE:FREQ=YEARLY;BYMONTH=4;BYDAY=-1SU;UNTIL=19860427T020000
+END:DAYLIGHT
+BEGIN:DAYLIGHT
+TZOFFSETTO:-070000
+TZOFFSETFROM:-080000
+TZNAME:America/Los_Angeles(DST)
+DTSTART:19870405T020000
+RRULE:FREQ=YEARLY;BYMONTH=4;BYDAY=1SU;UNTIL=20060402T020000
+END:DAYLIGHT
+BEGIN:STANDARD
+TZOFFSETTO:-080000
+TZOFFSETFROM:-070000
+TZNAME:America/Los_Angeles(STD)
+DTSTART:19621028T020000
+RRULE:FREQ=YEARLY;BYMONTH=10;BYDAY=-1SU;UNTIL=20061029T020000
+END:STANDARD
+BEGIN:DAYLIGHT
+TZOFFSETTO:-070000
+TZOFFSETFROM:-080000
+TZNAME:America/Los_Angeles(DST)
+DTSTART:20070311T020000
+RDATE:20070311T020000
+END:DAYLIGHT
+BEGIN:STANDARD
+TZOFFSETTO:-080000
+TZOFFSETFROM:-070000
+TZNAME:America/Los_Angeles(STD)
+DTSTART:20071104T020000
+RDATE:20071104T020000
+END:STANDARD
+BEGIN:DAYLIGHT
+TZOFFSETTO:-070000
+TZOFFSETFROM:-080000
+TZNAME:(DST)
+DTSTART:20080309T020000
+RRULE:FREQ=YEARLY;BYMONTH=3;BYDAY=2SU
+END:DAYLIGHT
+BEGIN:STANDARD
+TZOFFSETTO:-080000
+TZOFFSETFROM:-070000
+TZNAME:(STD)
+DTSTART:20081102T020000
+RRULE:FREQ=YEARLY;BYMONTH=11;BYDAY=1SU
+END:STANDARD
+END:VTIMEZONE
+BEGIN:VEVENT
+CREATED:20240823T120639Z
+LAST-MODIFIED:20240823T130425Z
+DTSTAMP:20240823T130425Z
+UID:97e57ce6-e446-4213-ab68-bf22b8a820a7
+SUMMARY:6:00-7:00 Amerika/Los Angeles 19th August
+DTSTART;TZID=America/Los_Angeles:20240819T060000
+DTEND;TZID=America/Los_Angeles:20240819T070000
+TRANSP:OPAQUE
+X-MOZ-GENERATION:4
+SEQUENCE:1
+END:VEVENT
+BEGIN:VTIMEZONE
+TZID:Europe/Berlin
+X-TZINFO:Europe/Berlin[2024a]
+BEGIN:STANDARD
+TZOFFSETTO:+010000
+TZOFFSETFROM:+005328
+TZNAME:Europe/Berlin(STD)
+DTSTART:18930401T000000
+RDATE:18930401T000000
+END:STANDARD
+BEGIN:DAYLIGHT
+TZOFFSETTO:+020000
+TZOFFSETFROM:+010000
+TZNAME:Europe/Berlin(DST)
+DTSTART:19160430T230000
+RDATE:19160430T230000
+END:DAYLIGHT
+BEGIN:STANDARD
+TZOFFSETTO:+010000
+TZOFFSETFROM:+020000
+TZNAME:Europe/Berlin(STD)
+DTSTART:19161001T010000
+RDATE:19161001T010000
+END:STANDARD
+BEGIN:DAYLIGHT
+TZOFFSETTO:+020000
+TZOFFSETFROM:+010000
+TZNAME:Europe/Berlin(DST)
+DTSTART:19170416T020000
+RRULE:FREQ=YEARLY;BYMONTH=4;BYDAY=3MO;UNTIL=19180415T020000
+END:DAYLIGHT
+BEGIN:STANDARD
+TZOFFSETTO:+010000
+TZOFFSETFROM:+020000
+TZNAME:Europe/Berlin(STD)
+DTSTART:19170917T030000
+RRULE:FREQ=YEARLY;BYMONTH=9;BYDAY=3MO;UNTIL=19180916T030000
+END:STANDARD
+BEGIN:DAYLIGHT
+TZOFFSETTO:+020000
+TZOFFSETFROM:+010000
+TZNAME:Europe/Berlin(DST)
+DTSTART:19400401T020000
+RDATE:19400401T020000
+END:DAYLIGHT
+BEGIN:STANDARD
+TZOFFSETTO:+010000
+TZOFFSETFROM:+020000
+TZNAME:Europe/Berlin(STD)
+DTSTART:19421102T030000
+RDATE:19421102T030000
+END:STANDARD
+BEGIN:DAYLIGHT
+TZOFFSETTO:+020000
+TZOFFSETFROM:+010000
+TZNAME:Europe/Berlin(DST)
+DTSTART:19430329T020000
+RDATE:19430329T020000
+END:DAYLIGHT
+BEGIN:DAYLIGHT
+TZOFFSETTO:+020000
+TZOFFSETFROM:+010000
+TZNAME:Europe/Berlin(DST)
+DTSTART:19440403T020000
+RRULE:FREQ=YEARLY;BYMONTH=4;BYDAY=1MO;UNTIL=19450402T020000
+END:DAYLIGHT
+BEGIN:DAYLIGHT
+TZOFFSETTO:+030000
+TZOFFSETFROM:+020000
+TZNAME:Europe/Berlin(DST)
+DTSTART:19450524T020000
+RDATE:19450524T020000
+END:DAYLIGHT
+BEGIN:STANDARD
+TZOFFSETTO:+010000
+TZOFFSETFROM:+020000
+TZNAME:Europe/Berlin(STD)
+DTSTART:19431004T030000
+RRULE:FREQ=YEARLY;BYMONTH=10;BYDAY=1MO;UNTIL=19441002T030000
+END:STANDARD
+BEGIN:DAYLIGHT
+TZOFFSETTO:+020000
+TZOFFSETFROM:+030000
+TZNAME:Europe/Berlin(DST)
+DTSTART:19450924T030000
+RDATE:19450924T030000
+END:DAYLIGHT
+BEGIN:STANDARD
+TZOFFSETTO:+010000
+TZOFFSETFROM:+020000
+TZNAME:Europe/Berlin(STD)
+DTSTART:19451118T030000
+RDATE:19451118T030000
+END:STANDARD
+BEGIN:DAYLIGHT
+TZOFFSETTO:+020000
+TZOFFSETFROM:+010000
+TZNAME:Europe/Berlin(DST)
+DTSTART:19460414T020000
+RDATE:19460414T020000
+END:DAYLIGHT
+BEGIN:DAYLIGHT
+TZOFFSETTO:+020000
+TZOFFSETFROM:+010000
+TZNAME:Europe/Berlin(DST)
+DTSTART:19470406T030000
+RDATE:19470406T030000
+END:DAYLIGHT
+BEGIN:DAYLIGHT
+TZOFFSETTO:+030000
+TZOFFSETFROM:+020000
+TZNAME:Europe/Berlin(DST)
+DTSTART:19470511T030000
+RDATE:19470511T030000
+END:DAYLIGHT
+BEGIN:STANDARD
+TZOFFSETTO:+010000
+TZOFFSETFROM:+020000
+TZNAME:Europe/Berlin(STD)
+DTSTART:19461007T030000
+RDATE:19461007T030000
+END:STANDARD
+BEGIN:DAYLIGHT
+TZOFFSETTO:+020000
+TZOFFSETFROM:+030000
+TZNAME:Europe/Berlin(DST)
+DTSTART:19470629T030000
+RDATE:19470629T030000
+END:DAYLIGHT
+BEGIN:DAYLIGHT
+TZOFFSETTO:+020000
+TZOFFSETFROM:+010000
+TZNAME:Europe/Berlin(DST)
+DTSTART:19480418T020000
+RDATE:19480418T020000
+END:DAYLIGHT
+BEGIN:DAYLIGHT
+TZOFFSETTO:+020000
+TZOFFSETFROM:+010000
+TZNAME:Europe/Berlin(DST)
+DTSTART:19490410T020000
+RDATE:19490410T020000
+END:DAYLIGHT
+BEGIN:STANDARD
+TZOFFSETTO:+010000
+TZOFFSETFROM:+020000
+TZNAME:Europe/Berlin(STD)
+DTSTART:19471005T030000
+RRULE:FREQ=YEARLY;BYMONTH=10;BYDAY=1SU;UNTIL=19491002T030000
+END:STANDARD
+BEGIN:DAYLIGHT
+TZOFFSETTO:+020000
+TZOFFSETFROM:+010000
+TZNAME:Europe/Berlin(DST)
+DTSTART:19800406T020000
+RDATE:19800406T020000
+END:DAYLIGHT
+BEGIN:STANDARD
+TZOFFSETTO:+010000
+TZOFFSETFROM:+020000
+TZNAME:Europe/Berlin(STD)
+DTSTART:19800928T030000
+RRULE:FREQ=YEARLY;BYMONTH=9;BYDAY=-1SU;UNTIL=19950924T030000
+END:STANDARD
+BEGIN:DAYLIGHT
+TZOFFSETTO:+020000
+TZOFFSETFROM:+010000
+TZNAME:Europe/Berlin(DST)
+DTSTART:19810329T020000
+RRULE:FREQ=YEARLY;BYMONTH=3;BYDAY=-1SU;UNTIL=19960331T020000
+END:DAYLIGHT
+BEGIN:STANDARD
+TZOFFSETTO:+010000
+TZOFFSETFROM:+020000
+TZNAME:Europe/Berlin(STD)
+DTSTART:19961027T030000
+RDATE:19961027T030000
+END:STANDARD
+BEGIN:DAYLIGHT
+TZOFFSETTO:+020000
+TZOFFSETFROM:+010000
+TZNAME:(DST)
+DTSTART:19970330T020000
+RRULE:FREQ=YEARLY;BYMONTH=3;BYDAY=-1SU
+END:DAYLIGHT
+BEGIN:STANDARD
+TZOFFSETTO:+010000
+TZOFFSETFROM:+020000
+TZNAME:(STD)
+DTSTART:19971026T030000
+RRULE:FREQ=YEARLY;BYMONTH=10;BYDAY=-1SU
+END:STANDARD
+END:VTIMEZONE
+BEGIN:VEVENT
+CREATED:20240823T120639Z
+LAST-MODIFIED:20240823T130444Z
+DTSTAMP:20240823T130444Z
+UID:b27ea261-f23d-4e03-a7ac-f8cb0d00f07f
+SUMMARY:6:00-7:00 Europe/Berlin 20th August
+DTSTART;TZID=Europe/Berlin:20240820T060000
+DTEND;TZID=Europe/Berlin:20240820T070000
+TRANSP:OPAQUE
+X-MOZ-GENERATION:3
+END:VEVENT
+BEGIN:VEVENT
+CREATED:20240823T120639Z
+LAST-MODIFIED:20240823T130711Z
+DTSTAMP:20240823T130711Z
+UID:6d7ff7f3-4bc4-4d89-afa0-771bd690518a
+SUMMARY:6:00-7:00 Amerika/Los Angeles 20th August
+DTSTART;TZID=America/Los_Angeles:20240820T060000
+DTEND;TZID=America/Los_Angeles:20240820T070000
+TRANSP:OPAQUE
+X-MOZ-GENERATION:4
+SEQUENCE:1
+END:VEVENT
+BEGIN:VEVENT
+CREATED:20240823T120639Z
+LAST-MODIFIED:20240823T130729Z
+DTSTAMP:20240823T130729Z
+UID:8ee9f394-9a01-41e5-941b-fbe08b317210
+SUMMARY:21:00-22:00 Amerika/Los Angeles 19th August
+DTSTART;TZID=America/Los_Angeles:20240819T210000
+DTEND;TZID=America/Los_Angeles:20240819T220000
+TRANSP:OPAQUE
+X-MOZ-GENERATION:6
+SEQUENCE:2
+END:VEVENT
+END:VCALENDAR

ics_query/tests/runs/calendars/empty-calendar.ics

@@ -0,0 +1,7 @@
+BEGIN:VCALENDAR
+VERSION:2.0
+PRODID:-//SabreDAV//SabreDAV//EN
+CALSCALE:GREGORIAN
+X-WR-CALNAME:empty-calendar
+X-APPLE-CALENDAR-COLOR:#e78074
+END:VCALENDAR