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())