DockOfOurOwn/app.py

#! /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

# 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={}):
    return chevron.render(template=open('index.html', 'r'), data=data)


@app.route("/api/serverstatus", methods=['GET'])
def getServerStatus():
    response = session.get(stream_data['stream_url'] + '/api/yp')
    response_data = response.json()
    return json.dumps({
        'name': response_data['name'],
        'online': response_data['online'],
        'overallMaxViewerCount': response_data['overallMaxViewerCount'],
        'sessionMaxViewerCount': response_data['sessionMaxViewerCount'],
        'streamTitle': response_data['streamTitle'],
        'viewerCount': response_data['viewerCount'],
        'description': response_data['description'],
        'tags': response_data['tags'],
        'nsfw': response_data['nsfw'],
    })


@app.route("/api/update/streamtitle", methods=['POST'])
def updateStreamTitle():
    response = session.post(
        stream_data['stream_url'] + '/api/admin/config/streamtitle',
        data=json.dumps({'value': request.json()})
    )
    print(response.json())
    return response.status_code


@app.route('/api/update/servertags', methods=['POST'])
def updateServerTags(tags_list: list):
    response = session.post(
        stream_data['stream_url'] + '/api/admin/config/tags',
        data=json.dumps({
            'value': tags_list
        })
    )
    return response.status_code, json.loads(response.text)


@app.route('/api/update/nsfw', methods=['POST'])
def updateServerNSFW(boolean: bool):
    response = session.post(
        stream_data['stream_url'] + '/api/admin/config/nsfw',
        data=json.dumps({
            'value': boolean
        })
    )
    return response.status_code, json.loads(response.text)


@app.route("/")
def index():
    return render(json.loads(getServerStatus()))


if __name__ == '__main__':
    app.run()