Support figure / image alignment

creole/rest2html/clean_writer.py

@@ -164,6 +164,28 @@ def visit_docinfo(self, node):
     def depart_docinfo(self, node):
         self.body.append('</table>\n')
 
+    #__________________________________________________________________________
+    # Clean image:
+
+    depart_figure = _do_nothing
+
+    def visit_image(self, node):
+        super(CleanHTMLTranslator, self).visit_image(node)
+        if self.body[-1].startswith('<img'):
+            align = None
+
+            if 'align' in node:
+                # image with alignment
+                align = node['align']
+
+            elif node.parent.tagname == 'figure' and 'align' in node.parent:
+                # figure with alignment
+                align = node.parent['align']
+
+            if align:
+                self.body[-1] = self.body[-1].replace(' />', ' align="%s" />' % align)
+
+
 
 def rest2html(content, enable_exit_status=None, **kwargs):
     """

creole/tests/test_rest2html.py

@@ -148,6 +148,28 @@ def test_raw_enabled(self):
             <hr width=50 size=10>
         """, raw_enabled=True)
 
+    def test_preserve_image_alignment(self):
+        self.assert_rest2html("""
+            Image alignment should be preserved.
+
+            .. image:: foo.png
+               :align: right
+        """, """
+            <p>Image alignment should be preserved.</p>
+            <img alt="foo.png" src="foo.png" align="right" />
+        """)
+
+    def test_preserve_figure_alignment(self):
+        self.assert_rest2html("""
+            Image alignment should be preserved.
+
+            .. figure:: bar.png
+               :align: right
+        """, """
+            <p>Image alignment should be preserved.</p>
+            <img alt="bar.png" src="bar.png" align="right" />
+        """)
+
 
 if __name__ == '__main__':
     unittest.main()