Mercurial > lbo > hg > photosync

changeset 26:6e53be5c652e

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Improve date parsing Allow non-zero-padded month/day
author Lewin Bormann <lbo@spheniscida.de>
date Mon, 23 Mar 2020 16:16:37 +0100
parents 59b608562a36
children cebda502c732 520fa4e1c537
files photosync.py
diffstat 1 files changed, 6 insertions(+), 1 deletions(-) [+]
line wrap: on
line diff
--- a/photosync.py	Tue Nov 05 18:22:32 2019 +0100
+++ b/photosync.py	Mon Mar 23 16:16:37 2020 +0100
@@ -28,6 +28,10 @@
         msg = msg.format(*args)
     print (level, "::", msg)
 
+def make_date_iso(d):
+    """Expects a date like 2019-1-4 and preprocesses it for ISO parsing.
+    """
+    return '-'.join('{:02d}'.format(int(p)) for p in d.split('-'))
 
 class TokenSource:
     """Return OAuth token for PhotosService to use.
@@ -418,10 +422,11 @@
             window = None
             if len(parts) == 2:
                 (a, b) = parts
+                (a, b) = (make_date_iso(a), make_date_iso(b))
                 (a, b) = p.isoparse(a), p.isoparse(b)
                 window = (a, b)
             elif len(parts) == 1:
-                date = p.isoparse(parts[0])
+                date = p.isoparse(make_date_iso(parts[0]))
                 window = (date, date)
             else:
                 print("Please use --date with argument yyyy-mm-dd:yyyy-mm-dd (from:to) or yyyy-mm-dd.")