Приветики! Всех с 2018!) Появился вопросик, как сделать проверку на flask файла? МНе необходимо проверить, что файл png и если это не так выполнить одно условие и прекратить программу, а также сделать проверку что размер изображения должен быть 240 на 250 пикселей и опять же если не так вернуть ошибку.
Сейчас у меня следующий код:
import os
from flask import Flask, request, redirect, url_for
from werkzeug.utils import secure_filename
UPLOAD_FOLDER = '/var/www/html/sample0/files'
ALLOWED_EXTENSIONS = set(['txt', 'pdf', 'png', 'jpg', 'jpeg', 'gif'])
app = Flask(__name__)
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
def allowed_file(filename):
return '.' in filename and \
filename.rsplit('.', 1)[1] in ALLOWED_EXTENSIONS
@app.route('/', methods=['GET', 'POST'])
def upload_file():
if request.method == 'POST':
file = request.files['file']
if file and allowed_file(file.filename):
filename = file.filename#secure_filename(file.filename)
file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
return '''
<script src="https://code.jquery.com/jquery-1.11.3.js"></script>
<script>
$.ajax({
url : 'http://mysait.com/sample0/upload.php',
success : function(data) {
document.body.innerHTML = data;
}
});
</script>
'''
return '''
<script src="https://code.jquery.com/jquery-1.11.3.js"></script>
<script>
$.ajax({
url : 'http://mysait.com/sample0/upload.php',
success : function(data) {
document.body.innerHTML = data;
}
});
</script>
'''
if __name__ == "__main__":
app.run(debug=True)
Я так понимаю в ALLOWED_EXTENTION нужно оставить только png? Заранее спасибо за помощь!
