Comment registration support for MSSQL engine

Signed-off-by: Ujfalusi Sándor <ujfalusi.sandor@gmail.com>

Author: ujfalusis

sqlmesh/core/engine_adapter/mssql.py

@@ -465,8 +465,8 @@ def _build_create_comment_table_exp(
         template = dedent("""
             DECLARE @comment sql_variant = {comment};
             DECLARE @property_name VARCHAR(128) = 'MS_Description';
-            DECLARE @schema_name VARCHAR(128) = '{schema_name}';
-            DECLARE @object_name VARCHAR(128) = '{object_name}';
+            DECLARE @schema_name VARCHAR(128) = {schema_name};
+            DECLARE @object_name VARCHAR(128) = {object_name};
             DECLARE @object_kind VARCHAR(128) = '{object_kind}';
             DECLARE @existing sql_variant;
 
@@ -486,11 +486,13 @@ def _build_create_comment_table_exp(
                 END
         """)
         tsql_text = template.format(
-            comment=exp.Literal.string(table_comment).sql(dialect=self.dialect)
-            if table_comment is not None
-            else "NULL",
-            schema_name=table.db if table.db else "dbo",
-            object_name=table.name,
+            comment=exp.Literal.string(table_comment or "NULL").sql(
+                dialect=self.dialect, identify=False
+            ),
+            schema_name=exp.Literal.string(table.db or "dbo").sql(
+                dialect=self.dialect, identify=False
+            ),
+            object_name=exp.Literal.string(table.name).sql(dialect=self.dialect, identify=False),
             object_kind=table_kind,
         )
         return tsql_text
@@ -501,10 +503,10 @@ def _build_create_comment_column_exp(
         template = dedent("""
             DECLARE @comment sql_variant = {comment};
             DECLARE @property_name VARCHAR(128) = 'MS_Description';
-            DECLARE @schema_name VARCHAR(128) = '{schema_name}';
-            DECLARE @object_name VARCHAR(128) = '{object_name}';
+            DECLARE @schema_name VARCHAR(128) = {schema_name};
+            DECLARE @object_name VARCHAR(128) = {object_name};
             DECLARE @object_kind VARCHAR(128) = '{object_kind}';
-            DECLARE @column_name VARCHAR(128) = '{column_name}';
+            DECLARE @column_name VARCHAR(128) = {column_name};
             DECLARE @existing sql_variant;
 
             SELECT TOP 1 @existing = CAST(VALUE AS NVARCHAR) FROM fn_listextendedproperty(@property_name, 'schema', @schema_name, @object_kind, @object_name, 'column', @column_name);
@@ -522,14 +524,17 @@ def _build_create_comment_column_exp(
                         EXEC sp_updateextendedproperty @property_name, @comment, 'schema', @schema_name, @object_kind, @object_name, 'column', @column_name;
                 END
         """)
+
         tsql_text = template.format(
-            comment=exp.Literal.string(column_comment).sql(dialect=self.dialect)
-            if column_comment is not None
-            else "NULL",
-            schema_name=table.db if table.db else "dbo",
-            object_name=table.name,
+            comment=exp.Literal.string(column_comment or "NULL").sql(
+                dialect=self.dialect, identify=False
+            ),
+            schema_name=exp.Literal.string(table.db or "dbo").sql(
+                dialect=self.dialect, identify=False
+            ),
+            object_name=exp.Literal.string(table.name).sql(dialect=self.dialect, identify=False),
             object_kind=table_kind,
-            column_name=column_name,
+            column_name=exp.Literal.string(column_name).sql(dialect=self.dialect, identify=False),
         )
 
         return tsql_text