xiaochuan's multiapp examples

desktop_env/evaluators/getters/file.py

@@ -1,7 +1,7 @@
 import os
 from typing import Dict, List, Set
 from typing import Optional, Any, Union
-
+from datetime import datetime
 import requests
 
 
@@ -58,14 +58,24 @@ def get_vm_file(env, config: Dict[str, Any]) -> Union[Optional[str], List[Option
         gives (List[int]): optional. defaults to [0]. which files are directly
           returned to the metric. if len==1, str is returned; else, list is
           returned.
+        only support for single file now:
+        time_suffix(bool): optional. defaults to False. if True, append the current time in required format.
+        time_format(str): optional. defaults to "%Y_%m_%d". format of the time suffix.
     """
-
+    time_format = "%Y_%m_%d"
     if not config.get("multi", False):
         paths: List[str] = [config["path"]]
         dests: List[str] = [config["dest"]]
     else:
         paths: List[str] = config["path"]
         dests: List[str] = config["dest"]
+        if "time_suffix" in config.keys() and config["time_suffix"]:
+            if "time_format" in config.keys():
+                time_format = config["time_format"]
+            # Insert time before . in file type suffix
+            paths = [p.split(".")[0] + datetime.now().strftime(time_format) + "." + p.split(".")[1] if "." in p else p for p in paths]
+            dests = [d.split(".")[0] + datetime.now().strftime(time_format) + "." + d.split(".")[1] if "." in d else d for d in dests]
+            
     cache_paths: List[str] = []
 
     gives: Set[int] = set(config.get("gives", [0]))
@@ -85,7 +95,8 @@ def get_vm_file(env, config: Dict[str, Any]) -> Union[Optional[str], List[Option
             cache_paths.append(_path)
         with open(_path, "wb") as f:
             f.write(file)
-
+    # debug
+    print(cache_paths)
     return cache_paths[0] if len(cache_paths)==1 else cache_paths