diff --git a/launch_standalone.py b/launch_standalone.py index 2681aff..db88dd9 100644 --- a/launch_standalone.py +++ b/launch_standalone.py @@ -3,11 +3,22 @@ import time from server import BAPSicleServer -if __name__ == '__main__': +def startServer(): + # On Windows calling this function is necessary. # Causes all kinds of loops if not present. multiprocessing.freeze_support() - server = multiprocessing.Process(target=BAPSicleServer).start() + + server = multiprocessing.Process(target=BAPSicleServer) + server.start() + while True: - time.sleep(1) - pass + time.sleep(2) + if server and server.is_alive(): + pass + else: + print("Server dead. Exiting.") + sys.exit(0) + +if __name__ == '__main__': + startServer() \ No newline at end of file diff --git a/player.py b/player.py index d00e56b..68f5fce 100644 --- a/player.py +++ b/player.py @@ -10,11 +10,13 @@ import setproctitle import copy import json import time +import os +import sys +os.environ['PYGAME_HIDE_SUPPORT_PROMPT'] = "hide" from pygame import mixer from state_manager import StateManager from mutagen.mp3 import MP3 -import os -os.environ['PYGAME_HIDE_SUPPORT_PROMPT'] = "hide" + class Player(): @@ -343,6 +345,7 @@ class Player(): print("Quiting player ", channel) self.quit() self._retMsg("EXIT") + sys.exit(0) def showOutput(in_q, out_q): diff --git a/server.py b/server.py index 5f733b3..8b2c05b 100644 --- a/server.py +++ b/server.py @@ -4,6 +4,8 @@ from flask import Flask, render_template, send_from_directory, request import json import sounddevice as sd import setproctitle +import logging +import sys setproctitle.setproctitle("BAPSicle - Server") @@ -18,10 +20,15 @@ class BAPSicleServer(): app = Flask(__name__, static_url_path='') +log = logging.getLogger('werkzeug') +log.disabled = True +app.logger.disabled = True + channel_to_q = [] channel_from_q = [] channel_p = [] +stopping = False @app.errorhandler(404) def page_not_found(e): @@ -147,6 +154,11 @@ def status(channel): return response +@app.route("/quit") +def quit(): + stopServer() + return "Shutting down..." + @app.route("/player/all/stop") def all_stop(): @@ -184,9 +196,23 @@ def stopServer(): for q in channel_to_q: q.put("QUIT") for player in channel_p: - player.join() - global app - app = None + try: + player.join() + except: + pass + print ("Stopped all players.") + global stopping + if stopping == False: + stopping = True + shutdown = request.environ.get('werkzeug.server.shutdown') + if shutdown is None: + print("Shutting down Server.") + + else: + print("Shutting down Flask.") + shutdown() + else: + print(str(stopping)) if __name__ == "__main__": diff --git a/state_manager.py b/state_manager.py index e81cffd..704a59a 100644 --- a/state_manager.py +++ b/state_manager.py @@ -8,6 +8,11 @@ class StateManager: __state = {} def __init__(self, name, default_state=None): + try: + os.mkdir(resolve_external_file_path("/state")) + except FileExistsError: + pass + self.filepath = resolve_external_file_path("/state/" + name + ".json") if not os.path.isfile(self.filepath): self.log("No file found for " + self.filepath)