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

Fixed possible exception thrown when balance field is missing during schema conversion (#4523)

Browse Source 
Co-authored-by: Draper <27962761+Drapersniper@users.noreply.github.com>
This commit is contained in:
Chloe 2020-10-23 12:39:14 -05:00 committed by GitHub
parent ca078e7cd9
commit e3359e9b3f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
redbot/core/bank.py
View File

@ -109,14 +109,16 @@ async def _schema_0_to_1():
group = _config._get_base_group(_config.USER) group = _config._get_base_group(_config.USER)
bank_user_data = await group.all() bank_user_data = await group.all()
for user_config in bank_user_data.values(): for user_config in bank_user_data.values():
user_config["balance"] = int(user_config["balance"]) if "balance" in user_config:
user_config["balance"] = int(user_config["balance"])
await group.set(bank_user_data) await group.set(bank_user_data)
group = _config._get_base_group(_config.MEMBER) group = _config._get_base_group(_config.MEMBER)
bank_member_data = await group.all() bank_member_data = await group.all()
for guild_data in bank_member_data.values(): for guild_data in bank_member_data.values():
for member_config in guild_data.values(): for member_config in guild_data.values():
member_config["balance"] = int(member_config["balance"]) if "balance" in member_config:
member_config["balance"] = int(member_config["balance"])
await group.set(bank_member_data) await group.set(bank_member_data)