Added save function for json dumping
This commit is contained in:
parent
f34e368ac3
commit
7ed1f2fcaa
@ -14,6 +14,14 @@ class FourasModule(BaseModule):
|
|||||||
self._client.answers = [line.strip() for line in a_file.readlines()]
|
self._client.answers = [line.strip() for line in a_file.readlines()]
|
||||||
print(f"Loaded {len(self._client.riddles)} riddles")
|
print(f"Loaded {len(self._client.riddles)} riddles")
|
||||||
|
|
||||||
|
def save(self):
|
||||||
|
dump = {}
|
||||||
|
for key, value in self._client.ongoing_riddles.items():
|
||||||
|
dump_channel = dict(value)
|
||||||
|
dump_channel["message"] = dump_channel["message"].id
|
||||||
|
dump[key.id] = dump_channel
|
||||||
|
return dump
|
||||||
|
|
||||||
def new_riddle(self, channel, index):
|
def new_riddle(self, channel, index):
|
||||||
current_riddle = dict(
|
current_riddle = dict(
|
||||||
index=index,
|
index=index,
|
||||||
@ -107,20 +115,15 @@ class FourasModule(BaseModule):
|
|||||||
)
|
)
|
||||||
return
|
return
|
||||||
|
|
||||||
if message_content == "debug fouras":
|
if message_content == "debug":
|
||||||
dump = {}
|
dump = {}
|
||||||
cooldowns = {}
|
|
||||||
for key, value in self.client.cooldown.items():
|
|
||||||
channel_name = await self.get_channel_name(key)
|
|
||||||
cooldowns[channel_name] = value - time.time()
|
|
||||||
dump["poil_au_cooldown"] = cooldowns
|
|
||||||
for key, value in self._client.ongoing_riddles.items():
|
for key, value in self._client.ongoing_riddles.items():
|
||||||
dump_channel = value
|
dump_channel = dict(value)
|
||||||
dump_channel.pop("message", None)
|
dump_channel.pop("message", None)
|
||||||
dump_channel["answer"] = "||{0}||".format(dump_channel["answer"])
|
dump_channel.pop("answer", None)
|
||||||
channel_name = await self.get_channel_name(key)
|
channel_name = await self.get_channel_name(key)
|
||||||
dump[channel_name] = dump_channel
|
dump[channel_name] = dump_channel
|
||||||
await message.author.send(json.dumps(dump, ensure_ascii=False, indent=4))
|
await message.author.send('```json\n{0}```'.format(json.dumps(dump, ensure_ascii=False, indent=4)))
|
||||||
return
|
return
|
||||||
|
|
||||||
# if current channel has ongoing riddle
|
# if current channel has ongoing riddle
|
||||||
|
@ -1,6 +1,7 @@
|
|||||||
from .base import BaseModule
|
from .base import BaseModule
|
||||||
import random
|
import random
|
||||||
import time
|
import time
|
||||||
|
import json
|
||||||
|
|
||||||
RHYMES_FILE = "rhymes.txt"
|
RHYMES_FILE = "rhymes.txt"
|
||||||
|
|
||||||
@ -34,6 +35,12 @@ class RhymesModule(BaseModule):
|
|||||||
self.rhyme_keys = keys
|
self.rhyme_keys = keys
|
||||||
self.rhyme_strings = rhymes
|
self.rhyme_strings = rhymes
|
||||||
|
|
||||||
|
def save(self):
|
||||||
|
dump = {}
|
||||||
|
for key, value in self.cooldown.items():
|
||||||
|
dump[key.id] = value - time.time()
|
||||||
|
return dump
|
||||||
|
|
||||||
def get_last_word(self, ch: str) -> str:
|
def get_last_word(self, ch: str) -> str:
|
||||||
truncated = ch
|
truncated = ch
|
||||||
while True:
|
while True:
|
||||||
@ -53,6 +60,15 @@ class RhymesModule(BaseModule):
|
|||||||
async def handle_message(self, message):
|
async def handle_message(self, message):
|
||||||
message_content = message.content.lower()
|
message_content = message.content.lower()
|
||||||
|
|
||||||
|
|
||||||
|
if message_content == "debug":
|
||||||
|
dump = {}
|
||||||
|
for key, value in self.cooldown.items():
|
||||||
|
channel_name = await self.get_channel_name(key)
|
||||||
|
dump[channel_name] = value - time.time()
|
||||||
|
await message.author.send('```json\n{0}```'.format(json.dumps(dump, ensure_ascii=False, indent=4)))
|
||||||
|
return
|
||||||
|
|
||||||
last_word = self.get_last_word(message_content)
|
last_word = self.get_last_word(message_content)
|
||||||
if message.author != self._client.user and last_word:
|
if message.author != self._client.user and last_word:
|
||||||
poil = self.poil_auquel(last_word)
|
poil = self.poil_auquel(last_word)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user