3
0
mirror of https://github.com/jlu5/PyLink.git synced 2024-12-26 04:32:51 +01:00

UserMapping: add in missing reference to the parent irc instance

This commit is contained in:
James Lu 2018-06-08 15:54:06 -07:00
parent 06d57a5b28
commit 6f3813d3a4

View File

@ -174,13 +174,14 @@ class UserMapping(collections.abc.MutableMapping, structures.CopyWrapper):
A mapping storing User objects by UID, as well as UIDs by nick via A mapping storing User objects by UID, as well as UIDs by nick via
the 'bynick' attribute the 'bynick' attribute
""" """
def __init__(self, *, data=None): def __init__(self, irc, data=None):
if data is not None: if data is not None:
assert isinstance(data, dict) assert isinstance(data, dict)
self._data = data self._data = data
else: else:
self._data = {} self._data = {}
self.bynick = collections.defaultdict(list) self.bynick = collections.defaultdict(list)
self._irc = irc
def __getitem__(self, key): def __getitem__(self, key):
return self._data[key] return self._data[key]
@ -188,7 +189,7 @@ class UserMapping(collections.abc.MutableMapping, structures.CopyWrapper):
def __setitem__(self, key, userobj): def __setitem__(self, key, userobj):
assert hasattr(userobj, 'lower_nick'), "Cannot add object without lower_nick attribute to UserMapping" assert hasattr(userobj, 'lower_nick'), "Cannot add object without lower_nick attribute to UserMapping"
if key in self._data: if key in self._data:
log.warning('(%s) Attempting to replace User object for %r: %r -> %r', self.name, log.warning('(%s) Attempting to replace User object for %r: %r -> %r', self._irc.name,
key, self._data.get(key), userobj) key, self._data.get(key), userobj)
self._data[key] = userobj self._data[key] = userobj
@ -310,7 +311,7 @@ class PyLinkNetworkCore(structures.CamelCaseToSnakeCase):
# Intialize the server, channel, and user indexes to be populated by # Intialize the server, channel, and user indexes to be populated by
# our protocol module. # our protocol module.
self.servers = {} self.servers = {}
self.users = UserMapping() self.users = UserMapping(self)
# Two versions of the channels index exist in PyLink 2.0, and they are joined together # Two versions of the channels index exist in PyLink 2.0, and they are joined together
# - irc._channels which implicitly creates channels on access (mostly used # - irc._channels which implicitly creates channels on access (mostly used