Skip to content

Commit ee7e1bd

Browse files
committed
Attempt to address PR issues
1 parent 768d9ad commit ee7e1bd

File tree

3 files changed

+7
-11
lines changed

3 files changed

+7
-11
lines changed

bot.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -204,7 +204,7 @@ async def on_ready(self):
204204
await self.threads.populate_cache()
205205
await self.config.update()
206206

207-
closures = self.config.get('closures', {}).copy()
207+
closures = self.config.closures.copy()
208208

209209
for recipient_id, items in closures.items():
210210
after = (datetime.datetime.fromisoformat(items['time']) -
@@ -218,9 +218,7 @@ async def on_ready(self):
218218

219219
if not thread:
220220
# If the recipient is gone or channel is deleted
221-
closures = self.config.get('closures', {})
222-
closures.pop(str(recipient_id))
223-
self.config['closures'] = closures
221+
self.config.closures.pop(str(recipient_id))
224222
await self.config.update()
225223
continue
226224

core/config.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ def populate_cache(self):
4343
'blocked': {},
4444
'notification_squad': {},
4545
'subscriptions': {},
46+
'closures': {},
4647
}
4748

4849
try:

core/thread.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,6 @@ async def close(self, *, closer, after=0, silent=False,
5959
# TODO: Add somewhere to clean up broken closures
6060
# (when channel is already deleted)
6161
await self.bot.config.update()
62-
closures = self.bot.config.get('closures', {})
6362
now = datetime.datetime.utcnow()
6463
items = {
6564
# 'initiation_time': now.isoformat(),
@@ -69,8 +68,7 @@ async def close(self, *, closer, after=0, silent=False,
6968
'delete_channel': delete_channel,
7069
'message': message
7170
}
72-
closures[str(self.id)] = items
73-
self.bot.config['closures'] = closures
71+
self.bot.config.closures[str(self.id)] = items
7472
await self.bot.config.update()
7573

7674
self.close_task = self.bot.loop.call_later(
@@ -84,10 +82,9 @@ async def _close(self, closer, silent=False, delete_channel=True,
8482
message=None, scheduled=False):
8583
del self.manager.cache[self.id]
8684

87-
closures = self.bot.config.get('closures', {})
88-
closures.pop(str(self.id))
89-
self.bot.config['closures'] = closures
90-
await self.bot.config.update()
85+
if scheduled:
86+
self.bot.config.closures.pop(str(self.id), None)
87+
await self.bot.config.update()
9188

9289
if str(self.id) in self.bot.config.subscriptions:
9390
del self.bot.config.subscriptions[str(self.id)]

0 commit comments

Comments
 (0)