import os
import os.path

from flask import Blueprint, request, send_file

training_documents_controller = Blueprint('training_documents_controller', __name__)

@training_documents_controller.route('/training_documents', methods=['GET'])
def get_training_documents():

	files = os.listdir('./storage/training')
	files.remove('.gitignore')

	return files

@training_documents_controller.route('/training_documents/<file_name>', methods=['GET'])
def get_training_document(file_name):
	
	if not os.path.isfile('./storage/training/' + file_name):
		return 'File not found', 404
	
	return send_file('../storage/training/' + file_name, as_attachment=True)

@training_documents_controller.route('/training_documents', methods=['POST'])
def store_training_document():

	file = request.files['file']
	file.save('./storage/training/' + file.filename)

	return 'Document stored'
	
@training_documents_controller.route('/training_documents/<file_name>', methods=['DELETE'])
def delete_training_document(file_name):

	if not os.path.isfile('./storage/training/' + file_name):
		return 'File not found', 404

	os.remove('./storage/training/' + file_name)

	return 'Document deleted'