storage: fix get_invite for non-guild invites

2019-02-27 00:11:11 -03:00 · 2019-02-27 00:11:11 -03:00 · f2bfb901c7
parent 8e6564dd7d
commit f2bfb901c7
1 changed files with 6 additions and 3 deletions
--- a/litecord/storage.py
+++ b/litecord/storage.py
@ -847,10 +847,13 @@ class Storage:
        WHERE id = $1
        """, invite['guild_id'])

-        dinv['guild'] = dict(guild)
+        if guild:
+            dinv['guild'] = dict(guild)

-        # TODO: query actual guild features
-        dinv['guild']['features'] = []
+            # TODO: query actual guild features
+            dinv['guild']['features'] = []
+        else:
+            dinv['guild'] = {}

        chan = await self.get_channel(invite['channel_id'])
        dinv['channel'] = {