Tlapbot/tlapbot/owncast_helpers.py

from flask import current_app
import requests
from sqlite3 import Error

def read_users_points(db, user_id):
    try:
        cursor = db.execute(
            "SELECT points FROM points WHERE id = ?",
            (user_id,)
        )
        return cursor.fetchone()[0]
    except Error as e:
        print("Error occured reading points:", e.args[0])
        print("To user:", user_id)

def give_points_to_user(db, user_id, points):
    try:
        db.execute(
        "UPDATE points SET points = points + ? WHERE id = ?",
        (points, user_id,)
        )
        db.commit()
    except Error as e:
        print("Error occured giving points:", e.args[0])
        print("To user:", user_id, "  amount of points:", points)

def use_points(db, user_id, points):
    try:
        db.execute(
        "UPDATE points SET points = points - ? WHERE id = ?",
        (points, user_id,)
        )
        db.commit()
        return True
    except Error as e:
        print("Error occured using points:", e.args[0])
        print("From user:", user_id, "  amount of points:", points)
        return False

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.post(url, headers=headers)
    for user_object in r.json():
        give_points_to_user(db,
                            user_object["user"]["id"],
                            current_app.config['POINTS_AMOUNT_GIVEN'])

def user_exists(db, user_id):
    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)

""" Adds a new user to the database. Does nothing if user is already in."""
def add_user_to_database(db, user_id, display_name):
    try:
        cursor = db.execute(
            "SELECT points FROM points WHERE id = ?",
            (user_id,)
        )
        if cursor.fetchone() == None:
            cursor.execute(
                "INSERT INTO points(id, name, points) VALUES(?, ?, 10)",
                (user_id, display_name)
            )
        db.commit()
    except Error as e:
        print("Error occured adding user to db:", e.args[0])
        print("To user:", user_id, display_name)

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:
        print("Error occured changing display name:", e.args[0])
        print("To user:", user_id, new_name)

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 add_to_redeem_queue(db, user_id, redeem_name):
    try:
        cursor = db.execute(
                "INSERT INTO redeem_queue(redeem, redeemer_id) VALUES(?, ?)",
                (redeem_name, user_id)
            )
        db.commit()
    except Error as e:
        print("Error occured adding to redeem queue:", e.args[0])
        print("To user:", user_id, " with redeem:", redeem_name)

def clear_redeem_queue(db):
    try:
        cursor = db.execute(
                "DELETE FROM redeem_queue"
            )
        db.commit()
    except Error as e:
        print("Error occured deleting redeem queue:", e.args[0])

def pretty_redeem_queue(db):
    try:
        cursor = db.execute(
            """SELECT redeem_queue.created, redeem_queue.redeem, points.name
            FROM redeem_queue
            INNER JOIN points
            on redeem_queue.redeemer_id = points.id"""
        )
        return cursor.fetchall()
    except Error as e:
        print("Error occured selecting pretty redeem queue:", e.args[0])

def whole_redeem_queue(db):
    try:
        cursor = db.execute(
                "SELECT * from redeem_queue"
            )
        return cursor.fetchall()
    except Error as e:
        print("Error occured selecting redeem queue:", e.args[0])