mirror of https://github.com/Kozea/pygal.git
Florian Mounier
10 years ago
20 changed files with 659 additions and 765 deletions
@ -1,220 +0,0 @@
|
||||
# -*- coding: utf-8 -*- |
||||
# This file is part of pygal |
||||
# |
||||
# A python svg graph plotting library |
||||
# Copyright © 2012-2014 Kozea |
||||
# |
||||
# 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/>. |
||||
""" |
||||
Ghost container |
||||
|
||||
It is used to delegate rendering to real objects but keeping config in place |
||||
|
||||
""" |
||||
|
||||
from __future__ import division |
||||
import io |
||||
import sys |
||||
from pygal._compat import u, is_list_like |
||||
from pygal.graph import CHARTS_NAMES |
||||
from pygal.config import Config, CONFIG_ITEMS |
||||
from pygal.util import prepare_values |
||||
from uuid import uuid4 |
||||
|
||||
|
||||
class ChartCollection(object): |
||||
pass |
||||
|
||||
|
||||
REAL_CHARTS = { |
||||
'DateY': 'pygal.graph.time.DateY', |
||||
'DateTimeLine': 'pygal.graph.time.DateTimeLine', |
||||
'DateLine': 'pygal.graph.time.DateLine', |
||||
'TimeLine': 'pygal.graph.time.TimeLine', |
||||
'TimeDeltaLine': 'pygal.graph.time.TimeDeltaLine' |
||||
} |
||||
|
||||
for NAME in CHARTS_NAMES: |
||||
if NAME in REAL_CHARTS: |
||||
mod_name = 'pygal.graph.time' |
||||
else: |
||||
mod_name = 'pygal.graph.%s' % NAME.lower() |
||||
|
||||
__import__(mod_name) |
||||
mod = sys.modules[mod_name] |
||||
chart = getattr(mod, NAME) |
||||
if issubclass(chart, ChartCollection): |
||||
for name, chart in chart.__dict__.items(): |
||||
if name.startswith('_'): |
||||
continue |
||||
REAL_CHARTS['%s_%s' % (NAME, name)] = chart |
||||
else: |
||||
REAL_CHARTS[NAME] = chart |
||||
|
||||
|
||||
class Ghost(object): |
||||
|
||||
def __init__(self, config=None, **kwargs): |
||||
"""Init config""" |
||||
name = self.__class__.__name__ |
||||
self.cls = REAL_CHARTS[name] |
||||
self.uuid = str(uuid4()) |
||||
if config and isinstance(config, type): |
||||
config = config() |
||||
|
||||
if config: |
||||
config = config.copy() |
||||
else: |
||||
config = Config() |
||||
|
||||
config(**kwargs) |
||||
self.config = config |
||||
self.raw_series = [] |
||||
self.raw_series2 = [] |
||||
self.xml_filters = [] |
||||
|
||||
def add(self, title, values, **kwargs): |
||||
"""Add a serie to this graph""" |
||||
if not is_list_like(values) and not isinstance(values, dict): |
||||
values = [values] |
||||
if kwargs.get('secondary', False): |
||||
self.raw_series2.append((title, values, kwargs)) |
||||
else: |
||||
self.raw_series.append((title, values, kwargs)) |
||||
|
||||
def add_xml_filter(self, callback): |
||||
self.xml_filters.append(callback) |
||||
|
||||
def make_series(self, series, offset=0): |
||||
return prepare_values(series, self.config, self.cls, offset) |
||||
|
||||
def make_instance(self, overrides=None): |
||||
for conf_key in CONFIG_ITEMS: |
||||
if conf_key.is_list: |
||||
if getattr(self, conf_key.name, None): |
||||
setattr(self, conf_key.name, |
||||
list(getattr(self, conf_key.name))) |
||||
|
||||
self.config(**self.__dict__) |
||||
self.config.__dict__.update(overrides or {}) |
||||
series = self.make_series(self.raw_series) |
||||
secondary_series = self.make_series( |
||||
self.raw_series2, len(series or [])) |
||||
self._last__inst = self.cls( |
||||
self.config, series, secondary_series, self.uuid, |
||||
self.xml_filters) |
||||
return self._last__inst |
||||
|
||||
# Rendering |
||||
def render(self, is_unicode=False, **kwargs): |
||||
return (self |
||||
.make_instance(overrides=kwargs) |
||||
.render(is_unicode=is_unicode)) |
||||
|
||||
def render_tree(self, **kwargs): |
||||
return self.make_instance(overrides=kwargs).render_tree() |
||||
|
||||
def render_table(self, **kwargs): |
||||
# Import here to avoid lxml import |
||||
try: |
||||
from pygal.table import Table |
||||
except ImportError: |
||||
raise ImportError('You must install lxml to use render table') |
||||
real_cls, self.cls = self.cls, Table |
||||
rv = self.make_instance().render(**kwargs) |
||||
self.cls = real_cls |
||||
return rv |
||||
|
||||
def render_pyquery(self): |
||||
"""Render the graph, and return a pyquery wrapped tree""" |
||||
from pyquery import PyQuery as pq |
||||
return pq(self.render(), parser='html') |
||||
|
||||
def render_in_browser(self, **kwargs): |
||||
"""Render the graph, open it in your browser with black magic""" |
||||
try: |
||||
from lxml.html import open_in_browser |
||||
except ImportError: |
||||
raise ImportError('You must install lxml to use render in browser') |
||||
open_in_browser(self.render_tree(**kwargs), encoding='utf-8') |
||||
|
||||
def render_response(self, **kwargs): |
||||
"""Render the graph, and return a Flask response""" |
||||
from flask import Response |
||||
return Response(self.render(**kwargs), mimetype='image/svg+xml') |
||||
|
||||
def render_django_response(self, **kwargs): |
||||
"""Render the graph, and return a Django response""" |
||||
from django.http import HttpResponse |
||||
return HttpResponse(self.render(**kwargs), content_type='image/svg+xml') |
||||
|
||||
def render_to_file(self, filename, **kwargs): |
||||
"""Render the graph, and write it to filename""" |
||||
with io.open(filename, 'w', encoding='utf-8') as f: |
||||
f.write(self.render(is_unicode=True, **kwargs)) |
||||
|
||||
def render_to_png(self, filename=None, dpi=72, **kwargs): |
||||
"""Render the graph, convert it to png and write it to filename""" |
||||
import cairosvg |
||||
return cairosvg.svg2png( |
||||
bytestring=self.render(**kwargs), write_to=filename, dpi=dpi) |
||||
|
||||
def render_sparktext(self, relative_to=None): |
||||
"""Make a mini text sparkline from chart""" |
||||
bars = u('▁▂▃▄▅▆▇█') |
||||
if len(self.raw_series) == 0: |
||||
return u('') |
||||
values = list(self.raw_series[0][1]) |
||||
if len(values) == 0: |
||||
return u('') |
||||
|
||||
chart = u('') |
||||
values = list(map(lambda x: max(x, 0), values)) |
||||
|
||||
vmax = max(values) |
||||
if relative_to is None: |
||||
relative_to = min(values) |
||||
|
||||
if (vmax - relative_to) == 0: |
||||
chart = bars[0] * len(values) |
||||
return chart |
||||
|
||||
divisions = len(bars) - 1 |
||||
for value in values: |
||||
chart += bars[int(divisions * |
||||
(value - relative_to) / (vmax - relative_to))] |
||||
return chart |
||||
|
||||
def render_sparkline(self, **kwargs): |
||||
spark_options = dict( |
||||
width=200, |
||||
height=50, |
||||
show_dots=False, |
||||
show_legend=False, |
||||
show_x_labels=False, |
||||
show_y_labels=False, |
||||
spacing=0, |
||||
margin=5, |
||||
explicit_size=True |
||||
) |
||||
spark_options.update(kwargs) |
||||
return self.make_instance(spark_options).render() |
||||
|
||||
def _repr_svg_(self): |
||||
"""Display svg in IPython notebook""" |
||||
return self.render(disable_xml_declaration=True) |
||||
|
||||
def _repr_png_(self): |
||||
"""Display png in IPython notebook""" |
||||
return self.render_to_png() |
||||
@ -0,0 +1,28 @@
|
||||
# -*- coding: utf-8 -*- |
||||
# This file is part of pygal |
||||
# |
||||
# A python svg graph plotting library |
||||
# Copyright © 2012-2014 Kozea |
||||
# |
||||
# 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/>. |
||||
""" |
||||
Class holding state during render |
||||
|
||||
""" |
||||
|
||||
|
||||
class State(object): |
||||
def __init__(self, graph): |
||||
self.__dict__.update(**graph.config.__dict__) |
||||
self.__dict__.update(**graph.__dict__) |
||||
Loading…
Reference in new issue