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

ics_query/__init__.py

@@ -1,11 +1,27 @@
-from .cli import main
+# 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/>.
+from .cli import cli, main
 from .version import __version__, __version_tuple__, version, version_tuple
 
 __all__ = [
-    "main",
+    "cli",
     "app",
     "__version__",
     "version",
     "__version_tuple__",
     "version_tuple",
+    "main",
 ]

ics_query/__main__.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/>.
 import sys
 
 from .cli import main

ics_query/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.1.1a0'
-__version_tuple__ = version_tuple = (0, 1, 1)
+__version__ = version = '0.3.2b0'
+__version_tuple__ = version_tuple = (0, 3, 2)

ics_query/cli.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/>.
 """The command line interface."""
 
 from __future__ import annotations
@@ -6,24 +21,45 @@ import functools
 import os  # noqa: TCH003
 import sys
 import typing as t
+import zoneinfo
 
 import click
-import recurring_ical_events
 from icalendar.cal import Calendar, Component
-from recurring_ical_events import CalendarQuery
+from tzlocal import get_localzone_name
 
 from . import parse
-from .version import __version__
+from .query import Query
+from .version import cli_version
 
 if t.TYPE_CHECKING:
     from io import FileIO
 
+    import recurring_ical_events
     from icalendar.cal import Component
 
-    from .parse import DateArgument
+    from .parse import Date
 
 print = functools.partial(print, file=sys.stderr)  # noqa: A001
 
+ENV_PREFIX = "ICS_QUERY"
+LICENSE = """
+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/>.
+"""
+
 
 class ComponentsResult:
     """Output interface for components."""
@@ -36,6 +72,11 @@ class ComponentsResult:
         """Return a component."""
         self._file.write(component.to_ical())
 
+    def add_components(self, components: t.Iterable[Component]):
+        """Add all components."""
+        for component in components:
+            self.add_component(component)
+
 
 class ComponentsResultArgument(click.File):
     """Argument for the result."""
@@ -52,15 +93,37 @@ class ComponentsResultArgument(click.File):
 
 
 class JoinedCalendars:
-    def __init__(self, calendars: list[Calendar]):
+    def __init__(
+        self, calendars: list[Calendar], timezone: str, components: t.Sequence[str]
+    ):
         """Join multiple calendars."""
-        self.queries = [recurring_ical_events.of(calendar) for calendar in calendars]
+        self.queries = [
+            Query(calendar, timezone, components=components) for calendar in calendars
+        ]
 
     def at(self, dt: tuple[int]) -> t.Generator[Component]:
         """Return the components."""
         for query in self.queries:
             yield from query.at(dt)
 
+    def first(self) -> t.Generator[Component]:
+        """Return the first events of all calendars."""
+        for query in self.queries:
+            for component in query.all():
+                yield component
+                break
+
+    def all(self) -> t.Generator[Component]:
+        """Return the first events of all calendars."""
+        for query in self.queries:
+            yield from query.all()
+
+    def between(
+        self, start: parse.Date, end: parse.DateAndDelta
+    ) -> t.Generator[Component]:
+        for query in self.queries:
+            yield from query.between(start, end)
+
 
 class CalendarQueryInputArgument(click.File):
     """Argument for the result."""
@@ -74,16 +137,106 @@ class CalendarQueryInputArgument(click.File):
         """Return a CalendarQuery."""
         file = super().convert(value, param, ctx)
         calendars = Calendar.from_ical(file.read(), multiple=True)
