Allowed file types validation in Flask - Performance & Optimization

Performance: Allowed file types validation

MEDIUM IMPACT

This affects page load speed and user interaction responsiveness by preventing unnecessary file uploads and server processing.

Validating uploaded file types to prevent invalid files

Flask

from flask import request
ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif'}
def allowed_file(filename):
    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS

file = request.files['file']
if file and allowed_file(file.filename):
    file.save('/uploads/' + file.filename)
else:
    return 'Invalid file type', 400

Checks file extension before saving, preventing unnecessary server processing of invalid files.

📈 Performance GainReduces server workload and response blocking, improving INP and overall user experience

Validating uploaded file types to prevent invalid files

Flask

from flask import request

file = request.files['file']
# No validation on file type
file.save('/uploads/' + file.filename)

No file type check causes server to process all files, including invalid or malicious ones.

📉 Performance CostBlocks server processing for all uploads, increasing response time and INP delays

Performance Comparison

Pattern	DOM Operations	Reflows	Paint Cost	Verdict
No file type validation	N/A	N/A	N/A	[X] Bad
Early file type validation on server	N/A	N/A	N/A	[OK] Good

Rendering Pipeline

File type validation happens on the server before file processing and response generation, reducing server-side blocking and improving interaction responsiveness.

→Server Processing

→Response Generation

⚠️ BottleneckServer Processing when handling invalid files

Core Web Vital Affected

INP

This affects page load speed and user interaction responsiveness by preventing unnecessary file uploads and server processing.

Optimization Tips

1Always validate file types early on the server to avoid unnecessary processing.

2Reject invalid files quickly to reduce server response blocking and improve interaction speed.

3Use a whitelist of allowed extensions to keep validation simple and fast.

Performance Quiz - 3 Questions

Why does validating allowed file types on the server improve performance?

A It increases the number of files uploaded, improving throughput.

B It prevents processing invalid files, reducing server load and response time.

C It delays the file upload to allow more user input.

D It reduces the size of the uploaded files automatically.

DevTools: Network

How to check: Upload files with allowed and disallowed types and observe server responses in the Network panel.

What to look for: Look for 400 errors on invalid files and faster response times on rejected uploads indicating good validation.

Practice

1. What is the main purpose of validating allowed file types in a Flask upload feature?

easy

A. To compress files before saving

B. To speed up the file upload process

C. To automatically rename files on upload

D. To ensure only safe and expected file formats are accepted

Solution

1. Step 1: Understand file upload risks

   Uploading files without validation can allow harmful files that may damage the server or application.

2. Step 2: Purpose of allowed file types

   Checking allowed file types restricts uploads to safe, expected formats, reducing security risks.

3. Final Answer:

   To ensure only safe and expected file formats are accepted -> Option D

4. Quick Check:

   Allowed file types = safety [OK]

Hint: Allowed types protect from unsafe uploads [OK]

Common Mistakes:

• Thinking validation speeds upload
• Confusing validation with renaming
• Assuming validation compresses files

2. Which of the following is the correct way to check if a filename has an allowed extension in Flask?

easy

A. if filename.split('.')[-1] in allowed_extensions:

B. if filename.endswith(allowed_extensions):

C. if filename in allowed_extensions:

D. if allowed_extensions.contains(filename):

Solution

1. Step 1: Extract file extension correctly

   Splitting the filename by '.' and taking the last part gets the extension.

2. Step 2: Check extension in allowed set

   Using 'in' checks if the extension is in the allowed_extensions set.

3. Final Answer:

   if filename.split('.')[-1] in allowed_extensions: -> Option A

4. Quick Check:

   Split and check extension [OK]

Hint: Split filename by '.' and check last part [OK]

Common Mistakes:

• Using endswith with a set (wrong syntax)
• Checking whole filename in allowed extensions
• Using non-existent 'contains' method

3. Given this Flask snippet, what will be the output if the uploaded file is named photo.png and allowed_extensions = {'png', 'jpg'}?

filename = 'photo.png'
if filename.split('.')[-1] in allowed_extensions:
    result = 'Allowed'
else:
    result = 'Not Allowed'

medium

A. Allowed

B. Not Allowed

C. Error: split method missing

D. Error: allowed_extensions undefined

Solution

1. Step 1: Extract extension from filename

   Splitting 'photo.png' by '.' gives ['photo', 'png'], last part is 'png'.

2. Step 2: Check if extension is allowed

   'png' is in the set {'png', 'jpg'}, so condition is True and result is 'Allowed'.

3. Final Answer:

   Allowed -> Option A

4. Quick Check:

   Extension 'png' in allowed set [OK]

Hint: Check last part after '.' in allowed set [OK]

Common Mistakes:

• Confusing filename with extension
• Assuming split returns error
• Forgetting to define allowed_extensions

4. Identify the error in this Flask file validation code:

allowed_extensions = {'txt', 'pdf'}
filename = 'report.pdf'
if filename.split('.')[-1] == allowed_extensions:
    print('File allowed')
else:
    print('File not allowed')

medium

A. allowed_extensions should be a list, not a set

B. filename.split('.')[-1] is incorrect syntax

C. Using '==' instead of 'in' to check extension

D. Missing import for split function

Solution

1. Step 1: Understand the comparison

   The code compares a string (extension) to a set using '==', which is always False.

2. Step 2: Correct way to check membership

   Use 'in' operator to check if extension is inside allowed_extensions set.

3. Final Answer:

   Using '==' instead of 'in' to check extension -> Option C

4. Quick Check:

   Use 'in' for membership check [OK]

Hint: Use 'in' to check if extension is allowed [OK]

Common Mistakes:

• Using '==' instead of 'in'
• Thinking split needs import
• Confusing set with list requirement

5. You want to allow only image files with extensions 'png', 'jpg', and 'jpeg' in your Flask app. Which code snippet correctly validates the uploaded file's extension and handles the case where the filename has no extension?

hard

A. allowed = {'png', 'jpg', 'jpeg'}
if filename.endswith(tuple(allowed)):
save_file()
else:
reject_file()

B. allowed = {'png', 'jpg', 'jpeg'}
if '.' in filename and filename.rsplit('.', 1)[1].lower() in allowed:
save_file()
else:
reject_file()

C. allowed = ['png', 'jpg', 'jpeg']
if filename.split('.')[-1] in allowed:
save_file()
else:
reject_file()

D. allowed = {'png', 'jpg', 'jpeg'}
if filename.split('.')[1] in allowed:
save_file()
else:
reject_file()

Solution

1. Step 1: Check for presence of extension

   Using '.' in filename ensures the file has an extension to avoid errors.

2. Step 2: Extract extension safely and check membership

   Using rsplit('.', 1)[1].lower() gets the extension after the last dot, then checks if it's in allowed set.

3. Step 3: Handle saving or rejecting file

   If extension is allowed, call save_file(), else reject_file().

4. Final Answer:

   allowed = {'png', 'jpg', 'jpeg'} if '.' in filename and filename.rsplit('.', 1)[1].lower() in allowed: save_file() else: reject_file() -> Option B

5. Quick Check:

   Check dot, split once from right, then 'in' allowed [OK]

Hint: Check '.' then rsplit once, then 'in' allowed [OK]

Common Mistakes:

• Not checking if filename has '.' before splitting
• Using split without limit causing errors
• Using endswith with tuple but missing case normalization