feat: sync topics to DB daily (#176)

OpenASL · Mar 14, 2021 · f975335 · f975335
1 parent fa59060
commit f975335
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 9 deletions.
diff --git a/bot/exts/topics.py b/bot/exts/topics.py
@@ -2,6 +2,7 @@
 import logging
 import random
 
+import discord
 from aiohttp import web
 from discord.ext.commands import Bot
 from discord.ext.commands import Cog
@@ -11,18 +12,25 @@
 
 from bot import settings
 from bot.database import store
+from bot.utils.datetimes import EASTERN
 from bot.utils.datetimes import utcnow
 from bot.utils.gsheets import get_gsheet_client
 
 logger = logging.getLogger(__name__)
 
 # -----------------------------------------------------------------------------
 
+DAILY_SYNC_TIME = dt.time(7, 0)  # Eastern time
+
 
 class Topics(Cog):
     def __init__(self, bot: Bot):
         self.bot = bot
 
+    @Cog.listener()
+    async def on_ready(self):
+        self.bot.loop.create_task(self.daily_sync())
+
     @command(
         name="synctopics",
         aliases=("st",),
@@ -34,6 +42,18 @@ async def sync_topics_command(self, ctx: Context):
         topics = await sync_topics()
         await ctx.reply(f"✅ Synced {len(topics)} topics.")
 
+    async def daily_sync(self):
+        while True:
+            now_eastern = dt.datetime.now(EASTERN)
+            date = now_eastern.date()
+            if now_eastern.time() > DAILY_SYNC_TIME:
+                date = now_eastern.date() + dt.timedelta(days=1)
+            then = EASTERN.localize(dt.datetime.combine(date, DAILY_SYNC_TIME))
+            logger.info(f"topics will be synced at at {then.isoformat()}")
+            await discord.utils.sleep_until(then.astimezone(dt.timezone.utc))
+            topics = await sync_topics()
+            logger.info(f"synced {len(topics)} topics")
+
 
 # -----------------------------------------------------------------------------
 

diff --git a/tests/bot/exts/__snapshots__/test_practice.ambr b/tests/bot/exts/__snapshots__/test_practice.ambr
@@ -209,16 +209,9 @@
       author=EmbedProxy(),
       color=<Colour value=15105570>,
       colour=<Colour value=15105570>,
-      description='
-        Today - Thursday, November 26 🦃
-
-        *There are no scheduled practices today!*
-
-        To schedule a practice, edit the schedule below or use the `!practice` command.
-        Example: `!practice today 2pm pst`
-
-      ',
+      description='Today - Thursday, November 26 🦃',
       fields=<class 'list'> [
+        EmbedProxy(inline=False, name='11 AM PST / 12 PM MST / 1 PM CST / 2 PM EST', value='[Add to Google Calendar](http://www.google.com/calendar/event?action=TEMPLATE&text=ASL+Practice%3A+Turkey+day+PRACTICE&dates=20201126T190000Z%2F20201126T200000Z&details=See+the+full+schedule+here%3A+https%3A%2F%2Fdocs.google.com%2Fspreadsheets%2Fd%2Fabc%2Fedit)\n> Notes: Turkey day PRACTICE'),
         EmbedProxy(inline=True, name='🗓 View or edit the schedule using the link below.', value='[Full schedule](https://docs.google.com/spreadsheets/d/abc/edit)'),
       ],
       footer=EmbedProxy(),

diff --git a/tests/bot/exts/test_practice.py b/tests/bot/exts/test_practice.py
@@ -39,6 +39,7 @@ async def mock_worksheet(monkeypatch, db):
             ["Wed 6pm edt", "", "another recurring", ""],
             ["9/26 2pm PDT 2020", "Steve", "one time", ""],
             ["Sunday, September 27 02:00 PM EDT 2020", "", "another 1️⃣", ""],
+            ["Nov 26 02:00 PM EDT 2020", "", "Turkey day PRACTICE", ""],
         ]
         mock_get_worksheet.return_value = mock_get_worksheet2.return_value = WorksheetMock
         yield WorksheetMock