Added File Storage server to Atract Server
This commit is contained in:
32
attract/application/file_server.py
Normal file
32
attract/application/file_server.py
Normal file
@@ -0,0 +1,32 @@
|
|||||||
|
import os
|
||||||
|
|
||||||
|
from flask import Blueprint
|
||||||
|
from flask import request
|
||||||
|
|
||||||
|
import config
|
||||||
|
|
||||||
|
file_server = Blueprint('file_server', __name__,
|
||||||
|
template_folder='templates',
|
||||||
|
static_folder='static/storage')
|
||||||
|
|
||||||
|
|
||||||
|
@file_server.route('/file', methods=['POST'])
|
||||||
|
@file_server.route('/file/<file_name>')
|
||||||
|
def index(file_name=None):
|
||||||
|
#GET file
|
||||||
|
if file_name:
|
||||||
|
folder_name = file_name[:2]
|
||||||
|
file_path = os.path.join("", folder_name, file_name)
|
||||||
|
print (file_path)
|
||||||
|
return file_server.send_static_file(file_path)
|
||||||
|
#POST file
|
||||||
|
file_name = request.form['name']
|
||||||
|
folder_name = file_name[:2]
|
||||||
|
file_folder_path = os.path.join(config.Development.FILE_STORAGE,
|
||||||
|
folder_name)
|
||||||
|
if not os.path.exists(file_folder_path):
|
||||||
|
os.mkdir(file_folder_path)
|
||||||
|
file_path = os.path.join(file_folder_path, file_name)
|
||||||
|
request.files['data'].save(file_path)
|
||||||
|
|
||||||
|
return "{}", 200
|
Reference in New Issue
Block a user