sphinxcontrib-screenshot 0.1.1__tar.gz → 0.1.2__tar.gz

{sphinxcontrib_screenshot-0.1.1 → sphinxcontrib_screenshot-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sphinxcontrib-screenshot
-Version: 0.1.1
+Version: 0.1.2
 Summary: A Shpinx extension to embed webpage screenshots.
 Home-page: https://github.com/tushuhei/sphinxcontrib-screenshot/
 Author: Shuhei Iitsuka

{sphinxcontrib_screenshot-0.1.1 → sphinxcontrib_screenshot-0.1.2}/sphinxcontrib/screenshot.py

@@ -32,7 +32,7 @@ Meta = typing.TypedDict('Meta', {
     'parallel_write_safe': bool
 })
 
-__version__ = '0.1.1'
+__version__ = '0.1.2'
 
 
 class ScreenshotDirective(SphinxDirective):
@@ -80,6 +80,7 @@ class ScreenshotDirective(SphinxDirective):
       'height': directives.positive_int,
       'width': directives.positive_int,
       'caption': directives.unchanged,
+      'figclass': directives.unchanged,
   }
   pool = ThreadPoolExecutor()
 
@@ -133,6 +134,7 @@ class ScreenshotDirective(SphinxDirective):
     height = self.options.get('height', 960)
     width = self.options.get('width', 1280)
     caption_text = self.options.get('caption', '')
+    figclass = self.options.get('figclass', '')
     interactions = '\n'.join(self.content)
 
     if urlparse(url).scheme not in {'http', 'https'}:
@@ -158,6 +160,9 @@ class ScreenshotDirective(SphinxDirective):
     image_node = nodes.image(uri=rel_filepath)
     figure_node = nodes.figure('', image_node)
 
+    if figclass:
+      figure_node['classes'].append(figclass)
+
     if caption_text:
       parsed = nodes.Element()
       self.state.nested_parse(

{sphinxcontrib_screenshot-0.1.1 → sphinxcontrib_screenshot-0.1.2}/sphinxcontrib_screenshot.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sphinxcontrib-screenshot
-Version: 0.1.1
+Version: 0.1.2
 Summary: A Shpinx extension to embed webpage screenshots.
 Home-page: https://github.com/tushuhei/sphinxcontrib-screenshot/
 Author: Shuhei Iitsuka

{sphinxcontrib_screenshot-0.1.1 → sphinxcontrib_screenshot-0.1.2}/tests/test_it.py

@@ -29,7 +29,7 @@ def test_default(app: SphinxTestApp, status: StringIO,
 
   # Every screenshot directive should become an image.
   imgs = soup.find_all('img')
-  assert len(list(imgs)) == 3
+  assert len(list(imgs)) == 4
 
   # The image size should be set as specified.
   img_obj = Image.open(app.outdir / imgs[0]['src'])
@@ -44,7 +44,7 @@ def test_default(app: SphinxTestApp, status: StringIO,
 
   # The images should be different after the specified user interaction.
   img_before_interaction = Image.open(app.outdir / imgs[0]['src'])
-  img_after_interaction = Image.open(app.outdir / imgs[-1]['src'])
+  img_after_interaction = Image.open(app.outdir / imgs[2]['src'])
   assert list(img_before_interaction.getdata()) != list(
       img_after_interaction.getdata())
 
@@ -52,3 +52,6 @@ def test_default(app: SphinxTestApp, status: StringIO,
   img_with_caption_a = imgs[0]
   img_with_caption_b = imgs[1]
   assert img_with_caption_a['src'] == img_with_caption_b['src']
+
+  # The figure node should have the class name specified.
+  assert 'round' in soup.find_all('figure')[-1]['class']