164 lines
4.6 KiB
Python
164 lines
4.6 KiB
Python
import os
|
|
from glob import glob
|
|
from time import sleep
|
|
from flask_cors import CORS
|
|
from flask import Flask, jsonify, request, redirect, render_template, Response
|
|
import Common
|
|
import threading
|
|
from liveDownloader import run as RUN
|
|
|
|
app = Flask("liveStatus")
|
|
app.config['JSON_AS_ASCII'] = False
|
|
CORS(app, supports_credentials=True)
|
|
# url_for('static', filename='index.html')
|
|
# url_for('static', filename='index.js')
|
|
|
|
|
|
@app.route("/")
|
|
def index():
|
|
return redirect("/static/index.html")
|
|
|
|
@app.route("/config", methods=["GET"])
|
|
def readConfig():
|
|
config = Common.config.copy()
|
|
config.pop("b_p")
|
|
config.pop("mtd")
|
|
config.pop("del")
|
|
config.pop("mv")
|
|
return jsonify(config)
|
|
|
|
|
|
@app.route("/config", methods=["POST"])
|
|
def writeConfig():
|
|
# TODO : 完善
|
|
Common.reloadConfig()
|
|
return jsonify({"message":"ok","code":200,"status":0,"data":request.form})
|
|
|
|
|
|
@app.route("/encode/insert", methods=["POST"])
|
|
def insertEncode():
|
|
if "filename" in request.form:
|
|
Common.encodeQueue.put(request.form["filename"])
|
|
return jsonify({"message":"ok","code":200,"status":0})
|
|
|
|
|
|
@app.route("/upload/insert", methods=["POST"])
|
|
def insertUpload():
|
|
if "filename" in request.form:
|
|
Common.uploadQueue.put(request.form["filename"])
|
|
return jsonify({"message":"ok","code":200,"status":0})
|
|
|
|
|
|
@app.route("/upload/finish", methods=["POST"])
|
|
def finishUpload():
|
|
Common.uploadQueue.put(True)
|
|
return jsonify({"message":"ok","code":200,"status":0})
|
|
|
|
|
|
@app.route("/stats", methods=["GET"])
|
|
def getAllStats():
|
|
return jsonify({"message":"ok","code":200,"status":0,"data":{
|
|
"download":Common.downloadStatus,
|
|
"encode": Common.encodeStatus,
|
|
"encodeQueueSize": Common.encodeQueue.qsize(),
|
|
"upload": Common.uploadStatus,
|
|
"uploadQueueSize": Common.uploadQueue.qsize(),
|
|
"error": Common.errors,
|
|
"broadcast": {
|
|
"broadcaster": Common.broadcaster.__str__(),
|
|
"isBroadcasting": Common.isBroadcasting,
|
|
"streamUrl": Common.streamUrl,
|
|
"updateTime": Common.updateTime
|
|
},
|
|
"config": {
|
|
"forceNotBroadcasting": Common.forceNotBroadcasting,
|
|
"forceStopDownload": Common.forceStopDownload,
|
|
"forceNotUpload": Common.forceNotUpload,
|
|
"forceNotEncode": Common.forceNotEncode,
|
|
}
|
|
}})
|
|
|
|
|
|
@app.route("/stats/broadcast", methods=["GET"])
|
|
def getBroadcastStats():
|
|
return jsonify({"message":"ok","code":200,"status":0,"data":{
|
|
"broadcast": {
|
|
"broadcaster": Common.broadcaster.__str__(),
|
|
"isBroadcasting": Common.isBroadcasting,
|
|
"streamUrl": Common.streamUrl,
|
|
"updateTime": Common.updateTime
|
|
}
|
|
}})
|
|
|
|
|
|
@app.route("/stats/config", methods=["GET"])
|
|
def getConfigStats():
|
|
return jsonify({"message":"ok","code":200,"status":0,"data":{
|
|
"config": {
|
|
"forceNotBroadcasting": Common.forceNotBroadcasting,
|
|
"forceStopDownload": Common.forceStopDownload,
|
|
"forceNotUpload": Common.forceNotUpload,
|
|
}
|
|
}})
|
|
|
|
|
|
@app.route("/stats/download", methods=["GET"])
|
|
def getDownloadStats():
|
|
return jsonify({"message":"ok","code":200,"status":0,"data":{
|
|
"download":Common.downloadStatus,
|
|
}})
|
|
|
|
|
|
@app.route("/stats/encode", methods=["GET"])
|
|
def getEncodeStats():
|
|
return jsonify({"message":"ok","code":200,"status":0,"data":{
|
|
"encode": Common.encodeStatus,
|
|
"encodeQueueSize": Common.encodeQueue.qsize(),
|
|
}})
|
|
|
|
|
|
@app.route("/stats/upload", methods=["GET"])
|
|
def getUploadStats():
|
|
return jsonify({"message":"ok","code":200,"status":0,"data":{
|
|
"upload": Common.uploadStatus,
|
|
"uploadQueueSize": Common.uploadQueue.qsize(),
|
|
}})
|
|
|
|
|
|
@app.route("/files/", methods=["GET"])
|
|
def fileIndex():
|
|
a = []
|
|
for i in (glob("*.mp4") + glob("*.flv")):
|
|
a.append({
|
|
"name": i,
|
|
"size": Common.parseSize(os.path.getsize(i))
|
|
})
|
|
return render_template("files.html",files=a)
|
|
|
|
|
|
@app.route("/files/download/<path>", methods=["GET"])
|
|
def fileDownload(path):
|
|
def generate(file):
|
|
with open(file, "rb") as f:
|
|
for row in f:
|
|
yield row
|
|
return Response(generate(path), mimetype='application/octet-stream')
|
|
|
|
|
|
def SubThread():
|
|
t = threading.Thread(target=RUN, args=(Common.config['l_u'],))
|
|
t.setDaemon(True)
|
|
t.start()
|
|
while True:
|
|
if t.is_alive():
|
|
sleep(240)
|
|
else:
|
|
t = threading.Thread(target=RUN, args=(Common.config['l_u'],))
|
|
t.setDaemon(True)
|
|
t.start()
|
|
|
|
|
|
p = threading.Thread(target = SubThread)
|
|
p.setDaemon(True)
|
|
p.start()
|