From 00f8f9c7de03de701eea482599e991f1b31333dd Mon Sep 17 00:00:00 2001 From: Saurabh Kumar Bajpai Date: Fri, 24 Jul 2026 13:40:33 +0530 Subject: [PATCH] fix: lazy imports in sql_connection_manager, remove hardcoded $set in mongo_document_update --- pysnippets/database/mongodb/mongo_document_update.py | 4 ++-- pysnippets/database/mysql/sql_connection_manager.py | 7 ++++--- 2 files changed, 6 insertions(+), 5 deletions(-) diff --git a/pysnippets/database/mongodb/mongo_document_update.py b/pysnippets/database/mongodb/mongo_document_update.py index c8587d65..c772270d 100644 --- a/pysnippets/database/mongodb/mongo_document_update.py +++ b/pysnippets/database/mongodb/mongo_document_update.py @@ -12,7 +12,7 @@ def update_document(collection, filter_criteria, update_data, multiple=False): dict: The update result details. """ if multiple: - result = collection.update_many(filter_criteria, {'$set': update_data}) + result = collection.update_many(filter_criteria, update_data) else: - result = collection.update_one(filter_criteria, {'$set': update_data}) + result = collection.update_one(filter_criteria, update_data) return result.raw_result diff --git a/pysnippets/database/mysql/sql_connection_manager.py b/pysnippets/database/mysql/sql_connection_manager.py index e70d213f..86c38589 100644 --- a/pysnippets/database/mysql/sql_connection_manager.py +++ b/pysnippets/database/mysql/sql_connection_manager.py @@ -1,6 +1,3 @@ -import psycopg2 -import mysql.connector - def connect_to_mysql(host, user, password, database): """ Connects to a MySQL database. @@ -14,6 +11,8 @@ def connect_to_mysql(host, user, password, database): Returns: mysql.connector.connection.MySQLConnection: Connection object if successful. """ + import mysql.connector + connection = mysql.connector.connect( host=host, user=user, @@ -35,6 +34,8 @@ def connect_to_postgresql(host, user, password, database): Returns: psycopg2.extensions.connection: Connection object if successful. """ + import psycopg2 + connection = psycopg2.connect( host=host, user=user,