-        return JoinedCalendars(calendars)
+        components = ctx.params.get("component", ("VEVENT", "VTODO", "VJOURNAL"))
+        timezone = ctx.params.get("tz", "")
+        return JoinedCalendars(calendars, timezone, components)
+
+
+opt_components = click.option(
+    "--component",
+    "-c",
+    multiple=True,
+    envvar=ENV_PREFIX + "_COMPONENT",
+    help=(
+        "Select the components which can be returned. "
+        "By default all supported components can be in the result. "
+        "Possible values are: VEVENT, VTODO, VJOURNAL. "
+    ),
+)
+
+opt_timezone = click.option(
+    "--tz",
+    envvar=ENV_PREFIX + "_TZ",
+    help=("Set the timezone. See also --available-timezones"),
+)
+
+
+def arg_calendar(func):
+    """Decorator for a calendar argument with all used options."""
+    arg = click.argument("calendar", type=CalendarQueryInputArgument("rb"))
+
+    @functools.wraps(func)
+    def wrapper(*args, component=(), tz="", **kw):  # noqa: ARG001
+        """Remove some parameters."""
+        return func(*args, **kw)
+
+    return opt_timezone(opt_components(arg(wrapper)))
+
+
+arg_output = click.argument("output", type=ComponentsResultArgument("wb"), default="-")
+# Option with many values and list as result
+# see https://click.palletsprojects.com/en/latest/options/#multiple-options
+
+
+def opt_available_timezones(*param_decls: str, **kwargs: t.Any) -> t.Callable:
+    """List available timezones.
+
+    This is copied from the --help option.
+
+    Commonly used timezone names are added first.
+    """
 
+    def callback(ctx: click.Context, param: click.Parameter, value: bool) -> None:  # noqa: FBT001, ARG001
+        if not value or ctx.resilient_parsing:
+            return
 
-arg_calendar = click.argument("calendar", type=CalendarQueryInputArgument("rb"))
-arg_output = click.argument("output", type=ComponentsResultArgument("wb"))
+        click.echo("localtime")  # special local time handle
+        click.echo(get_localzone_name())
+        click.echo("UTC")
+        all_zones = zoneinfo.available_timezones()
+        for zone in sorted(all_zones, key=str.lower):
+            click.echo(zone)
+        ctx.exit()
+
+    if not param_decls:
+        param_decls = ("--available-timezones",)
+
+    kwargs.setdefault("is_flag", True)
+    kwargs.setdefault("expose_value", False)
+    kwargs.setdefault("is_eager", True)
+    kwargs.setdefault("help", "List all available timezones and exit.")
+    kwargs["callback"] = callback
+    return click.option(*param_decls, **kwargs)
+
+
+def opt_license(*param_decls: str, **kwargs: t.Any) -> t.Callable:
+    """List available timezones.
+
+    This is copied from the --help option.
+    """
+
+    def callback(ctx: click.Context, param: click.Parameter, value: bool) -> None:  # noqa: FBT001, ARG001
+        if not value or ctx.resilient_parsing:
+            return
+        click.echo(LICENSE)
+        ctx.exit()
+
+    if not param_decls:
+        param_decls = ("--license",)
+
+    kwargs.setdefault("is_flag", True)
+    kwargs.setdefault("expose_value", False)
+    kwargs.setdefault("is_eager", True)
+    kwargs.setdefault("help", "Show the license and exit.")
+    kwargs["callback"] = callback
+    return click.option(*param_decls, **kwargs)
 
 
 @click.group()
