john/kwaylon
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

overrode some builtin functions of MsgData

Browse Source
This commit is contained in:
jsl12
2021-08-13 17:24:43 -05:00
parent 40af217ea5
commit 8ef4de693a
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
data.py
View File

@@ -64,6 +64,15 @@ class MsgData:
def __str__(self): def __str__(self):
return str(self.msgs) + '\n\n' + str(self.reactions) 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 def add_msg(self, message: discord.Message):
async with self.lock: async with self.lock:
mdict = message_dict(message) mdict = message_dict(message)