Weekly codewar challenge (#447)

Author: ovv

migrations/versions/2021-01-04T20-23-50Z_dece77ed5036_init_codewars_challenge.py

@@ -0,0 +1,66 @@
+"""init codewars challenge
+
+Revision ID: dece77ed5036
+Revises: 2ba5cc3efce8
+Create Date: 2021-01-04 20:23:50.479134+00:00
+
+"""
+from datetime import datetime
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = "dece77ed5036"
+down_revision = "2ba5cc3efce8"
+branch_labels = None
+depends_on = None
+
+QUESTIONS = [
+    "550f22f4d758534c1100025a",
+    "525f3eda17c7cd9f9e000b39",
+    "54521e9ec8e60bc4de000d6c",
+    "52bb6539a4cf1b12d90005b7",
+    "5296bc77afba8baa690002d7",
+    "5235c913397cbf2508000048",
+    "521c2db8ddc89b9b7a0000c1",
+    "55983863da40caa2c900004e",
+    "52742f58faf5485cae000b9a",
+    "52a78825cdfc2cfc87000005",
+    "54acd76f7207c6a2880012bb",
+    "5679d5a3f2272011d700000d",
+]
+
+codewars = sa.sql.table(
+    "codewars_challenge",
+    sa.Column("id", sa.Text, primary_key=True),
+    sa.Column("added_at", sa.DateTime(timezone=True), default=datetime.now, nullable=False),
+    sa.Column(
+        "posted_at",
+        sa.DateTime(timezone=True),
+        default=None,
+        onupdate=datetime.now,
+        nullable=True,
+    ),
+)
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table(
+        "codewars_challenge",
+        sa.Column("id", sa.Text(), nullable=False),
+        sa.Column("added_at", sa.DateTime(timezone=True), nullable=False, default=datetime.utcnow),
+        sa.Column("posted_at", sa.DateTime(timezone=True), nullable=True),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    # ### end Alembic commands ###
+
+    op.bulk_insert(codewars, [{"id": q} for q in QUESTIONS])
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_table("codewars_challenge")
+    # ### end Alembic commands ###

pyslackersweb/sirbot/__init__.py

@@ -1,6 +1,7 @@
 from aiohttp import web
 
 from .views import routes
+from .context import background_jobs
 
 
 async def app_factory() -> web.Application:
@@ -12,4 +13,7 @@ async def app_factory() -> web.Application:
     )
 
     sirbot.add_routes(routes)
+
+    sirbot.cleanup_ctx.extend([background_jobs])
+
     return sirbot

pyslackersweb/sirbot/context.py

@@ -0,0 +1,25 @@
+import datetime
+
+from typing import AsyncGenerator
+
+import pytz
+
+from aiohttp import web
+
+from . import tasks
+
+
+async def background_jobs(app: web.Application) -> AsyncGenerator[None, None]:
+    scheduler = app["scheduler"]
+    pg = app["pg"]
+    slack_client = app["slack_client"]
+
+    scheduler.add_job(
+        tasks.post_slack_codewars_challenge,
+        "interval",
+        start_date=datetime.datetime(year=2021, month=1, day=7, hour=18, tzinfo=pytz.UTC),
+        weeks=1,
+        args=(slack_client, pg),
+    )
+
+    yield

pyslackersweb/sirbot/database.py

@@ -0,0 +1,17 @@
+import logging
+
+import asyncpg
+
+from pyslackersweb.util.log import ContextAwareLoggerAdapter
+
+
+logger = ContextAwareLoggerAdapter(logging.getLogger(__name__))
+
+
+async def get_challenge(conn: asyncpg.connection.Connection) -> str:
+    return await conn.fetchval(
+        """
+UPDATE codewars_challenge SET posted_at=now() WHERE id=(
+        SELECT id FROM codewars_challenge WHERE posted_at IS NULL ORDER BY RANDOM() LIMIT 1
+) RETURNING id""",
+    )

pyslackersweb/sirbot/models.py

@@ -0,0 +1,19 @@
+from datetime import datetime
+
+import sqlalchemy as sa
+
+from pyslackersweb.models import metadata
+
+codewars = sa.Table(
+    "codewars_challenge",
+    metadata,
+    sa.Column("id", sa.Text, primary_key=True),
+    sa.Column("added_at", sa.DateTime(timezone=True), default=datetime.now, nullable=False),
+    sa.Column(
+        "posted_at",
+        sa.DateTime(timezone=True),
+        default=None,
+        onupdate=datetime.now,
+        nullable=True,
+    ),
+)

pyslackersweb/sirbot/tasks.py

@@ -0,0 +1,49 @@
+import logging
+
+import slack
+import asyncpg.pool
+
+from slack.events import Message
+from slack.io.abc import SlackAPI
+
+from pyslackersweb.util.log import ContextAwareLoggerAdapter
+
+from . import database
+
+logger = ContextAwareLoggerAdapter(logging.getLogger(__name__))
+
+
+async def post_slack_codewars_challenge(
+    slack_client: SlackAPI,
+    pg: asyncpg.pool.Pool,
+) -> None:
+    async with pg.acquire() as conn:
+        challenge = await database.get_challenge(conn)
+
+    message = Message()
+    message["channel"] = "CEFJ9TJNL"  # advent_of_code
+    if challenge:
+        message["attachments"] = [
+            {
+                "fallback": f"Codewars challenge: https://www.codewars.com/kata/{challenge}",
+                "color": "#ff0000",
+                "title": "Weekly Codewars challenge",
+                "title_link": f"https://www.codewars.com/kata/{challenge}",
+                "text": "Discuss in this thread",
+                "footer": "Codewars",
+                "footer_icon": "https://codewars.com/favicon.ico",
+            }
+        ]
+    else:
+        message["attachments"] = [
+            {
+                "fallback": "Codewars challenge: no challenge found",
+                "color": "#ff0000",
+                "title": "Weekly Codewars challenge",
+                "title_link": "https://www.codewars.com",
+                "text": "No challenge found :sad-panda:, please seed new challenges ",
+                "footer": "Codewars",
+                "footer_icon": "https://codewars.com/favicon.ico",
+            }
+        ]
+    await slack_client.query(slack.methods.CHAT_POST_MESSAGE, data=message)

tests/conftest.py

@@ -36,3 +36,4 @@ async def client(monkeypatch, aiohttp_client, slack_client_ctx):
         await conn.fetch(delete(pyslackersweb.models.domains))
         await conn.fetch(delete(pyslackersweb.models.SlackUsers))
         await conn.fetch(delete(pyslackersweb.models.SlackChannels))
+        await conn.fetch(delete(pyslackersweb.sirbot.models.codewars))

tests/test_sirbot.py

@@ -1,8 +1,10 @@
+import json
 import pytest
 
 from slack import methods
 
-from pyslackersweb.sirbot import settings
+from pyslackersweb.sirbot import settings, tasks, models
+from sqlalchemy.dialects.postgresql import insert as pg_insert
 
 
 @pytest.mark.parametrize(
@@ -57,3 +59,29 @@ async def test_readthedocs_notification_missing_name(client):
 
     # Assert we did not send a message to slack
     assert not client.app["slack_client"]._request.called
+
+
+async def test_task_codewars_challenge(client, caplog):
+    await tasks.post_slack_codewars_challenge(client.app["slack_client"], client.app["pg"])
+
+    mocked_request = client.app["slack_client"]._request
+    mocked_request.assert_called_once()
+    mocked_request_args = mocked_request.call_args.args
+
+    payload = json.loads(mocked_request_args[3])
+    assert payload["channel"] == "CEFJ9TJNL"
+    assert "No challenge found" in payload["attachments"][0]["text"]
+
+    async with client.app["pg"].acquire() as conn:
+        await conn.execute(pg_insert(models.codewars).values(id="foo"))
+
+    client.app["slack_client"]._request.reset_mock()
+    await tasks.post_slack_codewars_challenge(client.app["slack_client"], client.app["pg"])
+
+    mocked_request = client.app["slack_client"]._request
+    mocked_request.assert_called_once()
+    mocked_request_args = mocked_request.call_args.args
+
+    payload = json.loads(mocked_request_args[3])
+    assert payload["channel"] == "CEFJ9TJNL"
+    assert payload["attachments"][0]["title_link"] == "https://www.codewars.com/kata/foo"