X-Git-Url: https://plomlompom.com/repos/feed.xml?a=blobdiff_plain;f=plomtask%2Fdating.py;h=26b3ce3f4c2dae84fb44c8366d4a96882ced76ef;hb=f1795f22a6ea2409fc1c7c0ea921dc530e7625c2;hp=c55d847a1319a3687ac0ac06edc63b79e142ca09;hpb=eff89a3ebc0b3bf5b340b0ebd2b32fa136d8f640;p=plomtask

diff --git a/plomtask/dating.py b/plomtask/dating.py
index c55d847..26b3ce3 100644
--- a/plomtask/dating.py
+++ b/plomtask/dating.py
@@ -11,13 +11,11 @@ def valid_date(date_str: str) -> str:
     In any case, returns in DATE_FORMAT.
     """
     if date_str == 'today':
-        date_str = todays_date()
+        date_str = date_in_n_days(0)
     elif date_str == 'yesterday':
-        date = datetime.now() - timedelta(days=1)
-        date_str = date.strftime(DATE_FORMAT)
+        date_str = date_in_n_days(-1)
     elif date_str == 'tomorrow':
-        date = datetime.now() + timedelta(days=1)
-        date_str = date.strftime(DATE_FORMAT)
+        date_str = date_in_n_days(1)
     try:
         dt = datetime.strptime(date_str, DATE_FORMAT)
     except (ValueError, TypeError) as e:
@@ -26,6 +24,7 @@ def valid_date(date_str: str) -> str:
     return dt.strftime(DATE_FORMAT)
 
 
-def todays_date() -> str:
-    """Return current date in DATE_FORMAT."""
-    return datetime.now().strftime(DATE_FORMAT)
+def date_in_n_days(n: int) -> str:
+    """Return in DATE_FORMAT date from today + n days."""
+    date = datetime.now() + timedelta(days=n)
+    return date.strftime(DATE_FORMAT)