diff --git a/main.py b/main.py index 832d0ac..a7e7b7b 100644 --- a/main.py +++ b/main.py @@ -8,12 +8,16 @@ This file is part of discord-qrm2 and is released under the terms of the GNU General Public License, version 2. """ +from datetime import time, datetime +import random + +import pytz + import discord from discord.ext import commands, tasks import common as cmn import info - import data.options as opt import data.keys as keys @@ -113,6 +117,12 @@ async def _extctl_unload(ctx: commands.Context, extension: str): async def on_ready(): print(f"Logged in as: {bot.user} - {bot.user.id}") print("------") + if opt.status_mode == "time": + _ensure_activity_time.start() + elif opt.status_mode == "random": + _ensure_activity_random.start() + else: + _ensure_activity_fixed.start() @bot.event @@ -128,13 +138,38 @@ async def on_message(message): # --- Tasks --- @tasks.loop(minutes=5) -async def _ensure_activity(): - await bot.change_presence(activity=discord.Game(name=opt.game)) +async def _ensure_activity_time(): + status = opt.statuses[0] + + try: + tz = pytz.timezone(opt.status_tz) + except pytz.exceptions.UnknownTimeZoneError: + await bot.change_presence(activity=discord.Game(name="with invalid timezones.")) + return + + now = datetime.now(tz=tz).time() + + for sts in opt.time_statuses: + start_time = time(hour=sts[1][0], minute=sts[1][1], tzinfo=tz) + end_time = time(hour=sts[2][0], minute=sts[2][1], tzinfo=tz) + if start_time < now <= end_time: + status = sts[0] + + await bot.change_presence(activity=discord.Game(name=status)) -@_ensure_activity.before_loop -async def _before_ensure_activity(): - await bot.wait_until_ready() +@tasks.loop(minutes=5) +async def _ensure_activity_random(): + status = random.choice(opt.statuses) + + await bot.change_presence(activity=discord.Game(name=status)) + + +@tasks.loop(minutes=5) +async def _ensure_activity_fixed(): + status = opt.statuses[0] + + await bot.change_presence(activity=discord.Game(name=status)) # --- Run --- @@ -142,8 +177,6 @@ async def _before_ensure_activity(): for ext in opt.exts: bot.load_extension(ext_dir + '.' + ext) -_ensure_activity.start() - try: bot.run(keys.discord_token) diff --git a/requirements.txt b/requirements.txt index a31e785..21f069a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,3 +2,4 @@ discord.py ctyparser beautifulsoup4 lxml +pytz diff --git a/templates/data/options.py b/templates/data/options.py index 6fe15b7..194b7ad 100644 --- a/templates/data/options.py +++ b/templates/data/options.py @@ -29,8 +29,21 @@ owners_uids = (200102491231092736,) # The extensions to load when running the bot. exts = ['ae7q', 'base', 'fun', 'grid', 'ham', 'image', 'lookup', 'morse', 'qrz', 'study', 'weather'] -# The text to put in the "playing" status. -game = 'with lids on 7.200' +# Either "time", "random", or "fixed" (first item in statuses) +status_mode = "fixed" + +# Random statuses pool +statuses = ["with lids on the air", "with fire"] + +# Timezone for the status (string) +# See https://pythonhosted.org/pytz/ for more info +status_tz = 'US/Eastern' +# The text to put in the "playing" status, with start and stop times +time_statuses = [('with lids on 3.840', (00, 00), (6, 00)), + ('with lids on 7.200', (6, 00), (10, 00)), + ('with lids on 14.313', (10, 00), (18, 00)), + ('with lids on 7.200', (18, 00), (20, 00)), + ('with lids on 3.840', (20, 00), (23, 59))] # Emoji IDs and keywords for emoji reactions # Use the format {emoji_id (int): ('tuple', 'of', 'lowercase', 'keywords')}