dbt-sqlserver 1.7.2__tar.gz → 1.7.3__tar.gz

{dbt-sqlserver-1.7.2/dbt_sqlserver.egg-info → dbt-sqlserver-1.7.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dbt-sqlserver
-Version: 1.7.2
+Version: 1.7.3
 Summary: A Microsoft SQL Server adapter plugin for dbt
 Home-page: https://github.com/dbt-msft/dbt-sqlserver
 Author: Mikael Ene, Anders Swanson, Sam Debruyn, Cor Zuurmond

dbt-sqlserver-1.7.3/dbt/adapters/sqlserver/__version__.py

@@ -0,0 +1 @@
+version = "1.7.3"

{dbt-sqlserver-1.7.2 → dbt-sqlserver-1.7.3}/dbt/adapters/sqlserver/sql_server_adapter.py

@@ -1,5 +1,9 @@
 # https://github.com/microsoft/dbt-fabric/blob/main/dbt/adapters/fabric/fabric_adapter.py
+from typing import Optional
+
+import dbt.exceptions
 from dbt.adapters.fabric import FabricAdapter
+from dbt.contracts.graph.nodes import ConstraintType, ModelLevelConstraint
 
 from dbt.adapters.sqlserver.sql_server_column import SQLServerColumn
 from dbt.adapters.sqlserver.sql_server_configs import SQLServerConfigs
@@ -50,6 +54,33 @@ class SQLServerAdapter(FabricAdapter):
     #     return columns
     # endregion
 
+    @classmethod
+    def render_model_constraint(cls, constraint: ModelLevelConstraint) -> Optional[str]:
+        constraint_prefix = "add constraint "
+        column_list = ", ".join(constraint.columns)
+
+        if constraint.name is None:
+            raise dbt.exceptions.DbtDatabaseError(
+                "Constraint name cannot be empty. Provide constraint name  - column "
+                + column_list
+                + " and run the project again."
+            )
+
+        if constraint.type == ConstraintType.unique:
+            return constraint_prefix + f"{constraint.name} unique nonclustered({column_list})"
+        elif constraint.type == ConstraintType.primary_key:
+            return constraint_prefix + f"{constraint.name} primary key nonclustered({column_list})"
+        elif constraint.type == ConstraintType.foreign_key and constraint.expression:
+            return (
+                constraint_prefix
+                + f"{constraint.name} foreign key({column_list}) references "
+                + constraint.expression
+            )
+        elif constraint.type == ConstraintType.custom and constraint.expression:
+            return f"{constraint_prefix}{constraint.expression}"
+        else:
+            return None
+
     @classmethod
     def date_function(cls):
         return "getdate()"

dbt-sqlserver-1.7.3/dbt/include/sqlserver/macros/materializations/seeds/helpers.sql

@@ -0,0 +1,57 @@
+{% macro sqlserver__get_binding_char() %}
+  {{ return('?') }}
+{% endmacro %}
+
+{% macro sqlserver__get_batch_size() %}
+  {{ return(400) }}
+{% endmacro %}
+
+{% macro calc_batch_size(num_columns) %}
+    {#
+        SQL Server allows for a max of 2098 parameters in a single statement.
+        Check if the max_batch_size fits with the number of columns, otherwise
+        reduce the batch size so it fits.
+    #}
+    {% set max_batch_size = get_batch_size() %}
+    {% set calculated_batch = (2098 / num_columns)|int %}
+    {% set batch_size = [max_batch_size, calculated_batch] | min %}
+
+    {{ return(batch_size) }}
+{%  endmacro %}
+
+{% macro sqlserver__load_csv_rows(model, agate_table) %}
+  {% set cols_sql = get_seed_column_quoted_csv(model, agate_table.column_names) %}
+  {% set batch_size = calc_batch_size(agate_table.column_names|length) %}
+  {% set bindings = [] %}
+  {% set statements = [] %}
+
+  {{ log("Inserting batches of " ~ batch_size ~ " records") }}
+
+  {% for chunk in agate_table.rows | batch(batch_size) %}
+      {% set bindings = [] %}
+
+      {% for row in chunk %}
+          {% do bindings.extend(row) %}
+      {% endfor %}
+
+      {% set sql %}
+          insert into {{ this.render() }} ({{ cols_sql }}) values
+          {% for row in chunk -%}
+              ({%- for column in agate_table.column_names -%}
+                  {{ get_binding_char() }}
+                  {%- if not loop.last%},{%- endif %}
+              {%- endfor -%})
+              {%- if not loop.last%},{%- endif %}
+          {%- endfor %}
+      {% endset %}
+
+      {% do adapter.add_query(sql, bindings=bindings, abridge_sql_log=True) %}
+
+      {% if loop.index0 == 0 %}
+          {% do statements.append(sql) %}
+      {% endif %}
+  {% endfor %}
+
+  {# Return SQL so we can render it out into the compiled files #}
+  {{ return(statements[0]) }}
+{% endmacro %}

{dbt-sqlserver-1.7.2 → dbt-sqlserver-1.7.3/dbt_sqlserver.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dbt-sqlserver
-Version: 1.7.2
+Version: 1.7.3
 Summary: A Microsoft SQL Server adapter plugin for dbt
 Home-page: https://github.com/dbt-msft/dbt-sqlserver
 Author: Mikael Ene, Anders Swanson, Sam Debruyn, Cor Zuurmond

{dbt-sqlserver-1.7.2 → dbt-sqlserver-1.7.3}/dbt_sqlserver.egg-info/SOURCES.txt

@@ -16,6 +16,7 @@ dbt/include/sqlserver/macros/adapters/indexes.sql
 dbt/include/sqlserver/macros/adapters/relation.sql
 dbt/include/sqlserver/macros/materializations/models/table/clone.sql
 dbt/include/sqlserver/macros/materializations/models/table/create_table_as.sql
+dbt/include/sqlserver/macros/materializations/seeds/helpers.sql
 dbt/include/sqlserver/macros/materializations/snapshots/snapshot.sql
 dbt/include/sqlserver/macros/utils/split_part.sql
 dbt/include/sqlserver/macros/utils/timestamps.sql

dbt-sqlserver-1.7.2/dbt/adapters/sqlserver/__version__.py

@@ -1 +0,0 @@
-version = "1.7.2"