Bugfix - excluded primary key caused a KeyError

mx-moth · Feb 6, 2020 · 098fc12 · 098fc12
1 parent 18a8a8e
commit 098fc12
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/wtforms_sqlalchemy/orm.py b/wtforms_sqlalchemy/orm.py
@@ -256,7 +256,7 @@ def model_fields(model, db_session=None, only=None, exclude=None,
 
     if only:
         order = list(only)
-        properties = {key: properties[key] for key in only}
+        properties = {key: properties[key] for key in only if key in properties}
     elif exclude:
         properties = {key: prop for key, prop in properties.items() if key not in exclude}