Browse Source

add a param for get_time_str()

chenhaiyang 4 years ago
parent
commit
f0947122eb
1 changed files with 5 additions and 2 deletions
  1. 5 2
      app/utils/date.py

+ 5 - 2
app/utils/date.py

@@ -1,15 +1,18 @@
 # -*- coding: utf-8 -*-
 
+from typing import Optional
+
 import arrow
 
 TIME_FMT = 'YYYYMMDDHHmmss'
 
 
-def get_time_str(delta: int = 0, flag: str = 'now') -> str:
+def get_time_str(delta: int = 0, flag: str = 'now', fmt: Optional[str] = TIME_FMT) -> str:
     """
     Return a Beijing time strings.
     :param delta: time delta(seconds)
     :param flag:
+    :param fmt: time string's format
     :return: a '%Y%m%d%H%M%S' format strings
     """
     utc = arrow.utcnow()
@@ -22,7 +25,7 @@ def get_time_str(delta: int = 0, flag: str = 'now') -> str:
     else:
         t = local
 
-    time_str = t.format(TIME_FMT)
+    time_str = t.format(fmt)
 
     return time_str