How do I use context managers with databases in Python?

How do I use context managers with databases in Python?

You can use context managers in Python to handle database connections efficiently. They ensure that resources are properly managed, like closing connections after use. Would you like a code example?

Yes, please provide a code example.

Here's a simple example using SQLite:

import sqlite3

class Database:
    def __enter__(self):
        self.connection = sqlite3.connect('example.db')
        return self.connection

    def __exit__(self, exc_type, exc_val, exc_tb):
        self.connection.close()

with Database() as db:
    cursor = db.cursor()
    cursor.execute('CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY, name TEXT)')
    db.commit()

This code creates a context manager for the SQLite database.

What happens if an error occurs during the database operation?

If an error occurs, the __exit__ method will still be called, ensuring that the connection is closed properly. You can also handle exceptions within the context manager to manage errors gracefully.