Added DateOutOfRangeError

This commit is contained in:
Gourav Kumar 2022-02-26 20:41:21 +05:30
parent 7cac5cc307
commit c9bfa485f5

View File

@ -1,6 +1,21 @@
import datetime
from typing import Literal
class DateNotFoundError(Exception): class DateNotFoundError(Exception):
"""Exception to be raised when date is not found""" """Exception to be raised when date is not found"""
def __init__(self, message, date): def __init__(self, message, date):
message = f"{message}: {date}" message = f"{message}: {date}"
super().__init__(message) super().__init__(message)
class DateOutOfRangeError(Exception):
"""Exception to be raised when provided date is outside the range of dates in the time series"""
def __init__(self, date: datetime.datetime, type: Literal['min', 'max']) -> None:
if type == 'min':
message = f"Provided date {date} is before the first date in the TimeSeries"
if type == 'max':
message = f"Provided date {date} is after the last date in the TimeSeries"
super().__init__(message)