|
|
|
@ -249,16 +249,9 @@ class CringeBotClient(BotClient): |
|
|
|
|
def on_load_state(self): |
|
|
|
|
state = { |
|
|
|
|
"deletion_queue": {}, |
|
|
|
|
"cringe": [], |
|
|
|
|
"based": [], |
|
|
|
|
"unsure": [], |
|
|
|
|
"own": {}, |
|
|
|
|
**super().on_load_state()} |
|
|
|
|
|
|
|
|
|
state["cringe"] = set(state["cringe"]) |
|
|
|
|
state["based"] = set(state["based"]) |
|
|
|
|
state["unsure"] = set(state["unsure"]) |
|
|
|
|
|
|
|
|
|
for status_id, params in state["deletion_queue"].items(): |
|
|
|
|
# Load deletion queue into scheduler |
|
|
|
|
params["scheduler_event"] = self.deletion_scheduler.enterabs(datetime.fromisoformat(params["time"]).timestamp(), 1, self.queued_delete, argument=(status_id,)) |
|
|
|
@ -266,10 +259,6 @@ class CringeBotClient(BotClient): |
|
|
|
|
return state |
|
|
|
|
|
|
|
|
|
def on_save_state(self, state): |
|
|
|
|
state["cringe"] = list(state["cringe"]) |
|
|
|
|
state["based"] = list(state["based"]) |
|
|
|
|
state["unsure"] = list(state["unsure"]) |
|
|
|
|
|
|
|
|
|
# Transform deletion scheduler queue to a JSON friendly format |
|
|
|
|
state["deletion_queue"] = {event.argument[0]: {"time": datetime.fromtimestamp(event.time, timezone.utc).isoformat()} for event in self.deletion_scheduler.queue} |
|
|
|
|
|
|
|
|
|