104 lines
2.7 KiB
Python
104 lines
2.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
# This file is part of pygal
|
|
#
|
|
# A python svg graph plotting library
|
|
# Copyright © 2012-2016 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/>.
|
|
"""Various hacks for transparent python 2 / python 3 support"""
|
|
from __future__ import division
|
|
|
|
import sys
|
|
from collections import Iterable
|
|
from datetime import datetime, timedelta, tzinfo
|
|
|
|
if sys.version_info[0] == 3:
|
|
base = (str, bytes)
|
|
coerce = str
|
|
_ellipsis = eval('...')
|
|
else:
|
|
base = basestring
|
|
coerce = unicode
|
|
|
|
class EllipsisGetter(object):
|
|
def __getitem__(self, key):
|
|
return key
|
|
|
|
_ellipsis = EllipsisGetter()[...]
|
|
|
|
|
|
def is_list_like(value):
|
|
"""Return whether value is an iterable but not a mapping / string"""
|
|
return isinstance(value, Iterable) and not isinstance(value, (base, dict))
|
|
|
|
|
|
def is_str(string):
|
|
"""Return whether value is a string or a byte list"""
|
|
return isinstance(string, base)
|
|
|
|
|
|
def to_str(obj):
|
|
"""Cast obj to unicode string"""
|
|
if not is_str(obj):
|
|
return coerce(obj)
|
|
return obj
|
|
|
|
|
|
def to_unicode(string):
|
|
"""Force string to be a string in python 3 or a unicode in python 2"""
|
|
if not isinstance(string, coerce):
|
|
return string.decode('utf-8')
|
|
return string
|
|
|
|
|
|
def u(s):
|
|
"""Emulate u'str' in python 2, do nothing in python 3"""
|
|
if sys.version_info[0] == 2:
|
|
return s.decode('utf-8')
|
|
return s
|
|
|
|
|
|
try:
|
|
from datetime import timezone
|
|
utc = timezone.utc
|
|
except ImportError:
|
|
class UTC(tzinfo):
|
|
def tzname(self, dt):
|
|
return 'UTC'
|
|
|
|
def utcoffset(self, dt):
|
|
return timedelta(0)
|
|
|
|
def dst(self, dt):
|
|
return None
|
|
utc = UTC()
|
|
|
|
|
|
def timestamp(x):
|
|
"""Get a timestamp from a date in python 3 and python 2"""
|
|
if x.tzinfo is None:
|
|
# Naive dates to utc
|
|
x = x.replace(tzinfo=utc)
|
|
|
|
if hasattr(x, 'timestamp'):
|
|
return x.timestamp()
|
|
else:
|
|
return (x - datetime(1970, 1, 1, tzinfo=utc)).total_seconds()
|
|
|
|
|
|
try:
|
|
from urllib import quote_plus
|
|
except ImportError:
|
|
from urllib.parse import quote_plus
|