split out requests to owncast_requests.py
This commit is contained in:
parent
d6297c537f
commit
7235340a39
|
@ -3,8 +3,8 @@ import logging
|
||||||
from flask import Flask
|
from flask import Flask
|
||||||
from apscheduler.schedulers.background import BackgroundScheduler
|
from apscheduler.schedulers.background import BackgroundScheduler
|
||||||
from tlapbot.db import get_db
|
from tlapbot.db import get_db
|
||||||
from tlapbot.owncast_helpers import (is_stream_live, give_points_to_chat,
|
from tlapbot.owncast_requests import is_stream_live, give_points_to_chat
|
||||||
remove_inactive_redeems)
|
from tlapbot.owncast_helpers import remove_inactive_redeems
|
||||||
|
|
||||||
def create_app(test_config=None):
|
def create_app(test_config=None):
|
||||||
app = Flask(__name__, instance_relative_config=True)
|
app = Flask(__name__, instance_relative_config=True)
|
||||||
|
|
|
@ -1,5 +1,5 @@
|
||||||
from flask import current_app
|
from flask import current_app
|
||||||
from tlapbot.owncast_helpers import send_chat
|
from tlapbot.owncast_requests import send_chat
|
||||||
|
|
||||||
|
|
||||||
def send_help():
|
def send_help():
|
||||||
|
|
|
@ -1,37 +1,7 @@
|
||||||
from flask import current_app
|
from flask import current_app
|
||||||
import requests
|
|
||||||
from sqlite3 import Error
|
from sqlite3 import Error
|
||||||
from re import sub
|
from re import sub
|
||||||
|
|
||||||
# # # requests stuff # # #
|
|
||||||
def is_stream_live():
|
|
||||||
url = current_app.config['OWNCAST_INSTANCE_URL'] + '/api/status'
|
|
||||||
try:
|
|
||||||
r = requests.get(url)
|
|
||||||
except requests.exceptions.RequestException as e:
|
|
||||||
current_app.logger.error(f"Error occured checking if stream is live: {e.args[0]}")
|
|
||||||
return False
|
|
||||||
return r.json()["online"]
|
|
||||||
|
|
||||||
|
|
||||||
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'])
|
|
||||||
|
|
||||||
|
|
||||||
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()
|
|
||||||
|
|
||||||
|
|
||||||
# # # db stuff # # #
|
# # # db stuff # # #
|
||||||
def read_users_points(db, user_id):
|
def read_users_points(db, user_id):
|
||||||
"""Errors out if user doesn't exist."""
|
"""Errors out if user doesn't exist."""
|
||||||
|
|
|
@ -0,0 +1,32 @@
|
||||||
|
import requests
|
||||||
|
from flask import current_app
|
||||||
|
from tlapbot.owncast_helpers import give_points_to_user
|
||||||
|
|
||||||
|
|
||||||
|
def is_stream_live():
|
||||||
|
url = current_app.config['OWNCAST_INSTANCE_URL'] + '/api/status'
|
||||||
|
try:
|
||||||
|
r = requests.get(url)
|
||||||
|
except requests.exceptions.RequestException as e:
|
||||||
|
current_app.logger.error(f"Error occured checking if stream is live: {e.args[0]}")
|
||||||
|
return False
|
||||||
|
return r.json()["online"]
|
||||||
|
|
||||||
|
|
||||||
|
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'])
|
||||||
|
|
||||||
|
|
||||||
|
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()
|
||||||
|
|
|
@ -1,7 +1,8 @@
|
||||||
from flask import Flask, request, json, Blueprint, current_app
|
from flask import Flask, request, json, Blueprint, current_app
|
||||||
from tlapbot.db import get_db
|
from tlapbot.db import get_db
|
||||||
|
from tlapbot.owncast_requests import send_chat
|
||||||
from tlapbot.owncast_helpers import (add_user_to_database, change_display_name,
|
from tlapbot.owncast_helpers import (add_user_to_database, change_display_name,
|
||||||
send_chat, read_users_points, remove_duplicate_usernames)
|
read_users_points, remove_duplicate_usernames)
|
||||||
from tlapbot.help_message import send_help
|
from tlapbot.help_message import send_help
|
||||||
from tlapbot.redeems_handler import handle_redeem
|
from tlapbot.redeems_handler import handle_redeem
|
||||||
|
|
||||||
|
|
|
@ -1,7 +1,8 @@
|
||||||
from flask import current_app
|
from flask import current_app
|
||||||
from tlapbot.db import get_db
|
from tlapbot.db import get_db
|
||||||
|
from tlapbot.owncast_requests import send_chat
|
||||||
from tlapbot.owncast_helpers import (use_points, add_to_redeem_queue,
|
from tlapbot.owncast_helpers import (use_points, add_to_redeem_queue,
|
||||||
add_to_counter, read_users_points, send_chat, remove_emoji,
|
add_to_counter, read_users_points, remove_emoji,
|
||||||
add_to_milestone, check_apply_milestone_completion, milestone_complete)
|
add_to_milestone, check_apply_milestone_completion, milestone_complete)
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue