mirror of https://github.com/Kozea/pygal.git
Python to generate nice looking SVG graph
http://pygal.org/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
116 lines
3.8 KiB
116 lines
3.8 KiB
10 years ago
|
# -*- 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'}
|