-@click.version_option(__version__)
-def main():
+@click.version_option(cli_version)
+@opt_available_timezones()
+@opt_license()
+def cli():
     """Find out what happens in ICS calendar files.
 
     ics-query can query and filter RFC 5545 compatible .ics files.
@@ -107,8 +260,8 @@ def main():
     If OUTPUT is "-", then the standard output is used.
 
     \b
-    Notes on Calculation
-    --------------------
+    Calculation
+    -----------
 
     An event can be very long. If you request smaller time spans or a time as
     exact as a second, the event will still occur within this time span if it
@@ -121,20 +274,72 @@ def main():
     The START is INCLUSIVE, then END is EXCLUSIVE.
 
     \b
-    Notes on Timezones
-    ------------------
+    Timezones
+    ---------
+
+    We have several timezones available to choose from.
+    While the calendar entries might use their own timezone definitions,
+    the timezone parameters of ics-query use the timezone definitions of
+    Python's tzdata package.
+
+    You can list all timezones available with this command:
+
+    \b
+        ics-query --available-timezones
+
+    By default the local time of the components is assumed.
+    In this example, two events happen at 6am, one in Berlin and one in Los Angeles.
+    Both are hours apart though.
+
+    \b
+        $ ics-query at 2024-08-20 Berlin-Los-Angeles.ics - | grep -E 'DTSTART|SUMMARY'
+        SUMMARY:6:00-7:00 Europe/Berlin 20th August
+        DTSTART;TZID=Europe/Berlin:20240820T060000
+        SUMMARY:6:00-7:00 Amerika/Los Angeles 20th August
+        DTSTART;TZID=America/Los_Angeles:20240820T060000
+
+    If you however wish to get all events in a certain timezone, use the --tz parameter.
+    In this example, the event that happens at the 19th of August at 21:00 in
+    Los Angeles is actually happening on the 20th in local Berlin time.
+
+    \b
+        $ ics-query at --tz=Europe/Berlin 2024-08-20 Berlin-Los-Angeles.ics - \\
+            | grep -E 'DTSTART|SUMMARY'
+        SUMMARY:6:00-7:00 Europe/Berlin 20th August
+        DTSTART;TZID=Europe/Berlin:20240820T060000
+        SUMMARY:6:00-7:00 Amerika/Los Angeles 20th August
+        DTSTART;TZID=Europe/Berlin:20240820T150000
+        SUMMARY:21:00-22:00 Amerika/Los Angeles 19th August
+        DTSTART;TZID=Europe/Berlin:20240820T060000
+
+    If you wish to get events in your local time, use --tz localtime.
+    If you like UTC, use --tz UTC.
+
+    You can also set the environment variable ICS_QUERY_TZ to the timezone instead of
+    passing --tz.
+
+    \b
+    Components
+    ----------
+
+    We support different types of recurring components: VEVENT, VTODO, VJOURNAL.
+    You can specify which can be in the result using the --component parameter.
+
+    You can also set the environment variable ICS_QUERY_COMPONENT to the timezone
+    instead of passing --component.
+
     """  # noqa: D301
 
 
 pass_datetime = click.make_pass_decorator(parse.to_time)
 
 
-@main.command()
+@cli.command()
 @click.argument("date", type=parse.to_time)
 @arg_calendar
 @arg_output
-def at(calendar: CalendarQuery, output: ComponentsResult, date: DateArgument):
-    """Occurrences at a certain dates.
+def at(calendar: JoinedCalendars, output: ComponentsResult, date: Date):
+    """Print occurrences at a certain date or time.
 
     YEAR
 
@@ -156,7 +361,7 @@ def at(calendar: CalendarQuery, output: ComponentsResult, date: DateArgument):
 
     \b
         Formats:
-
+    \b
             YYYY-MM
             YYYY-M
             YYYYMM
@@ -175,7 +380,7 @@ def at(calendar: CalendarQuery, output: ComponentsResult, date: DateArgument):
 
     \b
         Formats:
-
+    \b
             YYYY-MM-DD
             YYYY-M-D
             YYYYMMDD
