Skip to content

Commit 9c2f2b2

Browse files
author
parkervg
committed
fix dialect inheritance
1 parent 4ca04ff commit 9c2f2b2

2 files changed

Lines changed: 4 additions & 10 deletions

File tree

blendsql/parse/dialect.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -185,17 +185,17 @@ def generate(self, expression, **opts) -> str:
185185
return _postprocess_blendsql_sql(sql)
186186

187187

188-
class BlendSQLDuckDB(BlendSQLDialect, DuckDB):
188+
class BlendSQLDuckDB(DuckDB, BlendSQLDialect):
189189
class Tokenizer(BlendSQLDialect.Tokenizer, DuckDB.Tokenizer):
190190
pass
191191

192192

193-
class BlendSQLPostgres(BlendSQLDialect, Postgres):
193+
class BlendSQLPostgres(Postgres, BlendSQLDialect):
194194
class Tokenizer(BlendSQLDialect.Tokenizer, Postgres.Tokenizer):
195195
pass
196196

197197

198-
class BlendSQLSQLite(BlendSQLDialect, SQLite):
198+
class BlendSQLSQLite(SQLite, BlendSQLDialect):
199199
class Tokenizer(BlendSQLDialect.Tokenizer, SQLite.Tokenizer):
200200
KEYWORDS = {
201201
**SQLite.Tokenizer.KEYWORDS,

blendsql/smoothie.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -46,17 +46,11 @@ def print_summary(self):
4646
from rich.table import Table
4747
from rich.columns import Columns
4848
from rich.box import ROUNDED
49-
from blendsql.parse.dialect import get_dialect
50-
import sqlglot
5149

5250
console = Console(force_terminal=True)
5351

54-
# Create SQL syntax highlighted query
55-
formatted_query = sqlglot.transpile(
56-
self.meta.query, read=get_dialect(self.meta.db_type), pretty=True
57-
)[0]
5852
query_syntax = Syntax(
59-
formatted_query, "sql", theme="default", dedent=True, word_wrap=True
53+
self.meta.query, "sql", theme="default", dedent=True, word_wrap=True
6054
)
6155

6256
# Create summary table

0 commit comments

Comments
 (0)