ChatGPTにファイルアップロードするPython-FlaskのWebアプリを作成してもらった

ChatGPTにファイルアップロードするPython-FlaskのWebアプリを作成してもらいました。

指示文(プロンプト):

PythonのFlsskで、ローカルからファイルをアップロードできるWebアプリを作成してください。アップロードできるファイルの拡張子はあらかじめ指定できるようにしてください。アップロードできるファイルサイズの上限をあらかじめ設定できるようにしてください。

回答:

以下は、PythonのFlaskを使用して、ローカルからファイルをアップロードできるWebアプリを作成するための例です。この例では、アップロードできるファイルの拡張子は.txtと.csvに限定され、ファイルサイズの上限は2MBに設定されています。必要に応じてこれらの制限を変更できます。


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

app = Flask(__name__)

UPLOAD_FOLDER = './uploads'
ALLOWED_EXTENSIONS = {'txt', 'csv', 'jpg', 'png'}
MAX_CONTENT_LENGTH = 1 * 1024 * 1024 # 1MB

app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
app.config['MAX_CONTENT_LENGTH'] = MAX_CONTENT_LENGTH

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:
            flash('No file part')
            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 == '':
            flash('No selected file')
            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>
    <html>
    <head>
    <title>Upload new File</title>
    </head>
    <body>
    <h1>Upload new File</h1>
    <form method=post enctype=multipart/form-data>
      <input type=file name=file>
      <input type=submit value=Upload>
    </form>
    </body>
    </html>
    '''

@app.route('/uploads/')
def uploaded_file(filename):
    return f'''
    <!doctype html>
    <html>
    <head>
    <title>Uploaded File</title>
    </head>
    <body>
    <h1>Uploaded File</h1>
    <p>File {filename} has been uploaded.</p>
    <p><a href="{url_for('upload_file')}">Upload another file</a></p>
    </body>
    </html>
    '''

if __name__ == '__main__':
    app.run(debug=True)

        

この例では、アップロードされたファイルは./uploadsフォルダに保存されます。フォルダが存在しない場合は、アプリが自動的に作成します。アップロードされたファイルのファイル名は、secure_filename()関数を使用して、安全な形式で保存されます。 また、WebアプリのHTMLテンプレートには、ファイルをアップロードするためのフォームが含まれています。このフォームは、multipart/form-dataを使用して送信されます。 最後に、アプリを実行するには、ターミナルでpythonファイルを実行します。

結果:

成功はしましたが、最初に提示されたコードだと、werkzeug.utilsのインポートが抜けていたので追加しました。また、サンプルだとtxtとcsvだけが許可されてますが、テストのためjpg、pngも許可に加えてます。

 

サイズ上限も機能しました。