重构延时方法

find_source/common/tools.py

@@ -104,22 +104,17 @@ def date2ts(date_str: str, fmt="%Y-%m-%d"):
     return timestamp
 
 
-def delay_by_hour(hour, fmt="%Y-%m-%d %H:%M:%S"):
-    """按小时延时"""
-    _hour = int(hour)
+def delay_by(delay=0, method='seconds', fmt="%Y-%m-%d %H:%M:%S"):
+    """按指定方式获得顺延时间"""
     _current_now = datetime.datetime.now()
-    return (_current_now + datetime.timedelta(hours=_hour)).strftime(fmt)
-
-
-def delay_by_minutes(minutes, fmt="%Y-%m-%d %H:%M:%S"):
-    """按分钟延时"""
-    _minutes = int(minutes)
-    _current_now = datetime.datetime.now()
-    return (_current_now + datetime.timedelta(minutes=_minutes)).strftime(fmt)
-
-
-def delay_by_day(days, fmt="%Y-%m-%d %H:%M:%S"):
-    """按天延时"""
-    _days = int(days)
-    _current_now = datetime.datetime.now()
-    return (_current_now + datetime.timedelta(days=_days)).strftime(fmt)
+    if method == 'days':
+        _timedelta = datetime.timedelta(days=delay)
+    elif method == 'hours':
+        _timedelta = datetime.timedelta(hours=delay)
+    elif method == 'minutes':
+        _timedelta = datetime.timedelta(minutes=delay)
+    elif method == 'microseconds':
+        _timedelta = datetime.timedelta(microseconds=delay)
+    else:
+        _timedelta = datetime.timedelta(seconds=delay)
+    return (_current_now + _timedelta).strftime(fmt)