Close phantomjs instance after take screenshot

collective · Dec 5, 2016 · c53b74c · c53b74c
1 parent 1dbec43
commit c53b74c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -4,5 +4,10 @@ Changelog
 1.0a1 (unreleased)
 ------------------
 
+- Close phantomjs instance after take screenshot (fixes `#34`_).
+  [rodfersou]
+
 - Fix extra space between text and quotes (pseudo element).
   [rodfersou]
+
+.. _`#34`: https://github.com/collective/collective.behavior.featuredimage/issues/34
diff --git a/src/collective/behavior/featuredimage/subscriber.py b/src/collective/behavior/featuredimage/subscriber.py
@@ -4,6 +4,7 @@
 from plone.namedfile.file import NamedBlobImage
 from selenium import webdriver
 
+import signal
 import transaction
 
 
@@ -28,6 +29,10 @@ def _get_screenshot(page):
     im.save(output, 'PNG')
     data = output.getvalue()
     output.close()
+    driver.close()
+    # http://stackoverflow.com/a/3849328
+    driver.service.process.send_signal(signal.SIGTERM)
+    driver.quit()
     return data