From b3cbe42450086964bbff38ab879504581fb23a1b Mon Sep 17 00:00:00 2001 From: Lili Date: Sun, 6 Oct 2024 15:23:30 +0200 Subject: [PATCH] add types to owncast_webhooks --- tlapbot/owncast_webhooks.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/tlapbot/owncast_webhooks.py b/tlapbot/owncast_webhooks.py index 8c34da4..8e17c32 100644 --- a/tlapbot/owncast_webhooks.py +++ b/tlapbot/owncast_webhooks.py @@ -1,4 +1,6 @@ from flask import Flask, request, json, Blueprint, current_app +from sqlite3 import Connection +from typing import Any from tlapbot.db import get_db from tlapbot.owncast_requests import send_chat from tlapbot.owncast_helpers import (add_user_to_database, change_display_name, @@ -11,9 +13,12 @@ bp = Blueprint('owncast_webhooks', __name__) @bp.route('/owncastWebhook', methods=['POST']) -def owncast_webhook(): - data = request.json - db = get_db() +def owncast_webhook() -> Any | None: + """Reads webhook json -- adds new users, removes duplicate usernames, + handles name changes and chat messages with commands. + Returns the 'data' json from the request.""" + data: Any | None = request.json + db: Connection = get_db() # Make sure user is in db before doing anything else. if data["type"] in ["CHAT", "NAME_CHANGED", "USER_JOINED"]: