From 8ef4de693a751e681750ed64cb4bb5e9e5d5139d Mon Sep 17 00:00:00 2001 From: jsl12 Date: Fri, 13 Aug 2021 17:24:43 -0500 Subject: [PATCH] overrode some builtin functions of MsgData --- data.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/data.py b/data.py index a920293..27b80e0 100644 --- a/data.py +++ b/data.py @@ -64,6 +64,15 @@ class MsgData: def __str__(self): return str(self.msgs) + '\n\n' + str(self.reactions) + def __repr__(self): + return f'<{__name__}.{self.__class__.__name__} with {self.msgs.shape[0]} messages and {self.reactions.shape[0]} reactions>' + + def __getitem__(self, item): + if isinstance(item, str): + return self.reactions.loc[pd.IndexSlice[:, item],].fillna(0).applymap(int) + elif isinstance(item, int): + return self.reactions.loc[pd.IndexSlice[item, :],].fillna(0).applymap(int) + async def add_msg(self, message: discord.Message): async with self.lock: mdict = message_dict(message)