Skip to content

Commit d6b5ce0

Browse files
committed
Avoid private SQLAlchemy API
1 parent 3afa36b commit d6b5ce0

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/flask_sqlalchemy/query.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,8 +27,9 @@ def get_or_404(self, ident: t.Any, description: str | None = None) -> t.Any:
2727
:param ident: The primary key to query.
2828
:param description: A custom message to show on the error page.
2929
"""
30-
mapper = self._only_full_mapper_zero("get")
31-
rv = self.session.get(mapper.class_, ident)
30+
model = (self.column_descriptions[0].get("entity")
31+
or self.column_descriptions[0]["type"])
32+
rv = self.session.get(model, ident)
3233

3334
if rv is None:
3435
abort(404, description=description)

0 commit comments

Comments
 (0)