1 """Various utilities for handling dates."""
2 from datetime import datetime, timedelta
3 from plomtask.exceptions import BadFormatException
5 DATE_FORMAT = '%Y-%m-%d'
6 MIN_RANGE_DATE = '2024-01-01'
7 MAX_RANGE_DATE = '2030-12-31'
10 def valid_date(date_str: str) -> str:
11 """Validate date against DATE_FORMAT or 'today'/'yesterday'/'tomorrow.
13 In any case, returns in DATE_FORMAT.
15 if date_str == 'today':
16 date_str = todays_date()
17 elif date_str == 'yesterday':
18 date = datetime.now() - timedelta(days=1)
19 date_str = date.strftime(DATE_FORMAT)
20 elif date_str == 'tomorrow':
21 date = datetime.now() + timedelta(days=1)
22 date_str = date.strftime(DATE_FORMAT)
24 dt = datetime.strptime(date_str, DATE_FORMAT)
25 except (ValueError, TypeError) as e:
26 msg = f'Given date of wrong format: {date_str}'
27 raise BadFormatException(msg) from e
28 return dt.strftime(DATE_FORMAT)
31 def todays_date() -> str:
32 """Return current date in DATE_FORMAT."""
33 return datetime.now().strftime(DATE_FORMAT)