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 = ?", (user_id, new_name,) ) 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])