How can I use context managers for database transactions in Python?

How can I use context managers for database transactions in Python?

You can use context managers to handle database transactions by ensuring that resources are properly managed. For example, using the with statement allows you to automatically commit or rollback transactions.

Can you show me a code example?

Sure! Here's a simple example:

import sqlite3

class Database:
    def __init__(self, db_name):
        self.db_name = db_name

    def __enter__(self):
        self.connection = sqlite3.connect(self.db_name)
        return self.connection

    def __exit__(self, exc_type, exc_val, exc_tb):
        if exc_type:
            self.connection.rollback()
        else:
            self.connection.commit()
        self.connection.close()

with Database('my_database.db') as db:
    cursor = db.cursor()
    cursor.execute('INSERT INTO my_table (column) VALUES (?)', (value,))

What happens if an error occurs during the transaction?

If an error occurs, the __exit__ method will trigger a rollback, ensuring that no partial changes are saved to the database.

Is it safe to use context managers for all types of databases?

Yes, context managers can be used with various databases, as long as the database library supports transactions and context management.