store functionality of meta data done

vgoehler · Mar 3, 2020 · 7f9a3bc · 7f9a3bc
1 parent bd290df
commit 7f9a3bc
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 3 deletions.
diff --git a/src/python_eulerian_video_magnification/metadata.py b/src/python_eulerian_video_magnification/metadata.py
@@ -1,5 +1,6 @@
-
+import json
 import os.path
+from datetime import datetime
 
 from python_eulerian_video_magnification.mode import Mode
 
@@ -21,7 +22,8 @@ def __init__(self, file_name: str, output_folder: str,
             'levels': levels,
             'amplification': amplification,
             'mode': mode.name,
-            'suffix': suffix
+            'suffix': suffix,
+            'date': None,
         }
 
     @staticmethod
@@ -32,3 +34,10 @@ def output_file_name(filename: str, suffix: str, path: str, generate_json=False)
 
     def __getitem__(self, item):
         return self.__data[item]
+
+    def save_meta_data(self):
+        """stores the meta data dictionary as a json"""
+        self.__data['date'] = str(datetime.now())
+        with open(self.__data['meta_target'], 'w') as fp:
+            json.dump(self.__data, fp=fp, sort_keys=True, indent=4, separators=(',', ': '))
+
diff --git a/tests/test_metadata.py b/tests/test_metadata.py
@@ -1,3 +1,5 @@
+import os.path
+
 from python_eulerian_video_magnification.metadata import MetaData
 from python_eulerian_video_magnification.mode import Mode
 
@@ -32,4 +34,19 @@ def test_for_data_as_expected():
                                                'target': "/out/put/gob_color_evm.avi",
                                                'meta_target': "/out/put/gob_color_evm.json", 'low': 0.1, 'high': 2.1,
                                                'levels': 4,
-                                               'amplification': 23, 'mode': 'COLOR', 'suffix': 'color'}
+                                               'amplification': 23, 'mode': 'COLOR', 'suffix': 'color', 'date': None}
+
+
+def test_json_save(tmp_path):
+    sut = MetaData(
+        file_name="/fu/bar/gob.avi",
+        output_folder=str(tmp_path),
+        mode=Mode.COLOR,
+        suffix="color",
+        low=0.1,
+        high=2.1,
+        levels=4,
+        amplification=23
+    )
+    sut.save_meta_data()
+    assert os.path.exists(sut['meta_target'])