def test_misc_funcs(self):
dic = MultikeyDict()
dic[1, 'bar'] = 2
dic[3, 'baz'] = 5
self.assertEqual(len(dic), 2)
del dic[1]
self.assertEqual(len(dic), 1)
dic.clear()
self.assertEqual(len(dic), 0)
class PlatformProtocol(protocol):
_next_id = 0
def __init__(self, *args, **kwargs):
self._buttons = MultikeyDict()
self._platforms = {}
self._distance_triggers = set()
self._autosave_loop = LoopingCall(self.dump_platform_json)
protocol.__init__(self, *args, **kwargs)
def add_distance_trigger(self, trigger):
self._distance_triggers.add(trigger)
trigger.signal_remove.connect(self._distance_triggers.remove)
def update_distance_triggers(self, player):
for trigger in self._distance_triggers:
trigger.update(player)
def on_world_update(self):
for player in self.players.values():
self.update_distance_triggers(player)
protocol.on_world_update(self)
def assign_id(self):
id_ = self._next_id
self._next_id += 1
return id_
def add_button(self, button):
self._buttons[(button.id, button.location)] = button
def create_button(self, location, color, label):
if self.is_platform_or_button(location):
return None
button = Button(self, self._next_id, location, color, label)
button.add_trigger(PressTrigger(self, False, button))
self._buttons[(self._next_id, location)] = button
self._next_id += 1
return button
def destroy_button(self, button):
button.destroy()
del self._buttons[button]
for player in self.players.values(
): # clear last button memory from players
if player.last_button is button:
player.last_button = None
def create_platform(self, location1, location2, z, color, label):
for location in prism_range(*location1, z, *location2, z + 1):
if self.is_platform_or_button(location):
return None
platform = Platform(self, self._next_id, location1, location2, z,
color, label)
self._platforms[self._next_id] = platform
self._next_id += 1
return platform
def destroy_platform(self, platform):
platform.destroy()
del self._platforms[platform.id]
for player in self.players.values(
): # clear last platform memory from players
if player.last_platform is platform:
player.last_platform = None
def get_platform(self, location_or_id):
if location_or_id in self._platforms:
return self._platforms[location_or_id]
for platform in self._platforms.values():
if platform.contains(location_or_id):
return platform
def get_button(self, location_or_id):
if location_or_id in self._buttons:
return self._buttons[location_or_id]
def is_platform_or_button(self, location):
return self.get_platform(location) or location in self._buttons
def on_map_change(self, map):
self._next_id = 0
self._platforms.clear()
self._buttons.clear()
self._distance_triggers.clear()
self.load_platform_json()
if AUTOSAVE_EVERY:
self._autosave_loop.start(AUTOSAVE_EVERY * 60.0, now=False)
protocol.on_map_change(self, map)
def on_map_leave(self):
if SAVE_ON_MAP_CHANGE:
self.dump_platform_json()
if self._autosave_loop.running:
self._autosave_loop.stop()
protocol.on_map_leave(self)
def _get_platform_json_path(self):
filename = self.map_info.rot_info.full_name + '_platform.txt'
return os.path.join(DEFAULT_LOAD_DIR, filename)
def load_platform_json(self):
path = self._get_platform_json_path()
if not os.path.isfile(path):
return
with open(path, 'r') as file:
data = json.load(file)
ids = []
for platform_data in data['platforms']:
platform = Platform.unserialize(self, platform_data)
self._platforms[platform.id] = platform
ids.append(platform.id)
for button_data in data['buttons']:
button = Button.unserialize(self, button_data)
self._buttons[(button.id, button.location)] = button
ids.append(button.id)
self._next_id = max(ids) + 1 if ids else 0
for button in self._buttons.values():
button.trigger_check()
def dump_platform_json(self):
data = {
'platforms': [
platform.serialize()
for platform in self._platforms.values()
],
'buttons':
[button.serialize() for button in self._buttons.values()]
}
path = self.get_platform_json_path()
with open(path, 'w') as file:
json.dump(data, file, indent=4)