mirror of
https://github.com/w-okada/voice-changer.git
synced 2025-01-23 13:35:12 +03:00
86 lines
2.3 KiB
Python
86 lines
2.3 KiB
Python
from flask import Flask, request, Markup, abort, jsonify, send_from_directory
|
|
from flask_cors import CORS
|
|
import logging
|
|
from logging.config import dictConfig
|
|
import sys, os
|
|
import base64
|
|
|
|
DATA_ROOT = "./"
|
|
dictConfig({
|
|
'version': 1,
|
|
'formatters': {'default': {
|
|
'format': '[%(asctime)s] %(levelname)s in %(module)s: %(message)s',
|
|
}},
|
|
'handlers': {'wsgi': {
|
|
'class': 'logging.StreamHandler',
|
|
'stream': 'ext://flask.logging.wsgi_errors_stream',
|
|
'formatter': 'default'
|
|
}},
|
|
'root': {
|
|
'level': 'DEBUG',
|
|
'handlers': ['wsgi']
|
|
}
|
|
})
|
|
|
|
|
|
app = Flask(__name__)
|
|
app.debug = True
|
|
@app.route("/<path:path>")
|
|
def static_dir(path):
|
|
return send_from_directory("../docs", path)
|
|
|
|
|
|
@app.route('/api/voice/<string:title>/<string:prefix>/<int:index>', methods=['POST'])
|
|
def post_voice(title, prefix, index):
|
|
filename = f"{prefix}{index:03}.zip"
|
|
data_dir = os.path.join(DATA_ROOT, title)
|
|
os.makedirs(data_dir,exist_ok=True)
|
|
fullpath = os.path.join(data_dir, filename)
|
|
data = base64.b64decode(request.json['data'])
|
|
f = open(fullpath, 'wb')
|
|
f.write(data)
|
|
f.close()
|
|
data = {
|
|
"message":"OK"
|
|
}
|
|
return jsonify(data)
|
|
|
|
@app.route('/api/voice/<string:title>/<string:prefix>/<int:index>', methods=['GET'])
|
|
def get_voice(title, prefix, index):
|
|
filename = f"{prefix}{index:03}.zip"
|
|
data_dir = os.path.join(DATA_ROOT, title)
|
|
fullpath = os.path.join(data_dir, filename)
|
|
|
|
is_file = os.path.isfile(fullpath)
|
|
if is_file == False:
|
|
data = {
|
|
"message":"NOT_FOUND",
|
|
}
|
|
return jsonify(data)
|
|
|
|
f = open(fullpath, 'rb')
|
|
data = f.read()
|
|
dataBase64 = base64.b64encode(data).decode('utf-8')
|
|
data = {
|
|
"message":"OK",
|
|
"data":dataBase64,
|
|
}
|
|
return jsonify(data)
|
|
|
|
CORS(app, resources={r"/*": {"origins": "*"}})
|
|
|
|
@app.after_request
|
|
def after_request(response):
|
|
# response.headers.add('Access-Control-Allow-Origin', '*')
|
|
response.headers.add('Access-Control-Allow-Headers', 'Content-Type,Authorization')
|
|
response.headers.add('Access-Control-Allow-Methods', 'GET,PUT,POST,DELETE,OPTIONS')
|
|
return response
|
|
|
|
|
|
if __name__ == '__main__':
|
|
args = sys.argv
|
|
PORT = args[1]
|
|
DATA_ROOT = args[2]
|
|
app.logger.info('START APP')
|
|
app.run(debug=True, host='0.0.0.0',port=PORT)
|