[V3 Config] Driver code initial cleanup (#1315)

* Remove get_driver

* Rename self.driver to self._driver

* Do not unnecessarily pass the cog identifier

* Remove unused import

* Fix type annotation

* Missed a keyword rename

* Modify signature of get/set methods in drivers
This commit is contained in:
Will
2018-02-18 22:30:32 -05:00
committed by palmtree5
parent 3984cb8f48
commit f9d846a704
5 changed files with 51 additions and 55 deletions

View File

@@ -41,21 +41,20 @@ class JSON(BaseDriver):
self.data = {}
self.jsonIO._save_json(self.data)
def get_driver(self):
return self
async def get(self, identifiers: Tuple[str]):
async def get(self, *identifiers: Tuple[str]):
partial = self.data
for i in identifiers:
full_identifiers = (self.unique_cog_identifier, *identifiers)
for i in full_identifiers:
partial = partial[i]
return partial
async def set(self, identifiers, value):
async def set(self, *identifiers: str, value=None):
partial = self.data
for i in identifiers[:-1]:
full_identifiers = (self.unique_cog_identifier, *identifiers)
for i in full_identifiers[:-1]:
if i not in partial:
partial[i] = {}
partial = partial[i]
partial[identifiers[-1]] = value
partial[full_identifiers[-1]] = value
await self.jsonIO._threadsafe_save_json(self.data)