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

[V3 Config] Fix for unnecessary writes on context mgr exit (#1859)

Browse Source 
* Fix for #1857

* And copy it so we don't have mutability issues

* Add a test
This commit is contained in:
Will
2018-06-22 22:15:22 -04:00
committed by Toby Harradine
parent afa08713e0
commit edadd8f2fd
2 changed files with 16 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
redbot/core/config.py
View File

@@ -29,12 +29,15 @@ class _ValueCtxManager:
def __init__(self, value_obj, coro):
self.value_obj = value_obj
self.coro = coro
self.raw_value = None
self.__original_value = None
def __await__(self):
return self.coro.__await__()
async def __aenter__(self):
self.raw_value = await self
self.__original_value = deepcopy(self.raw_value)
if not isinstance(self.raw_value, (list, dict)):
raise TypeError(
"Type of retrieved value must be mutable (i.e. "
@@ -44,7 +47,8 @@ class _ValueCtxManager:
return self.raw_value
async def __aexit__(self, *exc_info):
await self.value_obj.set(self.raw_value)
if self.raw_value != self.__original_value:
await self.value_obj.set(self.raw_value)
class Value: