Make timestamp helper timezone-aware

This commit is contained in:
Milkey Mouse 2020-11-05 00:59:00 -08:00
parent 9084f94ba2
commit 47143cae76
No known key found for this signature in database
GPG Key ID: C6EF5A02F5647987
2 changed files with 3 additions and 3 deletions

View File

@ -24,7 +24,7 @@ def timestamp(s):
try:
# is it pointing to a file / directory?
ts = safe_s(os.stat(s).st_mtime)
return datetime.utcfromtimestamp(ts)
return datetime.fromtimestamp(ts, tz=timezone.utc)
except OSError:
# didn't work, try parsing as timestamp. UTC, no TZ, no microsecs support.
for format in ('%Y-%m-%dT%H:%M:%SZ', '%Y-%m-%dT%H:%M:%S+00:00',
@ -33,7 +33,7 @@ def timestamp(s):
'%Y-%m-%d', '%Y-%j',
):
try:
return datetime.strptime(s, format)
return datetime.strptime(s, format).replace(tzinfo=timezone.utc)
except ValueError:
continue
raise ValueError

View File

@ -190,7 +190,7 @@ class TestLocationWithoutEnv:
"Location(proto='ssh', user=None, host='host', port=None, path='/path', archive='2016-12-31@23:59:59')"
def test_with_timestamp(self):
assert repr(Location('path::archive-{utcnow}').with_timestamp(datetime(2002, 9, 19))) == \
assert repr(Location('path::archive-{utcnow}').with_timestamp(datetime(2002, 9, 19, tzinfo=timezone.utc))) == \
"Location(proto='file', user=None, host=None, port=None, path='path', archive='archive-2002-09-19T00:00:00')"
def test_underspecified(self, monkeypatch):