Tlapbot/tlapbot/owncast_helpers.py

233 lines
7.5 KiB
Python
Raw Normal View History

2022-08-24 13:39:40 +02:00
from flask import current_app
import requests
from sqlite3 import Error
from re import sub
2022-10-17 18:47:30 +02:00
# # # requests stuff # # #
def is_stream_live():
url = current_app.config['OWNCAST_INSTANCE_URL'] + '/api/status'
r = requests.get(url)
return r.json()["online"]
2022-10-17 18:47:30 +02:00
def give_points_to_chat(db):
url = current_app.config['OWNCAST_INSTANCE_URL'] + '/api/integrations/clients'
headers = {"Authorization": "Bearer " + current_app.config['OWNCAST_ACCESS_TOKEN']}
r = requests.get(url, headers=headers)
unique_users = set(map(lambda user_object: user_object["user"]["id"], r.json()))
for user_id in unique_users:
give_points_to_user(db,
user_id,
current_app.config['POINTS_AMOUNT_GIVEN'])
2022-10-17 18:47:30 +02:00
def send_chat(message):
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()
def remove_emoji(message):
return sub(
r'<img class="emoji" alt="(:.*:)" title=":.*:" src="/img/emoji/.*">',
r'\1',
message
)
# # # db stuff # # #
2022-08-24 15:35:16 +02:00
def read_users_points(db, user_id):
2022-10-18 14:09:35 +02:00
"""Errors out if user doesn't exist."""
2022-08-24 15:35:16 +02:00
try:
cursor = db.execute(
"SELECT points FROM points WHERE id = ?",
(user_id,)
)
return cursor.fetchone()[0]
except Error as e:
current_app.logger.error(f"Error occured reading points: {e.args[0]}")
current_app.logger.error(f"To user: {user_id}")
2022-08-24 15:35:16 +02:00
2022-10-17 18:47:30 +02:00
def read_all_users_with_username(db, username):
2022-10-18 14:04:15 +02:00
try:
cursor = db.execute(
"SELECT name, points FROM points WHERE name = ?",
2022-10-18 14:04:15 +02:00
(username,)
)
users = cursor.fetchall()
return users
2022-10-18 14:04:15 +02:00
except Error as e:
current_app.logger.error(f"Error occured reading points by username: {e.args[0]}")
current_app.logger.error(f"To everyone with username: {username}")
2022-10-18 14:04:15 +02:00
2022-11-22 11:33:05 +01:00
2022-08-24 15:35:16 +02:00
def give_points_to_user(db, user_id, points):
try:
db.execute(
"UPDATE points SET points = points + ? WHERE id = ?",
(points, user_id,)
2022-08-24 15:35:16 +02:00
)
db.commit()
except Error as e:
current_app.logger.error(f"Error occured giving points: {e.args[0]}")
current_app.logger.error(f"To user: {user_id} amount of points: {points}")
2022-08-24 15:55:46 +02:00
2022-10-17 18:47:30 +02:00
2022-08-24 15:55:46 +02:00
def use_points(db, user_id, points):
try:
db.execute(
"UPDATE points SET points = points - ? WHERE id = ?",
(points, user_id,)
2022-08-24 15:55:46 +02:00
)
db.commit()
return True
2022-08-24 15:55:46 +02:00
except Error as e:
current_app.logger.error(f"Error occured using points: {e.args[0]}")
current_app.logger.error(f"From user: {user_id} amount of points: {points}")
return False
2022-08-24 15:35:16 +02:00
2022-10-17 18:47:30 +02:00
2022-08-24 15:35:16 +02:00
def user_exists(db, user_id):
2022-08-24 13:39:40 +02:00
try:
cursor = db.execute(
"SELECT points FROM points WHERE id = ?",
(user_id,)
)
2022-10-17 18:47:30 +02:00
if cursor.fetchone() is None:
2022-08-24 13:39:40 +02:00
return False
return True
except Error as e:
current_app.logger.error(f"Error occured checking if user exists: {e.args[0]}")
current_app.logger.error(f"To user: {user_id}")
2022-08-24 13:39:40 +02:00
2022-10-17 18:47:30 +02:00
def add_user_to_database(db, user_id, display_name):
2022-10-17 18:47:30 +02:00
""" Adds a new user to the database. Does nothing if user is already in."""
2022-08-24 13:39:40 +02:00
try:
cursor = db.execute(
"SELECT points, name FROM points WHERE id = ?",
2022-08-24 13:39:40 +02:00
(user_id,)
)
user = cursor.fetchone()
if user is None:
2022-08-24 13:39:40 +02:00
cursor.execute(
"INSERT INTO points(id, name, points) VALUES(?, ?, 10)",
(user_id, display_name)
2022-08-24 13:39:40 +02:00
)
2022-11-22 11:33:05 +01:00
if user is not None and user[1] is None:
cursor.execute(
"""UPDATE points
SET name = ?
WHERE id = ?""",
(display_name, user_id)
)
2022-08-24 13:39:40 +02:00
db.commit()
except Error as e:
current_app.logger.error(f"Error occured adding user to db: {e.args[0]}")
current_app.logger.error(f"To user id: {user_id}, with display name: {display_name}")
2022-10-17 18:47:30 +02:00
def change_display_name(db, user_id, new_name):
try:
cursor = db.execute(
"UPDATE points SET name = ? WHERE id = ?",
(new_name, user_id)
)
db.commit()
except Error as e:
current_app.logger.error(f"Error occured changing display name: {e.args[0]}")
current_app.logger.error(f"To user id: {user_id}, with display name: {new_name}")
2022-08-24 13:39:40 +02:00
def check_counter_exists(db, counter_name):
try:
cursor = db.execute(
"SELECT count FROM counters WHERE name = ?",
(counter_name,)
)
counter = cursor.fetchone()
if counter is None:
current_app.logger.warning("Counter not found in database.")
current_app.logger.warning("Maybe you forgot to run the refresh-counters CLI command "
"after you added a new counter to the config?")
return False
return True
except Error as e:
current_app.logger.error(f"Error occured checking if counter exists: {e.args[0]}")
current_app.logger.error(f"For counter: {counter_name}")
def add_to_counter(db, counter_name):
if check_counter_exists(db, counter_name):
try:
cursor = db.execute(
"UPDATE counters SET count = count + 1 WHERE name = ?",
(counter_name,)
)
db.commit()
return True
except Error as e:
current_app.logger.error(f"Error occured adding to counter: {e.args[0]}")
current_app.logger.error(f"To counter: {counter_name}")
return False
2022-10-17 18:47:30 +02:00
def add_to_redeem_queue(db, user_id, redeem_name, note=None):
try:
cursor = db.execute(
"INSERT INTO redeem_queue(redeem, redeemer_id, note) VALUES(?, ?, ?)",
(redeem_name, user_id, note)
)
db.commit()
return True
except Error as e:
current_app.logger.error(f"Error occured adding to redeem queue: {e.args[0]}")
current_app.logger.error(f"To user: {user_id} with redeem: {redeem_name} with note: {note}")
return False
2022-10-17 18:47:30 +02:00
def all_counters(db):
try:
cursor = db.execute(
"""SELECT counters.name, counters.count FROM counters"""
)
return cursor.fetchall()
except Error as e:
current_app.logger.error(f"Error occured selecting all counters: {e.args[0]}")
2022-10-17 18:47:30 +02:00
def pretty_redeem_queue(db):
try:
cursor = db.execute(
"""SELECT redeem_queue.created, redeem_queue.redeem, redeem_queue.note, points.name
FROM redeem_queue
INNER JOIN points
on redeem_queue.redeemer_id = points.id"""
)
return cursor.fetchall()
except Error as e:
current_app.logger.error(f"Error occured selecting pretty redeem queue: {e.args[0]}")
2022-10-17 18:47:30 +02:00
def whole_redeem_queue(db):
try:
cursor = db.execute(
"SELECT * from redeem_queue"
)
return cursor.fetchall()
except Error as e:
current_app.logger.error(f"Error occured selecting redeem queue: {e.args[0]}")
2022-09-26 11:10:48 +02:00
2022-10-17 18:47:30 +02:00
def remove_duplicate_usernames(db, user_id, username):
try:
cursor = db.execute(
"""UPDATE points
SET name = NULL
WHERE name = ? AND NOT id = ?""",
2022-11-22 11:33:05 +01:00
(username, user_id)
)
db.commit()
except Error as e:
current_app.logger.error(f"Error occured removing duplicate usernames: {e.args[0]}")