With statement execution flow in Python

Introduction

The with statement helps you run code that needs setup and cleanup automatically. It makes sure things like files or resources are properly opened and closed without extra work.

When opening a file to read or write and you want it to close automatically.

When working with locks in multi-threading to ensure they are released.

When managing database connections that need to be closed after use.

When handling resources like network connections that require cleanup.

Syntax

Python

with expression as variable:
    block_of_code

The expression must return a context manager object.

The variable is optional and holds the resource from the context manager.

Examples

Open a file and read its content. The file closes automatically after the block.

Python

with open('file.txt', 'r') as f:
    content = f.read()

Acquire a lock before running code and release it automatically after.

Python

with lock:
    # critical section
    do_something()

Use with without as if you don't need the resource variable.

Python

with open('file.txt'):
    print('File opened')

Sample Program

This program shows how the with statement calls __enter__ before the block and __exit__ after the block automatically.

Python

class SimpleContext:
    def __enter__(self):
        print('Entering the block')
        return 'Resource'
    def __exit__(self, exc_type, exc_val, exc_tb):
        print('Exiting the block')

with SimpleContext() as resource:
    print(f'Inside block with {resource}')

Important Notes

The __enter__ method runs first and can return a value used inside the block.

The __exit__ method runs after the block, even if an error happens inside the block.

This helps avoid forgetting to close or clean up resources.

Summary

The with statement manages setup and cleanup automatically.

It calls __enter__ before the block and __exit__ after the block.

Use it to safely handle files, locks, and other resources.