from icalendar.parser_tools import to_unicode
from collections import OrderedDict
[docs]
def canonsort_keys(keys, canonical_order=None):
"""Sorts leading keys according to canonical_order. Keys not specified in
canonical_order will appear alphabetically at the end.
"""
canonical_map = {k: i for i, k in enumerate(canonical_order or [])}
head = [k for k in keys if k in canonical_map]
tail = [k for k in keys if k not in canonical_map]
return sorted(head, key=lambda k: canonical_map[k]) + sorted(tail)
[docs]
def canonsort_items(dict1, canonical_order=None):
"""Returns a list of items from dict1, sorted by canonical_order.
"""
return [(k, dict1[k]) for k
in canonsort_keys(dict1.keys(), canonical_order)]
[docs]
class CaselessDict(OrderedDict):
"""A dictionary that isn't case sensitive, and only uses strings as keys.
Values retain their case.
"""
def __init__(self, *args, **kwargs):
"""Set keys to upper for initial dict.
"""
super().__init__(*args, **kwargs)
for key, value in self.items():
key_upper = to_unicode(key).upper()
if key != key_upper:
super().__delitem__(key)
self[key_upper] = value
def __getitem__(self, key):
key = to_unicode(key)
return super().__getitem__(key.upper())
def __setitem__(self, key, value):
key = to_unicode(key)
super().__setitem__(key.upper(), value)
def __delitem__(self, key):
key = to_unicode(key)
super().__delitem__(key.upper())
def __contains__(self, key):
key = to_unicode(key)
return super().__contains__(key.upper())
[docs]
def get(self, key, default=None):
key = to_unicode(key)
return super().get(key.upper(), default)
[docs]
def setdefault(self, key, value=None):
key = to_unicode(key)
return super().setdefault(key.upper(), value)
[docs]
def pop(self, key, default=None):
key = to_unicode(key)
return super().pop(key.upper(), default)
[docs]
def popitem(self):
return super().popitem()
def has_key(self, key):
key = to_unicode(key)
return super().__contains__(key.upper())
[docs]
def update(self, *args, **kwargs):
# Multiple keys where key1.upper() == key2.upper() will be lost.
mappings = list(args) + [kwargs]
for mapping in mappings:
if hasattr(mapping, 'items'):
mapping = iter(mapping.items())
for key, value in mapping:
self[key] = value
[docs]
def copy(self):
return type(self)(super().copy())
def __repr__(self):
return f'{type(self).__name__}({dict(self)})'
def __eq__(self, other):
return self is other or dict(self.items()) == dict(other.items())
def __ne__(self, other):
return not self == other
# A list of keys that must appear first in sorted_keys and sorted_items;
# must be uppercase.
canonical_order = None
[docs]
def sorted_keys(self):
"""Sorts keys according to the canonical_order for the derived class.
Keys not specified in canonical_order will appear at the end.
"""
return canonsort_keys(self.keys(), self.canonical_order)
[docs]
def sorted_items(self):
"""Sorts items according to the canonical_order for the derived class.
Items not specified in canonical_order will appear at the end.
"""
return canonsort_items(self, self.canonical_order)
__all__ = ["canonsort_keys", "canonsort_items", "CaselessDict"]