From 70469bf26092095590aa441f41f5bcde11968ffc Mon Sep 17 00:00:00 2001 From: Mike Bayer Date: Thu, 13 Jul 2006 16:36:31 +0000 Subject: [PATCH] added count/count_by to assignmapper, plus a test in activemapper to try it out --- CHANGES | 2 ++ lib/sqlalchemy/ext/assignmapper.py | 2 +- test/ext/activemapper.py | 2 ++ 3 files changed, 5 insertions(+), 1 deletion(-) diff --git a/CHANGES b/CHANGES index acf61f8702..d080ad9eab 100644 --- a/CHANGES +++ b/CHANGES @@ -9,6 +9,8 @@ activated when activemapper is imported - fixes to Session expunge/update/etc. - select_table mappers *still* werent always compiling - fixed up Boolean datatype +- added count()/count_by() to list of methods proxied by assignmapper; +this also adds them to activemapper 0.2.5 - fixed endless loop bug in select_by(), if the traversal hit diff --git a/lib/sqlalchemy/ext/assignmapper.py b/lib/sqlalchemy/ext/assignmapper.py index 29cc70b3e4..240a46e3b1 100644 --- a/lib/sqlalchemy/ext/assignmapper.py +++ b/lib/sqlalchemy/ext/assignmapper.py @@ -31,7 +31,7 @@ def assign_mapper(ctx, class_, *args, **kwargs): extension = ctx.mapper_extension m = mapper(class_, extension=extension, *args, **kwargs) class_.mapper = m - for name in ['get', 'select', 'select_by', 'selectone', 'get_by', 'join_to', 'join_via']: + for name in ['get', 'select', 'select_by', 'selectone', 'get_by', 'join_to', 'join_via', 'count', 'count_by']: monkeypatch_query_method(ctx, class_, name) for name in ['flush', 'delete', 'expire', 'refresh', 'expunge', 'merge', 'save', 'update', 'save_or_update']: monkeypatch_objectstore_method(ctx, class_, name) diff --git a/test/ext/activemapper.py b/test/ext/activemapper.py index 85466e29b6..1ff8b0d7b1 100644 --- a/test/ext/activemapper.py +++ b/test/ext/activemapper.py @@ -218,6 +218,8 @@ class testcase(testbase.PersistTest): ) self.assertEquals(len(results), 1) + self.assertEquals(Person.count(), 2) + class testselfreferential(testbase.PersistTest): def setUpAll(self): global TreeNode -- 2.47.3