dataface 0.1.2__py3-none-any.whl

dataface/core/execute/dialects/sqlserver.py

@@ -0,0 +1,92 @@
+"""SQL Server dialect implementation.
+
+SQL Server uses:
+- @p1, @p2, ... for parameter placeholders
+- SELECT * INTO #name FROM ... for temporary tables (prefixed with #)
+- Square brackets for identifier quoting
+"""
+
+from dataface.core.execute.dialects.base import SQLDialect
+
+
+class SQLServerDialect(SQLDialect):
+    """SQL Server dialect with @pN parameter style.
+
+    SQL Server uses named parameters with @ prefix and # prefix for temp tables.
+    Temp table creation uses SELECT INTO pattern.
+
+    Example:
+        >>> dialect = SQLServerDialect()
+        >>> dialect.param(1)
+        '@p1'
+        >>> dialect.temp_table('tmp', 'SELECT 1 as val')
+        'SELECT * INTO #tmp FROM (SELECT 1 as val) AS _t'
+        >>> dialect.temp_table_ref('tmp')
+        '#tmp'
+        >>> dialect.filter('category', '=', 1)
+        'category = @p1'
+    """
+
+    name = "sqlserver"
+    uses_named_params = True
+    dbapi_placeholder = "%s"
+
+    def param(self, index: int) -> str:
+        """Generate SQL Server parameter placeholder.
+
+        SQL Server uses @pN named style.
+
+        Args:
+            index: 1-based parameter index
+
+        Returns:
+            Parameter placeholder in @pN format
+        """
+        return f"@p{index}"
+
+    def temp_table(self, name: str, sql: str) -> str:
+        """Generate SQL Server temp table creation.
+
+        SQL Server uses SELECT INTO #name pattern for temp tables.
+        The source query is wrapped as a subquery with a unique alias
+        to avoid conflicts with the source SQL.
+
+        Args:
+            name: Name of the temporary table (without # prefix)
+            sql: SQL query to populate the table (must be a SELECT statement)
+
+        Returns:
+            SELECT INTO statement for temp table creation
+
+        Note:
+            The sql parameter must be a SELECT statement that can be
+            wrapped as a subquery. Other statement types (CREATE, INSERT, etc.)
+            will fail.
+        """
+        return f"SELECT * INTO #{name} FROM ({sql}) AS _dataface_src"
+
+    def temp_table_ref(self, name: str) -> str:
+        """Generate SQL Server temp table reference.
+
+        SQL Server temp tables are prefixed with #.
+
+        Args:
+            name: Name of the temporary table
+
+        Returns:
+            Temp table name with # prefix
+        """
+        return f"#{name}"
+
+    def quote_identifier(self, identifier: str) -> str:
+        """Quote an identifier using SQL Server square brackets.
+
+        Args:
+            identifier: The identifier to quote
+
+        Returns:
+            Square bracket-quoted identifier
+        """
+        # Escape right brackets by doubling them
+        escaped = identifier.replace("]", "]]")
+        return f"[{escaped}]"