diff options
Diffstat (limited to 'lib/sqlalchemy/ext/assignmapper.py')
| -rw-r--r-- | lib/sqlalchemy/ext/assignmapper.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/lib/sqlalchemy/ext/assignmapper.py b/lib/sqlalchemy/ext/assignmapper.py index 8c78e7024..dddb25bdc 100644 --- a/lib/sqlalchemy/ext/assignmapper.py +++ b/lib/sqlalchemy/ext/assignmapper.py @@ -1,4 +1,4 @@ -from sqlalchemy import mapper, util, Query +from sqlalchemy import mapper, util, Query, exceptions import types def monkeypatch_query_method(ctx, class_, name): @@ -17,9 +17,13 @@ def monkeypatch_objectstore_method(ctx, class_, name): setattr(class_, name, do) def assign_mapper(ctx, class_, *args, **kwargs): + validate = kwargs.pop('validate', False) if not isinstance(getattr(class_, '__init__'), types.MethodType): def __init__(self, **kwargs): for key, value in kwargs.items(): + if validate: + if not key in self.mapper.props: + raise exceptions.ArgumentError("Invalid __init__ argument: '%s'" % key) setattr(self, key, value) class_.__init__ = __init__ extension = kwargs.pop('extension', None) |