oncall-engine/engine/apps/alerts/escalation_snapshot/utils.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

33 lines
1.2 KiB
Python

import datetime
import typing
import pytz
from django.utils import timezone
def eta_for_escalation_step_notify_if_time(
from_time: datetime.time, to_time: datetime.time
) -> typing.Optional[datetime.datetime]:
"""
Counts eta for STEP_NOTIFY_IF_TIME
"""
eta: typing.Optional[datetime.datetime] = None
now = timezone.now()
current_date = now.date()
current_time = now.time()
if from_time < to_time:
if from_time > current_time:
eta = datetime.datetime.combine(current_date, from_time).astimezone(pytz.UTC)
elif current_time >= to_time:
eta = datetime.datetime.combine((now + datetime.timedelta(days=1)).date(), from_time).astimezone(pytz.UTC)
elif from_time > to_time:
if from_time > current_time >= to_time:
eta = datetime.datetime.combine(current_date, from_time).astimezone(pytz.UTC)
elif from_time == to_time:
if from_time > current_time:
eta = datetime.datetime.combine(current_date, from_time).astimezone(pytz.UTC)
elif current_time > to_time:
eta = datetime.datetime.combine((now + datetime.timedelta(days=1)).date(), from_time).astimezone(pytz.UTC)
return eta