76 lines
2.5 KiB
Python
76 lines
2.5 KiB
Python
|
from datetime import datetime, timedelta
|
||
|
|
|
||
|
|
from edmond.plugin import Plugin
|
||
|
|
from edmond.utils import proc
|
||
|
|
|
||
|
|
|
||
|
|
class SleepPlugin(Plugin):
|
||
|
|
"""Handle sleep state of the bot, snore a little bit."""
|
||
|
|
|
||
|
|
REQUIRED_CONFIGS = [
|
||
|
|
"sleep_time", "wakeup_time", "snore", "sleep_message", "wakeup_message",
|
||
|
|
"snore_rate"
|
||
|
|
]
|
||
|
|
|
||
|
|
def __init__(self, bot):
|
||
|
|
super().__init__(bot)
|
||
|
|
|
||
|
|
def on_ping(self, event):
|
||
|
|
awake = self.get_runtime_value("awake")
|
||
|
|
self.bot.log_w(f"{self.bot.channels}")
|
||
|
|
in_sleep_hours = self.is_sleep_time(datetime.now())
|
||
|
|
if awake and in_sleep_hours:
|
||
|
|
self.fall_asleep()
|
||
|
|
elif not awake:
|
||
|
|
if not in_sleep_hours:
|
||
|
|
self.wake_up()
|
||
|
|
else:
|
||
|
|
# Maybe snore.
|
||
|
|
if proc(self.config["snore_rate"]):
|
||
|
|
for channel in self.bot.channels:
|
||
|
|
self.bot.say(channel, self.config["snore"])
|
||
|
|
|
||
|
|
def fall_asleep(self):
|
||
|
|
for channel in self.bot.channels:
|
||
|
|
self.bot.say(channel, self.config["sleep_message"])
|
||
|
|
self.set_runtime_value("awake", False)
|
||
|
|
|
||
|
|
def wake_up(self):
|
||
|
|
self.set_runtime_value("awake", True)
|
||
|
|
for channel in self.bot.channels:
|
||
|
|
self.bot.say(channel, self.config["wakeup_message"])
|
||
|
|
|
||
|
|
def is_sleep_time(self, now):
|
||
|
|
"""Return True if the bot should be sleeping by now.
|
||
|
|
|
||
|
|
The sleep range can span over 2 days (e.g. 23 to 7) or be contained in a
|
||
|
|
day (e.g. 0 to 8).
|
||
|
|
"""
|
||
|
|
sleep_time = self.config["sleep_time"]
|
||
|
|
wakeup_time = self.config["wakeup_time"]
|
||
|
|
current_hour = now.hour
|
||
|
|
# If we're before the sleep hour, check that we are not in the
|
||
|
|
# previous sleep range. Else just check today's range.
|
||
|
|
if current_hour < sleep_time:
|
||
|
|
ref_dt = now - timedelta(days=1)
|
||
|
|
else:
|
||
|
|
ref_dt = now
|
||
|
|
sleep_dt, wakeup_dt = self._get_sleep_range(
|
||
|
|
ref_dt,
|
||
|
|
sleep_time,
|
||
|
|
wakeup_time
|
||
|
|
)
|
||
|
|
return sleep_dt <= now < wakeup_dt
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def _get_sleep_range(now, sleep_time, wakeup_time):
|
||
|
|
"""Return the (start, end) datetime tuple using config values.
|
||
|
|
|
||
|
|
Properly accounts for a sleep range going around midnight.
|
||
|
|
"""
|
||
|
|
sleep_dt = now.replace(hour=sleep_time, minute=0, second=0)
|
||
|
|
wakeup_dt = now.replace(hour=wakeup_time, minute=0, second=0)
|
||
|
|
if wakeup_time < sleep_time:
|
||
|
|
wakeup_dt += timedelta(days=1)
|
||
|
|
return sleep_dt, wakeup_dt
|