Skip to content

Commit

Permalink
Make interface compatible with Python 2.7
Browse files Browse the repository at this point in the history
  • Loading branch information
ods committed Jun 18, 2020
1 parent 57130d8 commit 5ad7f69
Show file tree
Hide file tree
Showing 6 changed files with 11 additions and 11 deletions.
2 changes: 1 addition & 1 deletion clickhouse_sqlalchemy/drivers/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -109,7 +109,7 @@ def limit_by_clause(self, select, **kw):
text += self.process(limit_by_clause.offset, **kw) + ', '
text += self.process(limit_by_clause.limit, **kw)
limit_by_exprs = limit_by_clause.by_clauses._compiler_dispatch(
self, **kw,
self, **kw
)
text += ' BY ' + limit_by_exprs

Expand Down
4 changes: 2 additions & 2 deletions clickhouse_sqlalchemy/ext/clauses.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,9 +33,9 @@ def sample_clause(element):

class LimitByClause:

def __init__(self, by_clauses, offset, limit):
def __init__(self, by_clauses, limit, offset):
self.by_clauses = ClauseList(
*by_clauses, _literal_as_text=_literal_as_label_reference,
*by_clauses, _literal_as_text=_literal_as_label_reference
)
self.offset = _offset_or_limit_clause(offset)
self.limit = _offset_or_limit_clause(limit)
Expand Down
4 changes: 2 additions & 2 deletions clickhouse_sqlalchemy/orm/query.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,8 +55,8 @@ def sample(self, sample):
self._sample = sample

@_generative()
def limit_by(self, *by_clauses, offset=None, limit):
self._limit_by = LimitByClause(by_clauses, offset, limit)
def limit_by(self, by_clauses, limit, offset=None):
self._limit_by = LimitByClause(by_clauses, limit, offset)

def join(self, *props, **kwargs):
type = kwargs.pop('type', None)
Expand Down
4 changes: 2 additions & 2 deletions clickhouse_sqlalchemy/sql/selectable.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,8 +50,8 @@ def sample(self, sample):
self._sample_clause = sample_clause(sample)

@_generative
def limit_by(self, *by_clauses, offset=None, limit):
self._limit_by_clause = LimitByClause(by_clauses, offset, limit)
def limit_by(self, by_clauses, limit, offset=None):
self._limit_by_clause = LimitByClause(by_clauses, limit, offset)

@_generative
def array_join(self, *columns):
Expand Down
4 changes: 2 additions & 2 deletions tests/orm/test_select.py
Original file line number Diff line number Diff line change
Expand Up @@ -107,7 +107,7 @@ def test_limit_by(self):
table = self._make_table()

query = self.session.query(table.c.x).order_by(table.c.x)\
.limit_by(table.c.x, limit=1)
.limit_by([table.c.x], limit=1)
self.assertEqual(
self.compile(query),
'SELECT t1.x AS t1_x FROM t1 ORDER BY t1.x '
Expand All @@ -122,7 +122,7 @@ def test_limit_by_with_offset(self):
table = self._make_table()

query = self.session.query(table.c.x).order_by(table.c.x)\
.limit_by(table.c.x, offset=1, limit=2)
.limit_by([table.c.x], offset=1, limit=2)
self.assertEqual(
self.compile(query),
'SELECT t1.x AS t1_x FROM t1 ORDER BY t1.x '
Expand Down
4 changes: 2 additions & 2 deletions tests/sql/test_selectable.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,7 @@ def test_limit_by(self):
table = self._make_table()

query = select([table.c.x]).order_by(table.c.x)\
.limit_by(table.c.x, limit=1)
.limit_by([table.c.x], limit=1)
self.assertEqual(
self.compile(query),
'SELECT t1.x FROM t1 ORDER BY t1.x LIMIT %(param_1)s BY t1.x'
Expand All @@ -75,7 +75,7 @@ def test_limit_by_with_offset(self):
table = self._make_table()

query = select([table.c.x]).order_by(table.c.x)\
.limit_by(table.c.x, offset=1, limit=2)
.limit_by([table.c.x], offset=1, limit=2)
self.assertEqual(
self.compile(query),
'SELECT t1.x FROM t1 ORDER BY t1.x '
Expand Down

0 comments on commit 5ad7f69

Please sign in to comment.