Update file save procedure

tdh8316 · Jun 4, 2020 · 58b7711 · 58b7711
1 parent a0b222d
commit 58b7711
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/triangler/mod.py b/triangler/mod.py
@@ -1,4 +1,4 @@
-import time
+from datetime import datetime
 from typing import Union
 
 from numpy.core.multiarray import ndarray
@@ -72,7 +72,7 @@ def save(self, source: Union[str, ndarray], output: str = None, **kwargs) -> Non
         """
         _is_source_string: bool = isinstance(source, str)
         _out = output or (
-            "Triangler_{}.jpg".format(int(time.time()))
+            "Triangler_{}.jpg".format(datetime.now().strftime("%H-%M-%b-%d-%G"))
             if not _is_source_string
             else (str().join(source.split(".")[:-1]) + "_tri." + source.split(".")[-1])
         )
@@ -87,4 +87,4 @@ def save(self, source: Union[str, ndarray], output: str = None, **kwargs) -> Non
         )
 
         if kwargs["print_log"]:
-            print("Saved {}".format(output))
+            print("Saved {}".format(_out))