summaryrefslogtreecommitdiff
path: root/lib/sqlalchemy/orm
diff options
context:
space:
mode:
authormike bayer <mike_mp@zzzcomputing.com>2021-10-29 14:22:41 +0000
committerGerrit Code Review <gerrit@ci3.zzzcomputing.com>2021-10-29 14:22:41 +0000
commit5aca98e65f471207439471210a1e232c07f0a3ba (patch)
tree367e8481c71841480a157c3069c646fe08a82383 /lib/sqlalchemy/orm
parentd127f4936acb28dc00efd84678336a2be935a312 (diff)
parent6a786297014c7b2a381084299ac3b4aabcf3c318 (diff)
downloadsqlalchemy-5aca98e65f471207439471210a1e232c07f0a3ba.tar.gz
Merge "Modernize tests - session_query_get" into main
Diffstat (limited to 'lib/sqlalchemy/orm')
-rw-r--r--lib/sqlalchemy/orm/session.py11
1 files changed, 8 insertions, 3 deletions
diff --git a/lib/sqlalchemy/orm/session.py b/lib/sqlalchemy/orm/session.py
index d5fb8a8e1..bb12f7021 100644
--- a/lib/sqlalchemy/orm/session.py
+++ b/lib/sqlalchemy/orm/session.py
@@ -2793,6 +2793,11 @@ class Session(_SessionClassMethods):
mapper = inspect(entity)
+ if not mapper or not mapper.is_mapper:
+ raise sa_exc.ArgumentError(
+ "Expected mapped class or mapper, got: %r" % entity
+ )
+
is_dict = isinstance(primary_key_identity, dict)
if not is_dict:
primary_key_identity = util.to_list(
@@ -2802,8 +2807,8 @@ class Session(_SessionClassMethods):
if len(primary_key_identity) != len(mapper.primary_key):
raise sa_exc.InvalidRequestError(
"Incorrect number of values in identifier to formulate "
- "primary key for query.get(); primary key columns are %s"
- % ",".join("'%s'" % c for c in mapper.primary_key)
+ "primary key for session.get(); primary key columns "
+ "are %s" % ",".join("'%s'" % c for c in mapper.primary_key)
)
if is_dict:
@@ -2817,7 +2822,7 @@ class Session(_SessionClassMethods):
util.raise_(
sa_exc.InvalidRequestError(
"Incorrect names of values in identifier to formulate "
- "primary key for query.get(); primary key attribute "
+ "primary key for session.get(); primary key attribute "
"names are %s"
% ",".join(
"'%s'" % prop.key