diff options
Diffstat (limited to 'lib/sqlalchemy/sql/base.py')
| -rw-r--r-- | lib/sqlalchemy/sql/base.py | 17 |
1 files changed, 11 insertions, 6 deletions
diff --git a/lib/sqlalchemy/sql/base.py b/lib/sqlalchemy/sql/base.py index da384bdab..7e9199bfa 100644 --- a/lib/sqlalchemy/sql/base.py +++ b/lib/sqlalchemy/sql/base.py @@ -43,13 +43,18 @@ def _from_objects(*elements): return itertools.chain(*[element._from_objects for element in elements]) -@util.decorator -def _generative(fn, *args, **kw): - """Mark a method as generative.""" +def _generative(fn): + @util.decorator + def _generative(fn, *args, **kw): + """Mark a method as generative.""" - self = args[0]._generate() - fn(self, *args[1:], **kw) - return self + self = args[0]._generate() + fn(self, *args[1:], **kw) + return self + + decorated = _generative(fn) + decorated.non_generative = fn + return decorated def _clone(element, **kw): |