pass down kwargs from saveImage(), so we can specify eg. the codec fo…

…r mp4 export

src/drawbot_skia/document.py

@@ -73,7 +73,7 @@ def saveImage(self, path, **kwargs):
         method = getattr(self, methodName, None)
         if method is None:
             raise ValueError(f"unsupported file type: {suffix}")
-        method(path)
+        method(path, **kwargs)
 
     def _saveImage_pdf(self, path, **kwargs):
         stream = skia.FILEWStream(os.fspath(path))
@@ -92,7 +92,7 @@ def _saveImage_jpeg(self, path, **kwargs):
 
     _saveImage_jpg = _saveImage_jpeg
 
-    def _saveImage_mp4(self, path, **kwargs):
+    def _saveImage_mp4(self, path, codec="libx264", **kwargs):
         from .ffmpeg import generateMP4
         if not self._pictures:
             # Empty mp4?
@@ -111,7 +111,7 @@ def _saveImage_mp4(self, path, **kwargs):
                 singlePage=False,
             )
             imagesTemplate = os.fspath(tempDir / "frame_%d.png")
-            generateMP4(imagesTemplate, path, frameRate)
+            generateMP4(imagesTemplate, path, frameRate, codec=codec)
 
 
 def _savePictures(pictures, path, format, whiteBackground=False, singlePage=None):

src/drawbot_skia/drawing.py

@@ -200,10 +200,10 @@ def savedState(self):
         self._canvas.restore()
         self._gstate = self._stack.pop()
 
-    def saveImage(self, fileName):
+    def saveImage(self, fileName, **kwargs):
         if self._document.isDrawing:
             self._document.endPage()
-        self._document.saveImage(fileName)
+        self._document.saveImage(fileName, **kwargs)
 
     # Helpers
 

tests/test_api.py

@@ -56,10 +56,12 @@ def test_saveImage_mp4(tmpdir):
     namespace = makeDrawbotNamespace(db)
     runScriptSource(multipageSource, "<string>", namespace)
     assert [] == sorted(tmpdir.glob("*.png"))
-    outputPath = tmpdir / "test.mp4"
-    db.saveImage(outputPath)
-    expected_filenames = ['test.mp4']
-    assert expected_filenames == [p.name for p in sorted(tmpdir.glob("*.mp4"))]
+    db.saveImage(tmpdir / "test.mp4")
+    db.saveImage(tmpdir / "test2.mp4", codec="mpeg4")
+    expected_filenames = ['test.mp4', 'test2.mp4']
+    paths = sorted(tmpdir.glob("*.mp4"))
+    assert paths[0].stat().st_size < paths[1].stat().st_size
+    assert expected_filenames == [p.name for p in paths]
 
 
 def test_noFont(tmpdir):