import sqlite3

def db_connection(func):
    def wrapper(*args, **kwargs):
        conn = sqlite3.connect('database.db')
        cursor = conn.cursor()
        
        result = func(cursor, *args, **kwargs)
        
        conn.commit()
        conn.close()
        
        return result
    
    return wrapper

@db_connection
def query(cursor, sql):
    cursor.execute(sql)
    return cursor.fetchall()

# Пример использования декоратора
result = query("SELECT * FROM users")
print(result)