[dt] create separate module for datetime utilities
This commit is contained in:
108
gallery_dl/dt.py
Normal file
108
gallery_dl/dt.py
Normal file
@@ -0,0 +1,108 @@
|
|||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
|
# Copyright 2025 Mike Fährmann
|
||||||
|
#
|
||||||
|
# This program is free software; you can redistribute it and/or modify
|
||||||
|
# it under the terms of the GNU General Public License version 2 as
|
||||||
|
# published by the Free Software Foundation.
|
||||||
|
|
||||||
|
"""Date/Time utilities"""
|
||||||
|
|
||||||
|
import sys
|
||||||
|
import time
|
||||||
|
from datetime import datetime, date, timedelta, timezone # noqa F401
|
||||||
|
|
||||||
|
EPOCH = datetime(1970, 1, 1)
|
||||||
|
SECOND = timedelta(0, 1)
|
||||||
|
|
||||||
|
|
||||||
|
def parse_datetime(date_string, format="%Y-%m-%dT%H:%M:%S%z", utcoffset=0):
|
||||||
|
"""Create a datetime object by parsing 'date_string'"""
|
||||||
|
try:
|
||||||
|
d = datetime.strptime(date_string, format)
|
||||||
|
o = d.utcoffset()
|
||||||
|
if o is not None:
|
||||||
|
# convert to naive UTC
|
||||||
|
d = d.replace(tzinfo=None, microsecond=0) - o
|
||||||
|
else:
|
||||||
|
if d.microsecond:
|
||||||
|
d = d.replace(microsecond=0)
|
||||||
|
if utcoffset:
|
||||||
|
# apply manual UTC offset
|
||||||
|
d += timedelta(0, utcoffset * -3600)
|
||||||
|
return d
|
||||||
|
except (TypeError, IndexError, KeyError):
|
||||||
|
return None
|
||||||
|
except (ValueError, OverflowError):
|
||||||
|
return date_string
|
||||||
|
|
||||||
|
|
||||||
|
def to_datetime(value):
|
||||||
|
"""Convert 'value' to a datetime object"""
|
||||||
|
if not value:
|
||||||
|
return EPOCH
|
||||||
|
|
||||||
|
if isinstance(value, datetime):
|
||||||
|
return value
|
||||||
|
|
||||||
|
if isinstance(value, str):
|
||||||
|
try:
|
||||||
|
if value[-1] == "Z":
|
||||||
|
# compat for Python < 3.11
|
||||||
|
value = value[:-1]
|
||||||
|
dt = datetime.fromisoformat(value)
|
||||||
|
if dt.tzinfo is None:
|
||||||
|
if dt.microsecond:
|
||||||
|
dt = dt.replace(microsecond=0)
|
||||||
|
else:
|
||||||
|
# convert to naive UTC
|
||||||
|
dt = dt.astimezone(timezone.utc).replace(
|
||||||
|
microsecond=0, tzinfo=None)
|
||||||
|
return dt
|
||||||
|
except Exception:
|
||||||
|
pass
|
||||||
|
|
||||||
|
return parse_timestamp(value, EPOCH)
|
||||||
|
|
||||||
|
|
||||||
|
def datetime_to_timestamp(dt):
|
||||||
|
"""Convert naive UTC datetime to Unix timestamp"""
|
||||||
|
return (dt - EPOCH) / SECOND
|
||||||
|
|
||||||
|
|
||||||
|
def datetime_to_timestamp_string(dt):
|
||||||
|
"""Convert naive UTC datetime to Unix timestamp string"""
|
||||||
|
try:
|
||||||
|
return str((dt - EPOCH) // SECOND)
|
||||||
|
except Exception:
|
||||||
|
return ""
|
||||||
|
|
||||||
|
|
||||||
|
if sys.hexversion < 0x30c0000:
|
||||||
|
# Python <= 3.11
|
||||||
|
datetime_utcfromtimestamp = datetime.utcfromtimestamp
|
||||||
|
datetime_utcnow = datetime.utcnow
|
||||||
|
datetime_from_timestamp = datetime_utcfromtimestamp
|
||||||
|
|
||||||
|
def parse_timestamp(ts, default=None):
|
||||||
|
"""Create a datetime object from a Unix timestamp"""
|
||||||
|
try:
|
||||||
|
return datetime_utcfromtimestamp(int(ts))
|
||||||
|
except Exception:
|
||||||
|
return default
|
||||||
|
else:
|
||||||
|
# Python >= 3.12
|
||||||
|
def datetime_from_timestamp(ts=None):
|
||||||
|
"""Convert Unix timestamp to naive UTC datetime"""
|
||||||
|
Y, m, d, H, M, S, _, _, _ = time.gmtime(ts)
|
||||||
|
return datetime(Y, m, d, H, M, S)
|
||||||
|
|
||||||
|
def parse_timestamp(ts, default=None):
|
||||||
|
"""Create a datetime object from a Unix timestamp"""
|
||||||
|
try:
|
||||||
|
return datetime_from_timestamp(int(ts))
|
||||||
|
except Exception:
|
||||||
|
return default
|
||||||
|
|
||||||
|
datetime_utcfromtimestamp = datetime_from_timestamp
|
||||||
|
datetime_utcnow = datetime_from_timestamp
|
||||||
@@ -6,5 +6,5 @@
|
|||||||
# it under the terms of the GNU General Public License version 2 as
|
# it under the terms of the GNU General Public License version 2 as
|
||||||
# published by the Free Software Foundation.
|
# published by the Free Software Foundation.
|
||||||
|
|
||||||
__version__ = "1.30.10"
|
__version__ = "1.30.11-dev"
|
||||||
__variant__ = None
|
__variant__ = None
|
||||||
|
|||||||
Reference in New Issue
Block a user