move helper functions to own file

This commit is contained in:
Lili (Tlapka) 2022-08-24 13:39:40 +02:00
parent ad692e5938
commit 2ab85bfb6d
2 changed files with 42 additions and 46 deletions

39
tlapbot/owncastHelpers.py Normal file
View File

@ -0,0 +1,39 @@
from flask import current_app
import requests
from sqlite3 import Error
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)
def sendChat(message): # TODO: url to constant?
url = current_app.config['OWNCAST_INSTANCE_URL'] + '/api/integrations/chat/send'
headers = {"Authorization": "Bearer " + current_app.config['OWNCAST_ACCESS_TOKEN']}
r = requests.post(url, headers=headers, json={"body": message})
return r.json()

View File

@ -1,66 +1,26 @@
from flask import Flask,request,json,g,Blueprint,current_app from flask import Flask,request,json,Blueprint
from sqlite3 import Error from sqlite3 import Error
from tlapbot.db import get_db from tlapbot.db import get_db
import requests from tlapbot.owncastHelpers import userExists, addUserToDatabase, sendChat
bp = Blueprint('owncastWebhooks', __name__) 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)
def sendChat(message): # TODO: url to constant?
url = current_app.config['OWNCAST_INSTANCE_URL'] + '/api/integrations/chat/send'
headers = {"Authorization": "Bearer " + current_app.config['OWNCAST_ACCESS_TOKEN']}
r = requests.post(url, headers=headers, json={"body": message})
return r.json()
@bp.route('/owncastWebhook',methods=['POST']) @bp.route('/owncastWebhook',methods=['POST'])
def owncastWebhook(): def owncastWebhook():
data = request.json data = request.json
db = get_db() db = get_db()
if data["type"] == "USER_JOINED": if data["type"] == "USER_JOINED":
user_id = data["eventData"]["user"]["id"] user_id = data["eventData"]["user"]["id"]
addUserToDatabase(user_id, db) addUserToDatabase(user_id, db)
elif data["type"] == "CHAT": elif data["type"] == "CHAT":
display_name = data["eventData"]["user"]["displayName"] display_name = data["eventData"]["user"]["displayName"]
print("New chat message:") print("New chat message:")
print(f'from {display_name}:') print(f'from {display_name}:')
print(f'{data["eventData"]["body"]}') print(f'{data["eventData"]["body"]}')
user_id = data["eventData"]["user"]["id"] user_id = data["eventData"]["user"]["id"]
if "!points" in data["eventData"]["body"]: if "!points" in data["eventData"]["body"]:
if not userExists(user_id, db): if not userExists(user_id, db):
addUserToDatabase(user_id, db) addUserToDatabase(user_id, db)
try: try:
cursor = db.execute( cursor = db.execute(
"SELECT points FROM points WHERE id = ?", "SELECT points FROM points WHERE id = ?",
@ -69,11 +29,9 @@ def owncastWebhook():
message = "{}'s points: {}".format(display_name, cursor.fetchone()[0]) message = "{}'s points: {}".format(display_name, cursor.fetchone()[0])
print(message) print(message)
sendChat(message) sendChat(message)
except Error as e: except Error as e:
print("Error occured reading points:", e.args[0]) print("Error occured reading points:", e.args[0])
print("To user:", user_id) print("To user:", user_id)
else: # DEBUG: give points for message else: # DEBUG: give points for message
try: try:
db.execute( db.execute(
@ -84,5 +42,4 @@ def owncastWebhook():
except Error as e: except Error as e:
print("Error occured giving DEBUG points:", e.args[0]) print("Error occured giving DEBUG points:", e.args[0])
print("To user:", user_id) print("To user:", user_id)
return data return data