CopyBot/Red-DiscordBot
1
0
Fork 0
mirror of https://github.com/Cog-Creators/Red-DiscordBot.git synced 2025-11-06 11:18:54 -05:00
Code Issues Packages Projects Releases Wiki Activity

[V3 Mongo] Correct dictionary rebuilding process for global all case (#2581)

Browse Source
This commit is contained in:
Will 2019-04-13 15:51:49 -04:00 committed by GitHub
parent 8b3c3e89e9
commit 0652dd344b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
redbot/core/drivers/red_mongo.py
View File

@ -87,10 +87,10 @@ class Mongo(BaseDriver):
async for doc in cursor: async for doc in cursor:
pkeys = doc["_id"]["RED_primary_key"] pkeys = doc["_id"]["RED_primary_key"]
del doc["_id"] del doc["_id"]
if len(pkeys) == 1: if len(pkeys) == 0:
# Global data # Global data
ret.update(**doc) ret.update(**doc)
elif len(pkeys) > 1: elif len(pkeys) > 0:
# All other data # All other data
partial = ret partial = ret
for key in pkeys[:-1]: for key in pkeys[:-1]:
@ -103,8 +103,6 @@ class Mongo(BaseDriver):
partial.update(**doc) partial.update(**doc)
else: else:
partial[pkeys[-1]] = doc partial[pkeys[-1]] = doc
else:
raise RuntimeError("This should not happen.")
return ret return ret
async def get(self, identifier_data: IdentifierData): async def get(self, identifier_data: IdentifierData):