icalendar/src/icalendar/parser_tools.py

35 wiersze
1.1 KiB
Python

# -*- coding: utf-8 -*-
from icalendar import compat
SEQUENCE_TYPES = (list, tuple)
DEFAULT_ENCODING = 'utf-8'
def to_unicode(value, encoding='utf-8'):
"""Converts a value to unicode, even if it is already a unicode string.
"""
if isinstance(value, compat.unicode_type):
return value
elif isinstance(value, compat.bytes_type):
try:
value = value.decode(encoding)
except UnicodeDecodeError:
value = value.decode('utf-8', 'replace')
return value
def data_encode(data, encoding=DEFAULT_ENCODING):
"""Encode all datastructures to the given encoding.
Currently unicode strings, dicts and lists are supported.
"""
# https://stackoverflow.com/questions/1254454/fastest-way-to-convert-a-dicts-keys-values-from-unicode-to-str
if isinstance(data, compat.unicode_type):
return data.encode(encoding)
elif isinstance(data, dict):
return dict(map(data_encode, compat.iteritems(data)))
elif isinstance(data, list) or isinstance(data, tuple):
return list(map(data_encode, data))
else:
return data