oncall-engine/engine/apps/schedules/ical_events/proxy/ical_proxy.py
Joey Orlando 9dde1805aa
add mypy static type checker to backend codebase (#2151)
# What this PR does

- Adds [`mypy` static type checking](https://mypy-lang.org/) to our CI
pipeline. Currently there is still a **ton** of errors being returned by
the tool, as we'll need to fix pre-existing errors. I think we can
slowly chip away at these errors in small PRs, doing them all in one
large PR is likely very risky.
- Also, this PR starts chipping away at one of the main type errors that
we have which is accessing the `datetime` class (from the `datetime`
library) or `timedelta` function on the `django.utils.timezone` module.
Basically we should be instead accessing these two objects from the
native `datetime` module. This makes sense because the [`__all__`
attribute](https://github.com/django/django/blob/main/django/utils/timezone.py#L14-L30)
in `django.utils.timezone` does not re-export `datetime` or `timedelta`.
- splits `engine` dependencies out into `requirements.txt` and
`requirements-dev.txt`

## Checklist

- [ ] Unit, integration, and e2e (if applicable) tests updated (N/A)
- [ ] Documentation added (or `pr:no public docs` PR label added if not
required) (N/A)
- [ ] `CHANGELOG.md` updated (or `pr:no changelog` PR label added if not
required) (N/A)
2023-06-12 12:50:33 -04:00

26 lines
1 KiB
Python

from abc import ABC, abstractmethod
from datetime import datetime
from typing import List, Tuple
from icalendar import Calendar, Event
class IcalService(ABC):
@abstractmethod
def get_events_from_ical_between(self, calendar: Calendar, start_date: datetime, end_date: datetime) -> List[Event]:
raise NotImplementedError
@abstractmethod
def get_start_and_end_with_respect_to_event_type(self, event: Event) -> Tuple[datetime, datetime]:
raise NotImplementedError
class IcalProxy(IcalService):
def __init__(self, ical_adapter: IcalService):
self.ical_adapter = ical_adapter
def get_events_from_ical_between(self, calendar: Calendar, start_date: datetime, end_date: datetime) -> List[Event]:
return self.ical_adapter.get_events_from_ical_between(calendar, start_date, end_date)
def get_start_and_end_with_respect_to_event_type(self, event: Event) -> Tuple[datetime, datetime]:
return self.ical_adapter.get_start_and_end_with_respect_to_event_type(event)