diff --git a/main.py b/main.py deleted file mode 100644 index 91bed90..0000000 --- a/main.py +++ /dev/null @@ -1,67 +0,0 @@ -from flask import Flask,request,json,g -from tlapbot.db import get_db -import requests - -app = Flask(__name__) - -@app.route('/owncastWebhook',methods=['POST']) -def owncastWebhook(): - data = request.json - db = get_db() - if data["type"] == "CHAT": - print("New chat message:") - print(f'from {data["eventData"]["user"]["displayName"]}:') - print(f'{data["eventData"]["body"]}') - - elif data["type"] == "USER_JOINED": - user_id = data["eventData"]["user"]["id"] - try: - cursor = db.execute( - "SELECT username FROM points WHERE id = ?", - (user_id) - ) - if cursor.fetchone() == None: - username = data["eventData"]["user"]["displayName"] - cursor.execute( - "INSERT INTO points(id, username, points) VALUES(?, ?, 0)", - (user_id, username) - ) - db.commit() - - elif data["type"] == "NAME_CHANGED": - user_id = data["eventData"]["user"]["id"] - new_username = data["eventData"]["user"]["newName"] - try: - db.execute( - "UPDATE points SET username = ? WHERE id = ?", - (new_username, user_id), - ) - db.commit() - except db.IntegrityError: - print("Integrity Error.") - print(f"User ID {user_id} probably does not exist.") - - elif data["type"] == "CHAT": - user_id = data["eventData"]["user"]["id"] - if "!points" in data["body"]: - try: - cursor = db.execute( - "SELECT points FROM points WHERE id = ?", - (user_id), - ) - print("{}'s points: {}".format(user_id, cursor.fetchone())) - else: # DEBUG: give points for message - try: - db.execute( - "UPDATE points SET points = points + 10 WHERE id = ?", - (user_id), - ) - db.commit() - except sqlite3.Error as e: - print("Error occured:", e.args[0]) - print("To user:", user_id) - - return data - -if __name__ == '__main__': - app.run(debug=True) \ No newline at end of file diff --git a/tlapbot/__init__.py b/tlapbot/__init__.py index 1fa45f0..0edf3ba 100644 --- a/tlapbot/__init__.py +++ b/tlapbot/__init__.py @@ -25,9 +25,13 @@ def create_app(test_config=None): pass from . import db + from . import owncastWebhooks + app.register_blueprint(owncastWebhooks.bp) db.init_app(app) return app + + if __name__ == '__main__': create_app() \ No newline at end of file diff --git a/tlapbot/owncastWebhooks.py b/tlapbot/owncastWebhooks.py new file mode 100644 index 0000000..2065ff6 --- /dev/null +++ b/tlapbot/owncastWebhooks.py @@ -0,0 +1,82 @@ +from flask import Flask,request,json,g,Blueprint +from sqlite3 import Error +from tlapbot.db import get_db +import requests + +bp = Blueprint('owncastWebhooks', __name__) + +def userExists(user_id, db): + try: + cursor = db.execute( + "SELECT points FROM points WHERE id = ?", + (user_id,) + ) + if cursor.fetchone() == None: + return False + return True + except Error as e: + print("Error occured checking if user exists:", e.args[0]) + print("To user:", user_id) + +# only adds user if they aren't already in. +def addUserToDatabase(user_id, db): + try: + cursor = db.execute( + "SELECT points FROM points WHERE id = ?", + (user_id,) + ) + if cursor.fetchone() == None: + cursor.execute( + "INSERT INTO points(id, points) VALUES(?, 10)", + (user_id,) + ) + db.commit() + except Error as e: + print("Error occured adding user to db:", e.args[0]) + print("To user:", user_id) + +@bp.route('/owncastWebhook',methods=['POST']) +def owncastWebhook(): + data = request.json + db = get_db() + + if data["type"] == "USER_JOINED": + user_id = data["eventData"]["user"]["id"] + addUserToDatabase(user_id, db) + + elif data["type"] == "CHAT": + display_name = data["eventData"]["user"]["displayName"] + print("New chat message:") + print(f'from {display_name}:') + print(f'{data["eventData"]["body"]}') + user_id = data["eventData"]["user"]["id"] + + if "!points" in data["eventData"]["body"]: + if not userExists(user_id, db): + addUserToDatabase(user_id, db) + + try: + cursor = db.execute( + "SELECT points FROM points WHERE id = ?", + (user_id,) + ) + print("{}'s points: {}".format(display_name, cursor.fetchone()[0])) + except Error as e: + print("Error occured reading points:", e.args[0]) + print("To user:", user_id) + + else: # DEBUG: give points for message + try: + db.execute( + "UPDATE points SET points = points + 10 WHERE id = ?", + (user_id,) + ) + db.commit() + except Error as e: + print("Error occured giving DEBUG points:", e.args[0]) + print("To user:", user_id) + + return data + +if __name__ == '__main__': + app.run(debug=True) \ No newline at end of file diff --git a/tlapbot/schema.sql b/tlapbot/schema.sql index 18da7b8..b168c78 100644 --- a/tlapbot/schema.sql +++ b/tlapbot/schema.sql @@ -2,6 +2,5 @@ DROP TABLE IF EXISTS points; CREATE TABLE points ( id TEXT PRIMARY KEY, - username TEXT NOT NULL, points INTEGER ); \ No newline at end of file