xevion/unbelievaselfbot
1
0
Fork 0
mirror of https://github.com/Xevion/unbelievaselfbot.git synced 2025-12-08 12:08:59 -06:00
Code Issues Packages Projects Releases Wiki Activity

add parsers check_valid implementation, TaskResponse optional logging, TaskCooldownMessage private vars and unparsed duration

Browse Source
This commit is contained in:
Xevion
2021-01-24 14:25:01 -06:00
parent 58a97c9a6b
commit 64688fdbb1
1 changed files with 35 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
bot/parsers.py
View File

@@ -14,6 +14,7 @@ class BaseMessage(object):
"""
Abstract base class for message data parsing.
"""
def __init__(self, message: discord.Message) -> None:
self.message = message
@@ -41,25 +42,33 @@ class TaskCooldownMessage(EmbedMessage):
DURATION_REGEX = re.compile(r'(\d+) (hour|minute|second)s?(?: and (\d+) (hour|minute|second)s?)?')
DELAY = 2
durations = {'hour': 3600, 'minute': 60, 'second': 1}
task_parsings = {'work': '$work', 'be a slut': '$slut', 'commit a crime': '$crime'}
__durations = {'hour': 3600, 'minute': 60, 'second': 1}
__task_parsings = {'work': '$work', 'be a slut': '$slut', 'commit a crime': '$crime'}
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
match = re.search(TaskCooldownMessage.COOLDOWN_REGEX, self.embed.description)
duration_match = re.match(TaskCooldownMessage.DURATION_REGEX, match.group(2))
self.duration_unparsed = match.group(2)
duration_match = re.match(TaskCooldownMessage.DURATION_REGEX, self.duration_unparsed)
# Process the 1-2 groups found and calculate the duration
self.duration = 0
groups = len(list(filter(lambda s: s is not None, duration_match.groups())))
for i in range(0, groups // 2):
x, y = (i * 2) + 1, (i * 2) + 2
self.duration += int(duration_match.group(x)) * TaskCooldownMessage.durations[duration_match.group(y)]
self.duration += int(duration_match.group(x)) * TaskCooldownMessage.__durations[duration_match.group(y)]
self.task_type = TaskCooldownMessage.task_parsings[match.group(1)]
self.task_type = TaskCooldownMessage.__task_parsings[match.group(1)]
self.available_at = self.message.created_at.timestamp() + TaskCooldownMessage.DELAY
@classmethod
def check_valid(cls, message: discord.Message) -> bool:
"""Checks whether a Message object is a bot's income task cooldown response."""
return len(message.embeds) > 0 \
and message.embeds[0].description != discord.Embed.Empty \
and message.embeds[0].description.startswith('<:stopwatch:630927808843218945> You cannot')
class TaskResponse(EmbedMessage):
MONEY_REGEX = re.compile(r'\$([0-9,]+)')
@@ -68,13 +77,28 @@ class TaskResponse(EmbedMessage):
super().__init__(*args, **kwargs)
self.change = 0
money_match = re.search(r'', self.embed.description)
if money_match and 'deposited' not in self.embed.description.lower():
money_match = re.search(TaskResponse.MONEY_REGEX, self.embed.description)
if money_match:
change = int(money_match.group(1).replace(',', ''))
if self.embed.colour.value == 6732650:
self.change += change
logger.info(f'{self.message.author} Gained ${change}')
if self.embed.colour.value == 15684432:
elif self.embed.colour.value == 15684432:
self.change -= change
logger.info(f'{self.message.author} Lost ${change}')
# logger.info(f'Lost ${change}')
def log_message(self, name: str = None) -> str:
return ('' if name is None else name + ' ') + ('Gained $' if self.change >= 0 else 'Lost $') + str(self.change)
@classmethod
def check_valid(cls, message: discord.Message) -> bool:
"""Checks that the message is a Bot Income Task response, denoting a amount of income or fine."""
if len(message.embeds) > 0:
embed = message.embeds[0]
return embed.colour.value in [6732650, 15684432] \
and all(sub not in embed.description.lower() for sub in ['deposited', 'received your', 'withdrew']) \
and not embed.description.startswith('<:') \
and re.search(cls.MONEY_REGEX, embed.description) is not None
def __repr__(self) -> str:
return f'TaskResponse(change={self.change})'