patching PDFCore.py to properly check for ids in self.objects when re…

…f JSObjects
hatching · May 18, 2018 · 38b1554 · 38b1554
1 parent 8d6a370
commit 38b1554
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 11 deletions.
diff --git a/peepdf/PDFCore.py b/peepdf/PDFCore.py
@@ -4323,16 +4323,24 @@ def updateObjects(self):
                             del(compressedObjectsDict)
         for id in self.referencedJSObjects:
             if id not in self.containingJS:
-                object = self.objects[id].getObject()
-                if object is None:
-                    errorMessage = 'Object is None'
-                    if isForceMode:
-                        pdfFile.addError(errorMessage)
-                        continue
-                    else:
-                        return (-1, errorMessage)
-                object.setReferencedJSObject(True)
-                self.updateStats(id, object)
+                if self.objects:
+                    if str(id) in self.objects:
+                        errorMessage = 'Object is None'
+                        if isForceMode:
+                            pdfFile.addError(errorMessage)
+                            continue
+                else:
+                    object = self.objects[str(id)].getObject()
+                    if object:
+                        errorMessage = 'Object is None'
+                        if isForceMode:
+                            pdfFile.addError("TEST")
+                            pdfFile.addError(errorMessage)
+                            continue
+                        else:
+                            return (-1, errorMessage)
+                    object.setReferencedJSObject(True)
+                    self.updateStats(id, object)
         if errorMessage != '':
             return (-1, errorMessage)
         return (0, '')

diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name="peepdf",
-    version="0.4.1",
+    version="0.4.2",
     author="Jose Miguel Esparza",
     license="GNU GPLv3",
     url="http://eternal-todo.com",