arianagiroux
03f2f966d7
- removes the nsfw tag function - adds functionality to the tags update endpoint to process raw data from client
133 lines
4 KiB
Python
Executable file
133 lines
4 KiB
Python
Executable file
#! /Users/arianagiroux/Documents/Coding/owncast-obs-streamer/venv/bin/python3
|
|
""" A simple flask app to provide a streamer UI for obs and owncast.
|
|
|
|
Usage:
|
|
$ ./app.py
|
|
> * Serving Flask app 'app' (lazy loading)
|
|
> * Environment: production
|
|
> WARNING: This is a development server. Do not use it in a production deployment.
|
|
> Use a production WSGI server instead.
|
|
> * Debug mode: off
|
|
> * Running on http://127.0.0.1:5000 (Press CTRL+C to quit)
|
|
|
|
> Open 127.0.0.1:5000 in your local browser
|
|
|
|
Note:
|
|
If port 5000 is already bound, use the following command to start the flask
|
|
app:
|
|
|
|
$ flask run -p INTEGER
|
|
|
|
The app will automatically fail if there is not file containing valid JSON data
|
|
located at ./resources/data.json. This json file should have the following data
|
|
structure:
|
|
|
|
{
|
|
"stream_url":"https://yourstream.url",
|
|
"user_name":"admin",
|
|
"stream_key":"your_stream_key"
|
|
}
|
|
"""
|
|
|
|
import chevron
|
|
import requests
|
|
import json
|
|
from os import path
|
|
from pprint import pprint
|
|
from flask import Flask, request, Response
|
|
|
|
# load json data, or raise exception.
|
|
if path.exists('resources/data.json'):
|
|
try:
|
|
stream_data = json.load(open('resources/data.json', 'r'))
|
|
except json.JSONDecodeError as e:
|
|
raise RuntimeError('JSON data did not pass validation.')
|
|
else:
|
|
raise RuntimeError('Could not find ./resources/data.json, required for '
|
|
'runtime!')
|
|
|
|
# initialize requests session
|
|
session = requests.Session()
|
|
session.auth = ('admin', stream_data['stream_key'])
|
|
|
|
# initialize flask app
|
|
app = Flask(__name__)
|
|
|
|
|
|
def render(data={}):
|
|
""" Template rendering function using mustache (via the pypi chevron
|
|
implementation.
|
|
|
|
:returns: A full HTML template with relevant data from the server.
|
|
"""
|
|
return chevron.render(template=open('index.mustache', 'r'), data=data)
|
|
|
|
|
|
@app.route("/api/serverstatus", methods=['GET'])
|
|
def getServerStatus():
|
|
""" Obtains useful information from the server from either /api/yp or
|
|
/api/status.
|
|
|
|
:returns: the collected data as a JSON formatted dict.
|
|
"""
|
|
response = session.get(stream_data['stream_url'] + '/api/yp')
|
|
response_data = response.json()
|
|
data = {
|
|
'name': response_data['name'],
|
|
'online': response_data['online'],
|
|
'overallMaxViewerCount': response_data['overallMaxViewerCount'],
|
|
'sessionMaxViewerCount': response_data['sessionMaxViewerCount'],
|
|
'viewerCount': response_data['viewerCount'],
|
|
'description': response_data['description'],
|
|
'tags': response_data['tags'],
|
|
'nsfw': response_data['nsfw'],
|
|
}
|
|
|
|
response = session.get(stream_data['stream_url'] + '/api/status')
|
|
response_data = response.json()
|
|
|
|
data['streamTitle'] = response_data['streamTitle']
|
|
|
|
return json.dumps(data)
|
|
|
|
|
|
@app.route("/api/update/streamtitle", methods=['POST'])
|
|
def updateStreamTitle():
|
|
""" An endpoint to allow the user to update the stream title.
|
|
|
|
:returns: the status code of the post request made to the server.
|
|
"""
|
|
response = session.post(
|
|
stream_data['stream_url'] + '/api/admin/config/streamtitle',
|
|
data=json.dumps({'value': request.get_json(force=True)})
|
|
)
|
|
return Response(status=response.status_code)
|
|
|
|
|
|
@app.route('/api/update/servertags', methods=['POST'])
|
|
def updateServerTags():
|
|
""" An endpoint to allow the user to update the stream title.
|
|
|
|
:returns: the status code of the post request made to the server.
|
|
"""
|
|
values = [i.strip() for i in request.get_json(force=True).split(',')]
|
|
response = session.post(
|
|
stream_data['stream_url'] + '/api/admin/config/tags',
|
|
data=json.dumps({
|
|
'value': values
|
|
})
|
|
)
|
|
return Response(status=response.status_code)
|
|
|
|
|
|
@app.route("/")
|
|
def index():
|
|
""" A simple initial endpoint that loads in relevant data from the server.
|
|
|
|
:returns: the rendered template (see app.render for more)
|
|
"""
|
|
return render(json.loads(getServerStatus()))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
app.run()
|