mirror of https://github.com/Kozea/pygal.git
Florian Mounier
10 years ago
3 changed files with 116 additions and 2 deletions
@ -0,0 +1,115 @@ |
|||||||
|
# -*- coding: utf-8 -*- |
||||||
|
# This file is part of pygal_sphinx_directives |
||||||
|
# |
||||||
|
# Pygal sphinx integration |
||||||
|
# Copyright © 2015 Florian Mounier |
||||||
|
# |
||||||
|
# This library is free software: you can redistribute it and/or modify it under |
||||||
|
# the terms of the GNU Lesser General Public License as published by the Free |
||||||
|
# Software Foundation, either version 3 of the License, or (at your option) any |
||||||
|
# later version. |
||||||
|
# |
||||||
|
# This library is distributed in the hope that it will be useful, but WITHOUT |
||||||
|
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS |
||||||
|
# FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more |
||||||
|
# details. |
||||||
|
# |
||||||
|
# You should have received a copy of the GNU Lesser General Public License |
||||||
|
# along with pygal. If not, see <http://www.gnu.org/licenses/>. |
||||||
|
|
||||||
|
|
||||||
|
from docutils.parsers.rst import Directive |
||||||
|
from traceback import format_exc |
||||||
|
from sphinx.directives.code import CodeBlock |
||||||
|
|
||||||
|
import base64 |
||||||
|
import docutils.core |
||||||
|
import pygal |
||||||
|
|
||||||
|
|
||||||
|
class PygalDirective(Directive): |
||||||
|
"""Execute the given python file and puts its result in the document.""" |
||||||
|
required_arguments = 0 |
||||||
|
optional_arguments = 2 |
||||||
|
final_argument_whitespace = True |
||||||
|
has_content = True |
||||||
|
|
||||||
|
def run(self): |
||||||
|
width, height = map(int, self.arguments[:2]) if len( |
||||||
|
self.arguments) >= 2 else (600, 400) |
||||||
|
if len(self.arguments) == 1: |
||||||
|
self.render_fix = bool(self.arguments[0]) |
||||||
|
elif len(self.arguments) == 3: |
||||||
|
self.render_fix = bool(self.arguments[2]) |
||||||
|
else: |
||||||
|
self.render_fix = False |
||||||
|
self.content = list(self.content) |
||||||
|
content = list(self.content) |
||||||
|
if self.render_fix: |
||||||
|
content[-1] = 'rv = ' + content[-1] |
||||||
|
code = '\n'.join(content) |
||||||
|
scope = {'pygal': pygal} |
||||||
|
try: |
||||||
|
exec(code, scope) |
||||||
|
except Exception: |
||||||
|
return [docutils.nodes.system_message( |
||||||
|
'An exception as occured during code parsing:' |
||||||
|
' \n %s' % format_exc(), type='ERROR', source='/', |
||||||
|
level=3)] |
||||||
|
if self.render_fix: |
||||||
|
rv = scope['rv'] |
||||||
|
else: |
||||||
|
chart = None |
||||||
|
for key, value in scope.items(): |
||||||
|
if isinstance(value, pygal.graph.graph.Graph): |
||||||
|
chart = value |
||||||
|
self.content.append(key + '.render()') |
||||||
|
break |
||||||
|
if chart is None: |
||||||
|
return [docutils.nodes.system_message( |
||||||
|
'No instance of graph found', level=3)] |
||||||
|
if not hasattr(chart, 'style'): |
||||||
|
chart.style = pygal.style.RTDStyle |
||||||
|
chart.config.width = width |
||||||
|
chart.config.height = height |
||||||
|
chart.explicit_size = True |
||||||
|
rv = chart.render() |
||||||
|
|
||||||
|
try: |
||||||
|
svg = ( |
||||||
|
'<embed src="data:image/svg+xml;charset=utf-8;base64,%s" />' % |
||||||
|
base64.b64encode(rv).decode('utf-8') |
||||||
|
.replace('\n', '')) |
||||||
|
except Exception: |
||||||
|
return [docutils.nodes.system_message( |
||||||
|
'An exception as occured during graph generation:' |
||||||
|
' \n %s' % format_exc(), type='ERROR', source='/', |
||||||
|
level=3)] |
||||||
|
return [docutils.nodes.raw('', svg, format='html')] |
||||||
|
|
||||||
|
|
||||||
|
class PygalWithCode(PygalDirective): |
||||||
|
width_code = True |
||||||
|
|
||||||
|
def run(self): |
||||||
|
node_list = super(PygalWithCode, self).run() |
||||||
|
node_list.extend(CodeBlock( |
||||||
|
self.name, |
||||||
|
['python'], |
||||||
|
self.options, |
||||||
|
self.content, |
||||||
|
self.lineno, |
||||||
|
self.content_offset, |
||||||
|
self.block_text, |
||||||
|
self.state, |
||||||
|
self.state_machine).run()) |
||||||
|
|
||||||
|
return [docutils.nodes.compound('', *node_list)] |
||||||
|
|
||||||
|
|
||||||
|
def setup(app): |
||||||
|
app.add_directive('pygal', PygalDirective) |
||||||
|
app.add_directive('pygal-code', PygalWithCode) |
||||||
|
app.add_directive('pygal-sparkline', PygalWithCode) |
||||||
|
|
||||||
|
return {'version': '1.0.1'} |
Loading…
Reference in new issue