[img_upload.git] / app.py

import os
from flask import Flask, flash, request, redirect, url_for
from werkzeug.utils import secure_filename

UPLOAD_FOLDER = "/tmp/"
ALLOWED_EXTENSIONS = {"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].lower() in ALLOWED_EXTENSIONS


@app.route("/", methods=["GET", "POST"])
def upload_file():
    if request.method == "POST":
        # check if the post request has the file part
        if "file" not in request.files:
            return redirect(request.url)
        file = request.files["file"]
        # if user does not select file, browser also
        # submit an empty part without filename
        if file.filename == "":
            return redirect(request.url)
        if file and allowed_file(file.filename):
            filename = secure_filename(file.filename)
            file.save(os.path.join(app.config["UPLOAD_FOLDER"], filename))
            return redirect(url_for("uploaded_file", filename=filename))

    return """
    <!doctype html>
    <title>Upload new File</title>
    <h1>Upload new File</h1>
    <form method=post enctype=multipart/form-data>
        <input type=file name=file>
        <input type=submit value=Upload>
    </form>
"""
Commit	Line	Data
be01a9ec JC	1	import os
	2	from flask import Flask, flash, request, redirect, url_for
	3	from werkzeug.utils import secure_filename
	4
	5	UPLOAD_FOLDER = "/tmp/"
	6	ALLOWED_EXTENSIONS = {"txt", "pdf", "png", "jpg", "jpeg", "gif"}
	7
	8	app = Flask(__name__)
	9	app.config["UPLOAD_FOLDER"] = UPLOAD_FOLDER
	10
	11
	12	def allowed_file(filename):
	13	return "." in filename and filename.rsplit(".", 1)[1].lower() in ALLOWED_EXTENSIONS
	14
	15
	16	@app.route("/", methods=["GET", "POST"])
	17	def upload_file():
	18	if request.method == "POST":
	19	# check if the post request has the file part
	20	if "file" not in request.files:
	21	return redirect(request.url)
	22	file = request.files["file"]
	23	# if user does not select file, browser also
	24	# submit an empty part without filename
	25	if file.filename == "":
	26	return redirect(request.url)
	27	if file and allowed_file(file.filename):
	28	filename = secure_filename(file.filename)
	29	file.save(os.path.join(app.config["UPLOAD_FOLDER"], filename))
	30	return redirect(url_for("uploaded_file", filename=filename))
	31
	32	return """
	33	<!doctype html>
	34	<title>Upload new File</title>
	35	<h1>Upload new File</h1>
	36	<form method=post enctype=multipart/form-data>
	37	<input type=file name=file>
	38	<input type=submit value=Upload>
	39	</form>
	40	"""