Add test and fix for Page.add_widget().

pymupdf · Jan 31, 2024 · 9e6e9cc · 9e6e9cc
1 parent 1ff29bb
commit 9e6e9cc
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/src/__init__.py b/src/__init__.py
@@ -16764,7 +16764,7 @@ def JM_new_javascript(pdf, value):
         # not convertible to char*
         return
 
-    res = mupdf.fz_new_buffer_from_copied_data(data)
+    res = mupdf.fz_new_buffer_from_copied_data(data.encode('utf8'))
     source = mupdf.pdf_add_stream(pdf, res, mupdf.PdfObj(), 0)
     newaction = mupdf.pdf_add_new_dict(pdf, 4)
     mupdf.pdf_dict_put(newaction, PDF_NAME('S'), mupdf.pdf_new_name('JavaScript'))
@@ -17577,7 +17577,7 @@ def GETATTR(name):
 
     # script (/A) -------------------------------------------------------
     value = GETATTR("script")
-    JM_put_script(annot_obj, PDF_NAME('A'), None, value)
+    JM_put_script(annot_obj, PDF_NAME('A'), mupdf.PdfObj(), value)
 
     # script (/AA/K) -------------------------------------------------------
     value = GETATTR("script_stroke")

diff --git a/tests/test_widgets.py b/tests/test_widgets.py
@@ -232,3 +232,16 @@ def test_2391():
 #     page = pdf[0]
 #     field = page.first_widget
 #     page.delete_widget(field)
+
+
+def test_add_widget():
+    doc = fitz.open()
+    page = doc.new_page()
+    w = fitz.Widget()
+    w.field_type = fitz.PDF_WIDGET_TYPE_BUTTON
+    w.rect = fitz.Rect(5, 5, 20, 20)
+    w.field_flags = fitz.PDF_BTN_FIELD_IS_PUSHBUTTON
+    w.field_name = "button"
+    w.fill_color = (0, 0, 1)
+    w.script = "app.alert('Hello, PDF!');"
+    page.add_widget(w)