fix parsing config `start_from` (#82)

This commit is contained in:
Dmitry Belyaev 2022-02-23 13:11:56 +03:00
parent 70a9dac822
commit 839a3ac0c3
Signed by: b4tman
GPG Key ID: 41A00BF15EA7E5F3
1 changed files with 8 additions and 4 deletions

View File

@ -1,4 +1,4 @@
from typing import Dict, Any
from typing import Dict, Any, Union
import yaml
@ -13,6 +13,8 @@ from . import (
CalendarSync
)
ConfigDate = Union[str, datetime.datetime]
def load_config() -> Dict[str, Any]:
with open('config.yml', 'r', encoding='utf-8') as f:
@ -20,11 +22,13 @@ def load_config() -> Dict[str, Any]:
return result
def get_start_date(date_str: str) -> datetime.datetime:
if 'now' == date_str:
def get_start_date(date: ConfigDate) -> datetime.datetime:
if isinstance(date, datetime.datetime):
return date
if 'now' == date:
result = datetime.datetime.utcnow()
else:
result = dateutil.parser.parse(date_str)
result = dateutil.parser.parse(date)
return result