[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

dbt/adapters/sqlserver/sql_server_connection_manager.py

@@ -1,6 +1,6 @@
+import datetime as dt
 import struct
 import time
-import datetime as dt
 from contextlib import contextmanager
 from itertools import chain, repeat
 from typing import Any, Callable, Dict, Mapping, Optional, Tuple
@@ -237,16 +237,16 @@ def bool_to_connection_string_arg(key: str, value: bool) -> str:
 def byte_array_to_datetime(value: bytes) -> dt.datetime:
     """
     Converts a DATETIMEOFFSET byte array to a timezone-aware datetime object
-    
+
     Parameters
     ----------
     value : buffer
         A binary value conforming to SQL_SS_TIMESTAMPOFFSET_STRUCT
-    
+
     Returns
     -------
     out : datetime
-    
+
     Source
     ------
     SQL_SS_TIMESTAMPOFFSET datatype and SQL_SS_TIMESTAMPOFFSET_STRUCT layout:
@@ -263,10 +263,11 @@ def byte_array_to_datetime(value: bytes) -> dt.datetime:
         hour=tup[3],
         minute=tup[4],
         second=tup[5],
-        microsecond=tup[6] // 1000, # https://bugs.python.org/issue15443
+        microsecond=tup[6] // 1000,  # https://bugs.python.org/issue15443
         tzinfo=dt.timezone(dt.timedelta(hours=tup[7], minutes=tup[8])),
     )
 
+
 class SQLServerConnectionManager(SQLConnectionManager):
     TYPE = "sqlserver"
 
@@ -431,7 +432,7 @@ def add_query(
             # convert DATETIMEOFFSET binary structures to datetime ojbects
             # https://github.com/mkleehammer/pyodbc/issues/134#issuecomment-281739794
             connection.handle.add_output_converter(-155, byte_array_to_datetime)
-            
+
             logger.debug(
                 "SQL status: {} in {:0.2f} seconds".format(
                     self.get_response(cursor), (time.time() - pre)

tests/unit/adapters/sqlserver/test_sql_server_connection_manager.py

@@ -1,5 +1,4 @@
 import datetime as dt
-import struct
 import json
 from unittest import mock
 
@@ -8,8 +7,8 @@
 
 from dbt.adapters.sqlserver.sql_server_connection_manager import (
     bool_to_connection_string_arg,
+    byte_array_to_datetime,
     get_pyodbc_attrs_before,
-    byte_array_to_datetime
 )
 from dbt.adapters.sqlserver.sql_server_credentials import SQLServerCredentials
 
@@ -79,35 +78,52 @@ def test_get_pyodbc_attrs_before_contains_access_token_key_for_cli_authenticatio
 def test_bool_to_connection_string_arg(key: str, value: bool, expected: str) -> None:
     assert bool_to_connection_string_arg(key, value) == expected
 
+
 @pytest.mark.parametrize(
-    "value, expected_datetime, expected_str", [
+    "value, expected_datetime, expected_str",
+    [
         (
-            bytes([
-                0xE5, 0x07,             #  2022       year            unsigned short
-                0x0C, 0x00,             #  12         month           unsigned short
-                0x11, 0x00,             #  17         day             unsigned short
-                0x16, 0x00,             #  17         hour            unsigned short
-                0x16, 0x00,             #  52         minute          unsigned short
-                0x12, 0x00,             #  18         second          unsigned short
-                0xBC, 0xCC, 0x5B, 0x07, #  123456700  10⁻⁷ second     unsigned long
-                0xFE, 0xFF,             # -2          offset hour     signed short
-                0xE2, 0xFF              # -30         offset minute   signed short
-            ]),
+            bytes(
+                [
+                    0xE5,
+                    0x07,  #  2022       year            unsigned short
+                    0x0C,
+                    0x00,  #  12         month           unsigned short
+                    0x11,
+                    0x00,  #  17         day             unsigned short
+                    0x16,
+                    0x00,  #  17         hour            unsigned short
+                    0x16,
+                    0x00,  #  52         minute          unsigned short
+                    0x12,
+                    0x00,  #  18         second          unsigned short
+                    0xBC,
+                    0xCC,
+                    0x5B,
+                    0x07,  #  123456700  10⁻⁷ second     unsigned long
+                    0xFE,
+                    0xFF,  # -2          offset hour     signed short
+                    0xE2,
+                    0xFF,  # -30         offset minute   signed short
+                ]
+            ),
             dt.datetime(
                 year=2022,
                 month=12,
                 day=17,
                 hour=17,
                 minute=52,
                 second=18,
-                microsecond=123456700 // 1000, # 10⁻⁶ second
-                tzinfo=dt.timezone(dt.timedelta(hours=-2, minutes=-30))
+                microsecond=123456700 // 1000,  # 10⁻⁶ second
+                tzinfo=dt.timezone(dt.timedelta(hours=-2, minutes=-30)),
             ),
-            "2021-12-17 17:52:18.123456-02:30"
+            "2021-12-17 17:52:18.123456-02:30",
         )
-    ]
+    ],
 )
-def test_byte_array_to_datetime(value: bytes, expected_datetime: dt.datetime, expected_str: str) -> None:
+def test_byte_array_to_datetime(
+    value: bytes, expected_datetime: dt.datetime, expected_str: str
+) -> None:
     """
     Assert SQL_SS_TIMESTAMPOFFSET_STRUCT bytes are converted to datetime and str
     https://learn.microsoft.com/sql/relational-databases/native-client-odbc-date-time/data-type-support-for-odbc-date-and-time-improvements#sql_ss_timestampoffset_struct