mirror of
https://github.com/beak-insights/felicity-lims.git
synced 2025-02-21 07:22:53 +08:00
72 lines
1.8 KiB
Python
72 lines
1.8 KiB
Python
from datetime import datetime, timedelta
|
|
|
|
from dateutil import parser
|
|
|
|
from felicity.core.config import get_settings
|
|
|
|
settings = get_settings()
|
|
|
|
|
|
def __timenow() -> datetime:
|
|
if settings.TIMEZONE_AWARE:
|
|
now = datetime.now(settings.TIMEZONE)
|
|
else:
|
|
now = datetime.now()
|
|
return now
|
|
|
|
|
|
def timenow_str() -> str:
|
|
return __timenow().strftime(format=settings.DATETIME_STR_FORMAT)
|
|
|
|
|
|
def timenow_dt() -> datetime:
|
|
return datetime.strptime(timenow_str(), settings.DATETIME_STR_FORMAT)
|
|
|
|
|
|
def to_datetime(date_value: str) -> datetime:
|
|
return parser.parse(date_value)
|
|
|
|
|
|
def datetime_math(
|
|
date_val: str | datetime, days: int, addition: bool = True
|
|
) -> datetime:
|
|
if isinstance(date_val, str):
|
|
date_dt: datetime = to_datetime(date_val)
|
|
else:
|
|
date_dt = date_val
|
|
|
|
if addition:
|
|
return date_dt + timedelta(days=days) # noqa
|
|
else:
|
|
return date_dt - timedelta(days=days) # noqa
|
|
|
|
|
|
def format_datetime(
|
|
dat_value: str | datetime, human_format: bool = False, with_time: bool = True
|
|
) -> str:
|
|
if not dat_value:
|
|
return ""
|
|
|
|
if human_format:
|
|
if with_time:
|
|
_format = settings.DATETIME_HUMAN_FORMAT
|
|
else:
|
|
_format = settings.DATE_HUMAN_FORMAT
|
|
else:
|
|
if with_time:
|
|
_format = settings.DATETIME_STR_FORMAT
|
|
else:
|
|
_format = settings.DATE_STR_FORMAT
|
|
|
|
if isinstance(dat_value, datetime):
|
|
return dat_value.strftime(_format)
|
|
|
|
return to_datetime(dat_value).strftime(_format)
|
|
|
|
|
|
def make_tz_aware(unaware: str | datetime) -> datetime:
|
|
if isinstance(unaware, datetime):
|
|
return unaware.replace(tzinfo=settings.TIMEZONE)
|
|
return datetime.strptime(unaware, settings.DATETIME_STR_FORMAT).replace(
|
|
tzinfo=settings.TIMEZONE
|
|
)
|