@@ -253,8 +458,237 @@ def at(calendar: CalendarQuery, output: ComponentsResult, date: DateArgument):
             ics-query at 19900101235959       # 1st January 1990, 23:59:59
             ics-query at `date +%Y%m%d%H%M%S` # now
     """  # noqa: D301
-    for event in calendar.at(date):
-        output.add_component(event)
+    output.add_components(calendar.at(date))
+
+
+@cli.command()
+@arg_calendar
+@arg_output
+def first(calendar: JoinedCalendars, output: ComponentsResult):
+    """Print only the first occurrence.
+
+    This prints the first occurrence in each calendar that is given.
+
+    \b
+    This example prints the first event in calendar.ics:
+    \b
+        ics-query first --component VEVENT calendar.ics -
+
+    """  # noqa: D301
+    output.add_components(calendar.first())
+
+
+@cli.command()
+@arg_calendar
+@arg_output
+def all(calendar: JoinedCalendars, output: ComponentsResult):  # noqa: A001
+    """Print all occurrences in a calendar.
+
+    The result is ordered by the start of the occurrences.
+    If you have multiple calendars, the result will contain
+    the occurrences of the first calendar before those of the second calendar
+    and so on.
+
+    \b
+    This example prints all events in calendar.ics:
+    \b
+        ics-query all --component VEVENT calendar.ics -
+
+    Note that calendars can create hundreds of occurrences and especially
+    contain endless repetitions. Use this with care as the output is
+    potentially enourmous. You can mitigate this by closing the OUTPUT
+    when you have enough e.g. with a head command.
+    """  # noqa: D301
+    output.add_components(calendar.all())
+
+
+@cli.command()
+@click.argument("start", type=parse.to_time)
+@click.argument("end", type=parse.to_time_and_delta)
+@arg_calendar
+@arg_output
+def between(
+    start: parse.Date,
+    end: parse.DateAndDelta,
+    calendar: JoinedCalendars,
+    output: ComponentsResult,
+):
+    """Print occurrences between a START and an END.
+
+    The start is inclusive, the end is exclusive.
+
+    This example returns the events within the next week:
+
+    \b
+        ics-query between --component VEVENT `date +%Y%m%d` +7d calendar.ics -
+
+    This example saves the events from the 1st of May 2024 to the 10th of June in
+    events.ics:
+
+    \b
+        ics-query between --component VEVENT 2024-5-1 2024-6-10 calendar.ics events.ics
+
+    In this example, you can check what is happening on New Years Eve 2025 around
+    midnight:
+
+    \b
+        ics-query between 2025-12-31T21:00 +6h calendar.ics events.ics
+
+    \b
+    Absolute Time
+    -------------
+
+    START must be specified as an absolute time.
+    END can be absolute or relative to START, see Relative Time below.
+
+    Each of the formats specify the earliest time e.g. the start of a day.
+    Thus, if START == END, there are 0 seconds in between and the result is
+    only what happens during that time or starts exactly at that time.
+
+    YEAR
+
+        Specifiy the start of the year.
+
+    \b
+        Formats:
+    \b
+            YYYY
+    \b
+        Examples:
+    \b
+            2024       # start of 2024
+            `date +%Y` # this year
+
+    MONTH
+
+        The start of the month.
+
+    \b
+        Formats:
+    \b
+            YYYY-MM
+            YYYY-M
+            YYYYMM
+    \b
+        Examples:
+    \b
+            2019-10      # October 2019
+            1990-01      # January 1990
+            1990-1       # January 1990
+            199001       # January 1990
+            `date +%Y%m` # this month
+
+    DAY
+
+        The start of the day
+
+    \b
+        Formats:
+    \b
+            YYYY-MM-DD
+            YYYY-M-D
+            YYYYMMDD
+    \b
+        Examples:
+    \b
+            1990-01-01     # 1st January 1990
+            1990-1-1       # 1st January 1990
+            19900101       # 1st January 1990
+            `date +%Y%m%d` # today
+
+    HOUR
+
+        The start of the hour.
+
+    \b
+        Formats:
+    \b
+            YYYY-MM-DD HH
+            YYYY-MM-DDTHH
+            YYYY-M-DTH
+            YYYYMMDDTHH
+            YYYYMMDDHH
+    \b
+        Examples:
+    \b
+            1990-01-01 01    # 1st January 1990, 1am
+            1990-01-01T01    # 1st January 1990, 1am
+            1990-1-1T17      # 1st January 1990, 17:00
+            19900101T23      # 1st January 1990, 23:00
+            1990010123       # 1st January 1990, 23:00
+            `date +%Y%m%d%H` # this hour
+
+    MINUTE
+
+        The start of a minute.
+
+    \b
+        Formats:
+    \b
+            YYYY-MM-DD HH:MM
+            YYYY-MM-DDTHH:MM
+            YYYY-M-DTH:M
+            YYYYMMDDTHHMM
+            YYYYMMDDHHMM
+    \b
+        Examples:
+    \b
+            1990-01-01 10:10   # 1st January 1990, 10:10am
+            1990-01-01T10:10   # 1st January 1990, 10:10am
+            1990-1-1T7:2       # 1st January 1990, 07:02
+            19900101T2359      # 1st January 1990, 23:59
+            199001012359       # 1st January 1990, 23:59
+            `date +%Y%m%d%H%M` # this minute
+
+    SECOND
+
+        A precise time. RFC 5545 calendars are specified to the second.
+        This is the most precise format to specify times.
+
+    \b
+        Formats:
+    \b
+            YYYY-MM-DD HH:MM:SS
+            YYYY-MM-DDTHH:MM:SS
+            YYYY-M-DTH:M:S
+            YYYYMMDDTHHMMSS
+            YYYYMMDDHHMMSS
+    \b
+        Examples:
+    \b
+            1990-01-01 10:10:00  # 1st January 1990, 10:10am
+            1990-01-01T10:10:00  # 1st January 1990, 10:10am
+            1990-1-1T7:2:30      # 1st January 1990, 07:02:30
+            19901231T235959      # 31st December 1990, 23:59:59
+            19900101235959       # 1st January 1990, 23:59:59
+            `date +%Y%m%d%H%M%S` # now
+    \b
+    Relative Time
+    -------------
+
+    The END argument can be a time range.
+    The + at the beginning is optional but makes for a better reading.
+
+    \b
+    Examples:
+    \b
+    Add 10 days to START: +10d
+    Add 24 hours to START: +1d or +24h
+    Add 3 hours to START: +3h
+    Add 30 minutes to START: +30m
+    Add 1000 seconds to START: +1000s
+    \b
+    You can also combine the ranges:
+    Add 1 day and 12 hours to START: +1d12h
+    Add 3 hours and 15 minutes to START: +3h15m
+
+    """  # noqa: D301
+    output.add_components(calendar.between(start, end))
+
+
+def main():
+    """Run the program."""
+    cli(auto_envvar_prefix=ENV_PREFIX)
 
 
-__all__ = ["main"]
+__all__ = ["main", "ENV_PREFIX", "cli"]

ics_query/parse.py

@@ -1,10 +1,28 @@
+# 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/>.
 """Functions for parsing the content."""
 
 from __future__ import annotations
 
+import datetime
 import re
+from typing import Union
 
-DateArgument = tuple[int]
+Date = tuple[int]
+DateAndDelta = Union[Date, datetime.timedelta]
 
 
 class InvalidTimeFormat(ValueError):
@@ -21,14 +39,22 @@ REGEX_TIME = re.compile(
     r"$"
 )
 
+REGEX_TIMEDELTA = re.compile(
+    r"^\+?(?:(?P<days>\d+)d)?"
+    r"(?:(?P<hours>\d+)h)?"
+    r"(?:(?P<minutes>\d+)m)?"
+    r"(?:(?P<seconds>\d+)s)?"
+    r"$"
+)
+
 
-def to_time(dt: str) -> DateArgument:
+def to_time(dt: str) -> Date:
     """Parse the time and date."""
     parsed_dt = REGEX_TIME.match(dt)
     if parsed_dt is None:
         raise InvalidTimeFormat(dt)
 
-    def group(group_name: str) -> tuple[int]:
+    def group(group_name: str) -> Date:
         """Return a group's value."""
         result = parsed_dt.group(group_name)
         while result and result[0] not in "0123456789":
@@ -47,4 +73,15 @@ def to_time(dt: str) -> DateArgument:
     )
 
 
-__all__ = ["to_time", "DateArgument"]
+def to_time_and_delta(dt: str) -> DateAndDelta:
+    """Parse to a absolute time or timedelta."""
+    parsed_td = REGEX_TIMEDELTA.match(dt)
+    if parsed_td is None:
+        return to_time(dt)
+    kw = {k: int(v) for k, v in parsed_td.groupdict().items() if v is not None}
+    if not kw:
+        raise InvalidTimeFormat(dt)
+    return datetime.timedelta(**kw)
+
+
+__all__ = ["to_time", "Date", "to_time_and_delta", "DateAndDelta"]