- restored old assign_mapper monkey patched query methods but with two differences:

   * added a deprecation warning
   * check if a method with that name already exist in the class
 - more foolproof deprecation warning for scalar kwarg

1 files changed, 16 insertions, 2 deletions

diff --git a/lib/sqlalchemy/ext/assignmapper.py b/lib/sqlalchemy/ext/assignmapper.py
index 238041702..cab5a9eae 100644
--- a/lib/sqlalchemy/ext/assignmapper.py
+++ b/lib/sqlalchemy/ext/assignmapper.py
@@ -1,7 +1,19 @@
 from sqlalchemy import util, exceptions
 import types
-from sqlalchemy.orm import mapper
+from sqlalchemy.orm import mapper, Query
     
+def _monkeypatch_query_method(name, ctx, class_):
+    def do(self, *args, **kwargs):
+        query = Query(class_, session=ctx.current)
+        util.warn_deprecated('Query methods on the class are deprecated; use %s.query.%s instead' % (class_.__name__, name))
+        return getattr(query, name)(*args, **kwargs)
+    try:
+        do.__name__ = name
+    except:
+        pass
+    if not hasattr(class_, name): 
+        setattr(class_, name, classmethod(do))
+
 def _monkeypatch_session_method(name, ctx, class_):
     def do(self, *args, **kwargs):
         session = ctx.current
@@ -41,7 +53,9 @@ def assign_mapper(ctx, class_, *args, **kwargs):
     if not hasattr(class_, 'query'): 
         class_.query = query()
     
-    for name in ['refresh', 'expire', 'delete', 'expunge', 'update']:
+    for name in ('get', 'filter', 'filter_by', 'select', 'select_by', 'selectfirst', 'selectfirst_by', 'selectone', 'selectone_by', 'get_by', 'join_to', 'join_via', 'count', 'count_by', 'options', 'instances'):
+        _monkeypatch_query_method(name, ctx, class_)
+    for name in ('refresh', 'expire', 'delete', 'expunge', 'update'):
         _monkeypatch_session_method(name, ctx, class_)
 
     m = mapper(class_, extension=extension, *args, **kwargs)