Added --ramdb option (#639)

tests/test_cli.py

@@ -6710,6 +6710,93 @@ def test_export_exportdb():
             in result.output
         )
 
+def test_export_exportdb_ramdb():
+    """test --exportdb --ramdb"""
+    import glob
+    import os
+    import os.path
+    import re
+
+    import osxphotos
+    from osxphotos.cli import export
+
+    runner = CliRunner()
+    cwd = os.getcwd()
+    # pylint: disable=not-context-manager
+    with runner.isolated_filesystem():
+        result = runner.invoke(
+            export,
+            [os.path.join(cwd, CLI_PHOTOS_DB), ".", "-V", "--exportdb", "export.db", "--ramdb"],
+        )
+        assert result.exit_code == 0
+        assert re.search(r"Created export database.*export\.db", result.output)
+        files = glob.glob("*")
+        assert "export.db" in files
+
+        result = runner.invoke(
+            export,
+            [
+                os.path.join(cwd, CLI_PHOTOS_DB),
+                ".",
+                "-V",
+                "--exportdb",
+                "export.db",
+                "--update",
+                "--ramdb"
+            ],
+        )
+        assert result.exit_code == 0
+        assert re.search(r"Using export database.*export\.db", result.output)
+        assert "exported: 0" in result.output
+
+
+def test_export_ramdb():
+    """test --ramdb"""
+    import glob
+    import os
+    import os.path
+    import re
+
+    import osxphotos
+    from osxphotos.cli import export
+
+    runner = CliRunner()
+    cwd = os.getcwd()
+    # pylint: disable=not-context-manager
+    with runner.isolated_filesystem():
+        result = runner.invoke(
+            export,
+            [os.path.join(cwd, CLI_PHOTOS_DB), ".", "-V", "--ramdb"],
+        )
+        assert result.exit_code == 0
+
+        # run again, update should update no files if db written back to disk
+        result = runner.invoke(
+            export,
+            [
+                os.path.join(cwd, CLI_PHOTOS_DB),
+                ".",
+                "-V",
+                "--update",
+                "--ramdb"
+            ],
+        )
+        assert result.exit_code == 0
+        assert "exported: 0" in result.output
+
+        # run again without --ramdb, update should update no files if db written back to disk
+        result = runner.invoke(
+            export,
+            [
+                os.path.join(cwd, CLI_PHOTOS_DB),
+                ".",
+                "-V",
+                "--update",
+            ],
+        )
+        assert result.exit_code == 0
+        assert "exported: 0" in result.output
+
 
 def test_export_finder_tag_keywords():
     """test --finder-tag-keywords"""

tests/test_export_db.py

@@ -177,6 +177,69 @@ def test_export_db_in_memory():
     assert uuids == [uuid]
 
 
+def test_export_db_in_memory_write_to_disk():
+    """test ExportDBInMemory with write back to disk"""
+    tempdir = tempfile.TemporaryDirectory(prefix="osxphotos_")
+    dbname = os.path.join(tempdir.name, ".osxphotos_export.db")
+    db = ExportDB(dbname, tempdir.name)
+    assert os.path.isfile(dbname)
+
+    filepath = os.path.join(tempdir.name, "test.JPG")
+
+    uuid = "FOOBAR"
+    record = db.create_file_record(filepath, uuid)
+    record.photoinfo = INFO_DATA
+    record.exifdata = EXIF_DATA
+    record.digest = DIGEST_DATA
+    record.src_sig = (7, 8, 9)
+    record.dest_sig = (10, 11, 12)
+    db.close()
+
+    # create in memory version
+    dbram = ExportDBInMemory(dbname, tempdir.name)
+    record2 = dbram.get_file_record(filepath)
+    assert record2.uuid == uuid
+    assert record2.photoinfo == INFO_DATA
+    assert record2.exifdata == EXIF_DATA
+    assert record2.digest == DIGEST_DATA
+    assert record2.src_sig == (7, 8, 9)
+    assert record2.dest_sig == (10, 11, 12)
+
+    # change some values
+    record2.photoinfo = INFO_DATA2
+    record2.exifdata = EXIF_DATA2
+    record2.digest = DIGEST_DATA2
+    record2.src_sig = (13, 14, 15)
+    record2.dest_sig = (16, 17, 18)
+
+    assert record2.photoinfo == INFO_DATA2
+    assert record2.exifdata == EXIF_DATA2
+    assert record2.digest == DIGEST_DATA2
+    assert record2.src_sig == (13, 14, 15)
+    assert record2.dest_sig == (16, 17, 18)
+
+    # all uuids
+    uuids = dbram.get_previous_uuids()
+    assert uuids == [uuid]
+
+    # write to disk
+    dbram.write_to_disk()
+    dbram.close()
+
+    # re-open original, assert changes are written back
+    db = ExportDB(dbname, tempdir.name)
+    record = db.get_file_record(filepath)
+    assert record.photoinfo == INFO_DATA2
+    assert record.exifdata == EXIF_DATA2
+    assert record.digest == DIGEST_DATA2
+    assert record.src_sig == (13, 14, 15)
+    assert record.dest_sig == (16, 17, 18)
+
+    # all uuids
+    uuids = db.get_previous_uuids()
+    assert uuids == [uuid]
+
+
 def test_export_db_in_memory_nofile():
     """test ExportDBInMemory with no dbfile"""
     tempdir = tempfile.TemporaryDirectory(prefix="osxphotos_")