diff --git a/local.py b/local.py index 8da433f20b4b5a1d3c29ca8aa93720c7bc4f7257..4c21939982b22b603e1f088b4e087229f21f7f1b 100644 --- a/local.py +++ b/local.py @@ -1,5 +1,5 @@ from minor import App -app = App() +app = App({"UPLOAD_DIR": "./upload"}) app.run() \ No newline at end of file diff --git a/minor.wsgi b/minor.wsgi index a61e415c47b696c716149e7367507903accdb2c2..1ae827c9034f9ceabe27f97d8ddb464a0add9460 100644 --- a/minor.wsgi +++ b/minor.wsgi @@ -1,4 +1,5 @@ from minor import App -app = App() +# change upload dir! +app = App({"UPLOAD_DIR": "./upload"}) application = app.get_application() \ No newline at end of file diff --git a/minor/app.py b/minor/app.py index a2540b663d9819fefe4b661198aba4c23c02029e..ec6e709551c1233464a937d2459d9151f4c619ae 100644 --- a/minor/app.py +++ b/minor/app.py @@ -3,21 +3,19 @@ from minor.controller import ApiController, MusicController class App(object): - def __init__(self): + def __init__(self, config): self._flask = Flask(__name__) self._flask.config['MONGODB_SETTINGS'] = {'db': 'minor'} + self._flask.config.update(config) from minor.model import db db.init_app(self._flask) ApiController(self).register(self._flask) - MusicController().register(self._flask) + MusicController(self._flask.config["UPLOAD_DIR"]).register(self._flask) def run(self): self._flask.run() - def check_key(self, key): - return key == "sekrit1234" - def get_application(self): return self._flask diff --git a/minor/controller/api.py b/minor/controller/api.py index 265c90a27458d30548f5d000a69c50c58d3bce1c..9e7c3b270468e212d14bc3b5f57a1e21f268b72a 100644 --- a/minor/controller/api.py +++ b/minor/controller/api.py @@ -19,16 +19,3 @@ class ApiController(FlaskController): @route("/ping") def ping(self): return "pong" - - @json_required - @route("/register", methods=["POST"]) - def register_major(self): - json = request.get_json() - if "key" in json: - if self._app.check_key(json["key"]): - # add the instance the the trusted list - new_instance = MajorInstance( - issuer=json["issuer"], url=json["url"]) - new_instance.save() - return jsonify({"code": 200, "message": "ok"}), 200 - return jsonify({"code": 400, "message": "bad_request"}), 400 diff --git a/minor/controller/music.py b/minor/controller/music.py index 24d641548e3b7fca9c09b558c06d450dd3e2b549..66bff44a5d9b6285f4eaa4df6f4d7276a4a96ec9 100644 --- a/minor/controller/music.py +++ b/minor/controller/music.py @@ -12,6 +12,8 @@ from minor.util import get_ext @route("/api/music") class MusicController(FlaskController): + def __init__(self, upload_dir): + self._upload_dir = upload_dir @auth_required @route("/listen") @@ -19,7 +21,7 @@ class MusicController(FlaskController): if "song" in payload: song = Song.objects.get(id=payload["song"]) if song is not None: - filename = "./uploads/" + str(song.id) + ".mp3" + filename = self._upload_dir + str(song.id) + ".mp3" if os.path.isfile(filename): def generate(): with open(filename, "rb") as fogg: @@ -45,7 +47,7 @@ class MusicController(FlaskController): def song_has_file(self, id, payload=None): song = Song.objects.get(id=id) if song is not None: - filename = "./uploads/" + str(song.id) + ".mp3" + filename = self._upload_dir + str(song.id) + ".mp3" if os.path.isfile(filename): return jsonify({"code": 200, "message": "ok", "data": True}), 200 else: @@ -90,7 +92,7 @@ class MusicController(FlaskController): ext = get_ext(file.filename) if ext is not None: name = secure_filename(str(db_song.id) + "." + ext) - path = os.path.join("./uploads", name) + path = os.path.join(self._upload_dir, name) file.save(path) return jsonify({"code": 200, "message": "ok"}), 200