format

Author: dlawin

data_diff/databases/_connect.py

@@ -30,7 +30,7 @@
     "trino": Trino,
     "clickhouse": Clickhouse,
     "vertica": Vertica,
-    "mssql": MsSql
+    "mssql": MsSql,
 }
 
 

data_diff/sqeleton/abcs/database_types.py

@@ -224,7 +224,9 @@ def current_schema(self) -> str:
         "Provide SQL for returning the current default schema."
 
     @abstractmethod
-    def offset_limit(self, offset: Optional[int] = None, limit: Optional[int] = None, has_order_by: Optional[bool] = None) -> str:
+    def offset_limit(
+        self, offset: Optional[int] = None, limit: Optional[int] = None, has_order_by: Optional[bool] = None
+    ) -> str:
         "Provide SQL fragment for limit and offset inside a select"
 
     @abstractmethod

data_diff/sqeleton/databases/_connect.py

@@ -87,7 +87,7 @@ def match_path(self, dsn):
     "trino": Trino,
     "clickhouse": Clickhouse,
     "vertica": Vertica,
-    "mssql": MsSQL
+    "mssql": MsSQL,
 }
 
 

data_diff/sqeleton/databases/base.py

@@ -155,7 +155,9 @@ class BaseDialect(AbstractDialect):
 
     PLACEHOLDER_TABLE = None  # Used for Oracle
 
-    def offset_limit(self, offset: Optional[int] = None, limit: Optional[int] = None, has_order_by: Optional[bool] = None) -> str:
+    def offset_limit(
+        self, offset: Optional[int] = None, limit: Optional[int] = None, has_order_by: Optional[bool] = None
+    ) -> str:
         if offset:
             raise NotImplementedError("No support for OFFSET in query")
 

data_diff/sqeleton/databases/mssql.py

@@ -54,16 +54,16 @@ def normalize_number(self, value: str, coltype: FractionalType) -> str:
 
         return f"FORMAT({value}, 'N{coltype.precision}')"
 
+
 class Mixin_MD5(AbstractMixin_MD5):
     def md5_as_int(self, s: str) -> str:
         # TODO FYI
         # I didn't see another way to do this
         # keep in mind this also required reducing
         # CHECKSUM_HEXDIGITS = 15 -> 14 due to the convert needing an even number of hex digits
         # (affects all dbs)
-        return (
-            f"convert(bigint, convert(varbinary, '0x' + RIGHT(CONVERT(NVARCHAR(32), HashBytes('MD5', {s}), 2), {CHECKSUM_HEXDIGITS}), 1))"
-        )
+        return f"convert(bigint, convert(varbinary, '0x' + RIGHT(CONVERT(NVARCHAR(32), HashBytes('MD5', {s}), 2), {CHECKSUM_HEXDIGITS}), 1))"
+
 
 class Dialect(BaseDialect, Mixin_Schema, Mixin_OptimizerHints):
     name = "MsSQL"
@@ -139,11 +139,11 @@ def random(self) -> str:
     def is_distinct_from(self, a: str, b: str) -> str:
         # IS (NOT) DISTINCT FROM is available only since SQLServer 2022.
         # See: https://stackoverflow.com/a/18684859/857383
-        return (
-            f"(({a}<>{b} OR {a} IS NULL OR {b} IS NULL) AND NOT({a} IS NULL AND {b} IS NULL))"
-        )
+        return f"(({a}<>{b} OR {a} IS NULL OR {b} IS NULL) AND NOT({a} IS NULL AND {b} IS NULL))"
 
-    def offset_limit(self, offset: Optional[int] = None, limit: Optional[int] = None, has_order_by: Optional[bool] = None) -> str:
+    def offset_limit(
+        self, offset: Optional[int] = None, limit: Optional[int] = None, has_order_by: Optional[bool] = None
+    ) -> str:
         if offset:
             raise NotImplementedError("No support for OFFSET in query")
 

data_diff/sqeleton/databases/oracle.py

@@ -104,7 +104,9 @@ def quote(self, s: str):
     def to_string(self, s: str):
         return f"cast({s} as varchar(1024))"
 
-    def offset_limit(self, offset: Optional[int] = None, limit: Optional[int] = None, has_order_by: Optional[bool] = None) -> str:
+    def offset_limit(
+        self, offset: Optional[int] = None, limit: Optional[int] = None, has_order_by: Optional[bool] = None
+    ) -> str:
         if offset:
             raise NotImplementedError("No support for OFFSET in query")
 

tests/sqeleton/common.py

@@ -30,7 +30,7 @@
     TEST_DUCKDB_CONN_STRING,
     N_THREADS,
     TEST_ACROSS_ALL_DBS,
-    TEST_MSSQL_CONN_STRING
+    TEST_MSSQL_CONN_STRING,
 )
 
 

tests/sqeleton/test_database.py

@@ -13,7 +13,6 @@
 
 TEST_DATABASES = {
     # TODO dev
-
     # dbs.MySQL,
     # dbs.PostgreSQL,
     # dbs.Oracle,

tests/sqeleton/test_query.py

@@ -47,7 +47,9 @@ def current_database(self) -> str:
     def current_schema(self) -> str:
         return "current_schema()"
 
-    def offset_limit(self, offset: Optional[int] = None, limit: Optional[int] = None, has_order_by: Optional[bool] = None) -> str:
+    def offset_limit(
+        self, offset: Optional[int] = None, limit: Optional[int] = None, has_order_by: Optional[bool] = None
+    ) -> str:
         x = offset and f"OFFSET {offset}", limit and f"LIMIT {limit}"
         return " ".join(filter(None, x))