Ran black

Author: erezsh

sqeleton/abcs/mixins.py

@@ -146,14 +146,12 @@ def time_travel(
         Must specify exactly one of `timestamp`, `offset` or `statement`.
         """
 
+
 class AbstractMixin_OptimizerHints(AbstractMixin):
     @abstractmethod
-    def optimizer_hints(
-        self,
-        optimizer_hints: str
-    ) -> str:
+    def optimizer_hints(self, optimizer_hints: str) -> str:
         """Creates a compatible optimizer_hints string
 
         Parameters:
             optimizer_hints - string of optimizer hints
-        """
+        """

sqeleton/databases/base.py

@@ -40,8 +40,8 @@
     AbstractMixin_Schema,
     AbstractMixin_RandomSample,
     AbstractMixin_NormalizeValue,
-    AbstractMixin_OptimizerHints
-) 
+    AbstractMixin_OptimizerHints,
+)
 from ..bound_exprs import bound_table
 
 logger = logging.getLogger("database")
@@ -140,10 +140,7 @@ def random_sample_ratio_approx(self, tbl: AbstractTable, ratio: float) -> Abstra
 
 
 class Mixin_OptimizerHints(AbstractMixin_OptimizerHints):
-    def optimizer_hints(
-        self,
-        hints: str
-    ) -> str:
+    def optimizer_hints(self, hints: str) -> str:
         return f"/*+ {hints} */ "
 
 
@@ -181,7 +178,6 @@ def current_timestamp(self) -> str:
     def explain_as_text(self, query: str) -> str:
         return f"EXPLAIN {query}"
 
-
     def _constant_value(self, v):
         if v is None:
             return "NULL"

sqeleton/databases/databricks.py

@@ -95,7 +95,6 @@ def set_timezone_to_utc(self) -> str:
         return "SET TIME ZONE 'UTC'"
 
 
-
 class Databricks(ThreadedDatabase):
     dialect = Dialect()
     CONNECT_URI_HELP = "databricks://:<access_token>@<server_hostname>/<http_path>"

sqeleton/databases/oracle.py

@@ -17,7 +17,15 @@
 from ..abcs.mixins import AbstractMixin_MD5, AbstractMixin_NormalizeValue, AbstractMixin_Schema
 from ..abcs import Compilable
 from ..queries import this, table, SKIP
-from .base import BaseDialect, Mixin_OptimizerHints, ThreadedDatabase, import_helper, ConnectError, QueryError, Mixin_RandomSample
+from .base import (
+    BaseDialect,
+    Mixin_OptimizerHints,
+    ThreadedDatabase,
+    import_helper,
+    ConnectError,
+    QueryError,
+    Mixin_RandomSample,
+)
 from .base import TIMESTAMP_PRECISION_POS
 
 SESSION_TIME_ZONE = None  # Changed by the tests

sqeleton/repl.py

@@ -48,7 +48,7 @@ def repl(uri):
                 continue
             try:
                 path = db.parse_table_name(table_name)
-                print('->', path)
+                print("->", path)
                 schema = db.query_table_schema(path)
             except Exception as e:
                 logging.error(e)

sqeleton/utils.py

@@ -1,4 +1,16 @@
-from typing import Iterable, Iterator, MutableMapping, Union, Any, Sequence, Dict, Hashable, TypeVar, TYPE_CHECKING, List
+from typing import (
+    Iterable,
+    Iterator,
+    MutableMapping,
+    Union,
+    Any,
+    Sequence,
+    Dict,
+    Hashable,
+    TypeVar,
+    TYPE_CHECKING,
+    List,
+)
 from abc import abstractmethod
 from weakref import ref
 import math
@@ -256,7 +268,6 @@ def __eq__(self, other):
             return NotImplemented
         return self._str == other._str
 
-
     def new(self, *args, **kw):
         return type(self)(*args, **kw, max_len=self._max_len)
 

tests/test_query.py

@@ -197,17 +197,19 @@ def test_select_with_optimizer_hints(self):
         c = Compiler(MockDatabase())
         t = table("a")
 
-        q = c.compile(t.select(this.b, optimizer_hints='PARALLEL(a 16)'))
+        q = c.compile(t.select(this.b, optimizer_hints="PARALLEL(a 16)"))
         assert q == "SELECT /*+ PARALLEL(a 16) */ b FROM a"
 
-        q = c.compile(t.where(this.b > 10).select(this.b, optimizer_hints='PARALLEL(a 16)'))
+        q = c.compile(t.where(this.b > 10).select(this.b, optimizer_hints="PARALLEL(a 16)"))
         self.assertEqual(q, "SELECT /*+ PARALLEL(a 16) */ b FROM a WHERE (b > 10)")
 
-        q = c.compile(t.limit(10).select(this.b, optimizer_hints='PARALLEL(a 16)'))
+        q = c.compile(t.limit(10).select(this.b, optimizer_hints="PARALLEL(a 16)"))
         self.assertEqual(q, "SELECT /*+ PARALLEL(a 16) */ b FROM (SELECT * FROM a LIMIT 10) tmp1")
 
-        q = c.compile(t.select(this.a).group_by(this.b).agg(this.c).select(optimizer_hints='PARALLEL(a 16)'))
-        self.assertEqual(q, "SELECT /*+ PARALLEL(a 16) */ * FROM (SELECT b, c FROM (SELECT a FROM a) tmp2 GROUP BY 1) tmp3")
+        q = c.compile(t.select(this.a).group_by(this.b).agg(this.c).select(optimizer_hints="PARALLEL(a 16)"))
+        self.assertEqual(
+            q, "SELECT /*+ PARALLEL(a 16) */ * FROM (SELECT b, c FROM (SELECT a FROM a) tmp2 GROUP BY 1) tmp3"
+        )
 
     def test_table_ops(self):
         c = Compiler(MockDatabase())