sphinxcontrib-screenshot 0.0.3__tar.gz → 0.0.4__tar.gz

{sphinxcontrib-screenshot-0.0.3 → sphinxcontrib-screenshot-0.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sphinxcontrib-screenshot
-Version: 0.0.3
+Version: 0.0.4
 Summary: A Shpinx extension to embed webpage screenshots.
 Author: Shuhei Iitsuka
 Author-email: tushuhei@gmail.com

{sphinxcontrib-screenshot-0.0.3 → sphinxcontrib-screenshot-0.0.4}/setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = sphinxcontrib-screenshot
-version = 0.0.3
+version = attr: sphinxcontrib.screenshot.__version__
 description = A Shpinx extension to embed webpage screenshots.
 long_description = file: README.md
 long_description_content_type = text/markdown

{sphinxcontrib-screenshot-0.0.3 → sphinxcontrib-screenshot-0.0.4}/sphinxcontrib/screenshot.py

@@ -19,6 +19,7 @@ import typing
 from docutils import nodes
 from docutils.parsers.rst import directives
 from docutils.statemachine import ViewList
+from playwright.sync_api import Browser, Playwright
 from playwright.sync_api import TimeoutError as PlaywrightTimeoutError
 from playwright.sync_api import sync_playwright
 from sphinx.application import Sphinx
@@ -30,8 +31,10 @@ Meta = typing.TypedDict('Meta', {
     'parallel_write_safe': bool
 })
 
-p = sync_playwright().start()
-browser = p.chromium.launch()
+p: typing.Optional[Playwright] = None
+browser: typing.Optional[Browser] = None
+
+__version__ = '0.0.4'
 
 
 class ScreenshotDirective(SphinxDirective):
@@ -108,6 +111,9 @@ class ScreenshotDirective(SphinxDirective):
     filename = hashlib.md5(hash_input.encode()).hexdigest() + '.png'
     filepath = os.path.join(ss_dirpath, filename)
 
+    if not browser:
+      raise RuntimeError('Browser has not been initiated.')
+
     # Check if the file already exists. If not, take a screenshot
     if not os.path.exists(filepath):
       page = browser.new_page()
@@ -151,17 +157,29 @@ class ScreenshotDirective(SphinxDirective):
     return [figure_node]
 
 
+def on_builder_inited(app: Sphinx):
+  global p
+  global browser
+  p = sync_playwright().start()
+  browser = p.chromium.launch()
+
+
 def on_build_finished(app: Sphinx, exception: Exception):
-  browser.close()
-  p.stop()
+  global p
+  global browser
+  if browser:
+    browser.close()
+  if p:
+    p.stop()
 
 
 def setup(app: Sphinx) -> Meta:
   app.add_directive('screenshot', ScreenshotDirective)
   app.connect('build-finished', on_build_finished)
+  app.connect('builder-inited', on_builder_inited)
   app.add_config_value('screenshot_init_script', '', 'env')
   return {
-      'version': '0.0.3',
+      'version': __version__,
       'parallel_read_safe': True,
       'parallel_write_safe': True,
   }

{sphinxcontrib-screenshot-0.0.3 → sphinxcontrib-screenshot-0.0.4}/sphinxcontrib_screenshot.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sphinxcontrib-screenshot
-Version: 0.0.3
+Version: 0.0.4
 Summary: A Shpinx extension to embed webpage screenshots.
 Author: Shuhei Iitsuka
 Author-email: tushuhei@gmail.com