diff options
| -rw-r--r-- | gemato/manifest.py | 11 | ||||
| -rw-r--r-- | tests/test_manifest.py | 6 |
2 files changed, 17 insertions, 0 deletions
diff --git a/gemato/manifest.py b/gemato/manifest.py index 2c1f4b7..b0168a0 100644 --- a/gemato/manifest.py +++ b/gemato/manifest.py @@ -314,6 +314,17 @@ class ManifestFile(object): for e in self.entries: f.write(u' '.join(e.to_list()) + '\n') + def find_timestamp(self): + """ + Find a timestamp entry and return it. Returns None if there + is no timestamp. + """ + + for e in self.entries: + if isinstance(e, ManifestEntryTIMESTAMP): + return e + return None + def find_path_entry(self, path): """ Find a matching entry for path @path and return it. Returns diff --git a/tests/test_manifest.py b/tests/test_manifest.py index 5e72a8a..43c7e18 100644 --- a/tests/test_manifest.py +++ b/tests/test_manifest.py @@ -58,6 +58,12 @@ class ManifestTest(unittest.TestCase): m.dump(outf) self.assertEqual(outf.getvalue().strip(), TEST_DEPRECATED_MANIFEST.strip()) + def test_find_timestamp(self): + m = gemato.manifest.ManifestFile() + m.load(io.StringIO(TEST_MANIFEST)) + self.assertEqual(m.find_timestamp().ts, + datetime.datetime(2017, 10, 22, 18, 06, 41)) + def test_find_path_entry(self): m = gemato.manifest.ManifestFile() m.load(io.StringIO(TEST_MANIFEST)) |