GitHub/PyLink
3
0
Fork 0
mirror of https://github.com/jlu5/PyLink.git synced 2024-12-28 13:42:37 +01:00
Code Issues Projects Releases Wiki Activity

structures: fix _keymangle when key isn't a string

Browse Source 
I really need to add unit tests for these...
This commit is contained in:
James Lu 2017-10-10 22:14:33 -07:00
parent 66c762b63f
commit 762ec3a0eb
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
structures.py
View File

@ -72,7 +72,9 @@ class CaseInsensitiveFixedSet(collections.abc.Set, CopyWrapper):
@staticmethod
def _keymangle(key):
"""Converts the given key to lowercase."""
return key.lower()
if isinstance(key, str):
return key.lower()
return key
def __repr__(self):
return "%s(%s)" % (self.__class__.__name__, self._data)
@ -121,7 +123,9 @@ class IRCCaseInsensitiveDict(CaseInsensitiveDict):
def _keymangle(self, key):
"""Converts the given key to lowercase."""
return self._irc.to_lower(key)
if isinstance(key, str):
return self._irc.to_lower(key)
return key
def __copy__(self):
return self.__class__(self._irc, data=self._data.copy())
@ -147,7 +151,9 @@ class IRCCaseInsensitiveSet(CaseInsensitiveSet):
def _keymangle(self, key):
"""Converts the given key to lowercase."""
return self._irc.to_lower(key)
if isinstance(key, str):
return self._irc.to_lower(key)
return key
def __copy__(self):
return self.__class__(self._irc, data=self._data.copy())