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