mssql-python 0.1.6__cp310-cp310-win_amd64.whl

mssql_python/db_connection.py

@@ -0,0 +1,38 @@
+"""
+Copyright (c) Microsoft Corporation.
+Licensed under the MIT license.
+This module provides a way to create a new connection object to interact with the database.
+"""
+from mssql_python.connection import Connection
+
+
+def connect(connection_str: str = "", autocommit: bool = True, attrs_before: dict = None, **kwargs) -> Connection:
+    """
+    Constructor for creating a connection to the database.
+
+    Args:
+        connection_str (str): The connection string to connect to.
+        autocommit (bool): If True, causes a commit to be performed after each SQL statement.
+    TODO: Add the following parameters to the function signature:
+        timeout (int): The timeout for the connection attempt, in seconds.
+        readonly (bool): If True, the connection is set to read-only.
+        attrs_before (dict): A dictionary of connection attributes to set before connecting.
+    Keyword Args:
+        **kwargs: Additional key/value pairs for the connection string.
+    Below attributes are not implemented in the internal driver:
+    - encoding (str): The encoding for the connection string.
+    - ansi (bool): If True, indicates the driver does not support Unicode.
+
+    Returns:
+        Connection: A new connection object to interact with the database.
+
+    Raises:
+        DatabaseError: If there is an error while trying to connect to the database.
+        InterfaceError: If there is an error related to the database interface.
+
+    This function provides a way to create a new connection object, which can then
+    be used to perform database operations such as executing queries, committing
+    transactions, and closing the connection.
+    """
+    conn = Connection(connection_str, autocommit=autocommit, attrs_before=attrs_before, **kwargs)
+    return conn

mssql_python/ddbc_bindings.py

@@ -0,0 +1,38 @@
+import os
+import importlib.util
+import sys
+import platform
+
+# Get current Python version and architecture
+python_version = f"cp{sys.version_info.major}{sys.version_info.minor}"
+if platform.machine().lower() in ('amd64', 'x86_64', 'x64'):
+    architecture = "amd64"
+elif platform.machine().lower() in ('arm64', 'aarch64'):
+    architecture = "arm64"
+else:
+    architecture = platform.machine().lower()
+
+# Find the specifically matching PYD file
+module_dir = os.path.dirname(__file__)
+expected_pyd = f"ddbc_bindings.{python_version}-{architecture}.pyd"
+pyd_path = os.path.join(module_dir, expected_pyd)
+
+if not os.path.exists(pyd_path):
+    # Fallback to searching for any matching PYD if the specific one isn't found
+    pyd_files = [f for f in os.listdir(module_dir) if f.startswith('ddbc_bindings.') and f.endswith('.pyd')]
+    if not pyd_files:
+        raise ImportError(f"No ddbc_bindings PYD module found for {python_version}-{architecture}")
+    pyd_path = os.path.join(module_dir, pyd_files[0])
+    print(f"Warning: Using fallback PYD file {pyd_files[0]} instead of {expected_pyd}")
+
+# Use the original module name 'ddbc_bindings' that the C extension was compiled with
+name = "ddbc_bindings"
+spec = importlib.util.spec_from_file_location(name, pyd_path)
+module = importlib.util.module_from_spec(spec)
+sys.modules[name] = module
+spec.loader.exec_module(module)
+
+# Copy all attributes from the loaded module to this module
+for attr in dir(module):
+    if not attr.startswith('__'):
+        globals()[attr] = getattr(module, attr)