From 712fee7eb228b4b2edcd10656b13d71a34b9b637 Mon Sep 17 00:00:00 2001 From: phxntxm Date: Wed, 31 Aug 2016 23:54:20 -0500 Subject: [PATCH] Corrected error when bot_data was not found --- cogs/statsupdate.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/cogs/statsupdate.py b/cogs/statsupdate.py index 7d0facc..f26745c 100644 --- a/cogs/statsupdate.py +++ b/cogs/statsupdate.py @@ -47,27 +47,27 @@ class StatsUpdate: async def on_server_join(self, server): data = await config.get_content('bot_data') - shard_data = data.get('shard_{}'.format(config.shard_id)) + shard_data = data.get('shard_{}'.format(config.shard_id)) or {} shard_data['server_count'] = len(self.bot.servers) - shard_data['member_count'] = len(list(self.bot.get_all_members())) + shard_data['member_count'] = len(set(self.bot.get_all_members())) data['shard_{}'.format(config.shard_id)] = shard_data await config.save_content('bot_data', data) await self.update(shard_data) async def on_server_leave(self, server): data = await config.get_content('bot_data') - shard_data = data.get('shard_{}'.format(config.shard_id)) + shard_data = data.get('shard_{}'.format(config.shard_id)) or {} shard_data['server_count'] = len(self.bot.servers) - shard_data['member_count'] = len(list(self.bot.get_all_members())) + shard_data['member_count'] = len(set(self.bot.get_all_members())) data['shard_{}'.format(config.shard_id)] = shard_data await config.save_content('bot_data', data) await self.update(shard_data) async def on_ready(self): data = await config.get_content('bot_data') - shard_data = data.get('shard_{}'.format(config.shard_id)) + shard_data = data.get('shard_{}'.format(config.shard_id)) or {} shard_data['server_count'] = len(self.bot.servers) - shard_data['member_count'] = len(list(self.bot.get_all_members())) + shard_data['member_count'] = len(set(self.bot.get_all_members())) data['shard_{}'.format(config.shard_id)] = shard_data await config.save_content('bot_data', data) await self.update(shard_data)