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.
66 lines
1.7 KiB
66 lines
1.7 KiB
# -*- coding: utf-8 -*- |
|
# This file is part of pygal |
|
# |
|
# A python svg graph plotting library |
|
# Copyright © 2012-2013 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/>. |
|
""" |
|
Value adapters to use when a chart doesn't accept all value types |
|
|
|
""" |
|
import datetime |
|
from numbers import Number |
|
from pygal.i18n import COUNTRIES |
|
|
|
|
|
def positive(x): |
|
if x is None: |
|
return |
|
if x < 0: |
|
return 0 |
|
return x |
|
|
|
|
|
def not_zero(x): |
|
if x == 0: |
|
return |
|
return x |
|
|
|
|
|
def none_to_zero(x): |
|
return x or 0 |
|
|
|
|
|
def int_to_country(x): |
|
# This is used for test compatibility |
|
if isinstance(x, Number): |
|
try: |
|
x = int(x) |
|
except: |
|
return x |
|
if x >= 0 and x < len(COUNTRIES): |
|
return list(COUNTRIES.keys())[x] |
|
return x |
|
|
|
|
|
def date(x): |
|
# Make int work for date graphs by counting days number from now |
|
if isinstance(x, Number): |
|
try: |
|
d = datetime.date.today() + datetime.timedelta(days=x) |
|
return datetime.datetime.combine(d, datetime.time(0, 0, 0)) |
|
except OverflowError: |
|
return None |
|
return x
|
|
|