Skip to content

Commit 889ad5f

Browse files
authored
Add "request" as a parameter in the methods get_list_query and get_count_query (#592)
* Add "request" as a parameter in the method get_list_query and get_count_query * Add "request" as a parameter in the method get_list_query and get_count_query
1 parent 5ccd356 commit 889ad5f

File tree

1 file changed

+8
-8
lines changed
  • starlette_admin/contrib/sqla

1 file changed

+8
-8
lines changed

starlette_admin/contrib/sqla/view.py

+8-8
Original file line numberDiff line numberDiff line change
@@ -176,7 +176,7 @@ async def handle_row_action(
176176
except SQLAlchemyError as exc:
177177
raise ActionFailed(str(exc)) from exc
178178

179-
def get_list_query(self) -> Select:
179+
def get_list_query(self, request: Request) -> Select:
180180
"""
181181
Return a Select expression which is used as base statement for
182182
[find_all][starlette_admin.views.BaseModelView.find_all] method.
@@ -185,10 +185,10 @@ def get_list_query(self) -> Select:
185185
```python hl_lines="3-4"
186186
class PostView(ModelView):
187187
188-
def get_list_query(self):
188+
def get_list_query(self, request: Request):
189189
return super().get_list_query().where(Post.published == true())
190190
191-
def get_count_query(self):
191+
def get_count_query(self, request: Request):
192192
return super().get_count_query().where(Post.published == true())
193193
```
194194
@@ -198,7 +198,7 @@ def get_count_query(self):
198198
"""
199199
return select(self.model)
200200

201-
def get_count_query(self) -> Select:
201+
def get_count_query(self, request: Request) -> Select:
202202
"""
203203
Return a Select expression which is used as base statement for
204204
[count][starlette_admin.views.BaseModelView.count] method.
@@ -207,10 +207,10 @@ def get_count_query(self) -> Select:
207207
```python hl_lines="6-7"
208208
class PostView(ModelView):
209209
210-
def get_list_query(self):
210+
def get_list_query(self, request: Request):
211211
return super().get_list_query().where(Post.published == true())
212212
213-
def get_count_query(self):
213+
def get_count_query(self, request: Request):
214214
return super().get_count_query().where(Post.published == true())
215215
```
216216
"""
@@ -252,7 +252,7 @@ async def count(
252252
where: Union[Dict[str, Any], str, None] = None,
253253
) -> int:
254254
session: Union[Session, AsyncSession] = request.state.session
255-
stmt = self.get_count_query()
255+
stmt = self.get_count_query(request)
256256
if where is not None:
257257
if isinstance(where, dict):
258258
where = build_query(where, self.model)
@@ -274,7 +274,7 @@ async def find_all(
274274
order_by: Optional[List[str]] = None,
275275
) -> Sequence[Any]:
276276
session: Union[Session, AsyncSession] = request.state.session
277-
stmt = self.get_list_query().offset(skip)
277+
stmt = self.get_list_query(request).offset(skip)
278278
if limit > 0:
279279
stmt = stmt.limit(limit)
280280
if where is not None:

0 commit comments

Comments
 (0)