main.py

import time
from contextlib import contextmanager

# Context Manager for database connection handling
@contextmanager
def database_connection():
    print("Connecting to database...")
    time.sleep(1)  # Simulate connection delay
    yield "DB Connection Object"  # Mock database connection object
    print("Closing database connection...")  # Simulate disconnection

# Decorator for logging query execution time
def log_query(func):
    def wrapper(*args, **kwargs):
        print(f"Executing query: {func.__name__}")
        start = time.time()
        result = func(*args, **kwargs)
        end = time.time()
        print(f"Query executed in {end - start:.4f} seconds")
        return result
    return wrapper

# Example usage
with database_connection() as conn:
    @log_query
    def fetch_users(connection):
        time.sleep(2)  # Simulate query delay
        print(f"Fetching users from {connection}")
        return ["User1", "User2", "User3"]

    users = fetch_users(conn)
    print(users)