From: Elrond <elrond+mediagoblin.org@samba-tng.org>
Date: Sat, 24 Dec 2011 18:08:20 +0000 (+0100)
Subject: Add search level one() method
X-Git-Url: https://vcs.fsf.org/?a=commitdiff_plain;h=2608982885477e2f41579240d24a26864f718123;p=mediagoblin.git

Add search level one() method

And create a _fix_query_dict which converts '_id' to 'id'.
---

diff --git a/mediagoblin/db/sql/base.py b/mediagoblin/db/sql/base.py
index b8d5cc96..38b04334 100644
--- a/mediagoblin/db/sql/base.py
+++ b/mediagoblin/db/sql/base.py
@@ -4,13 +4,26 @@ from sqlalchemy.orm import scoped_session, sessionmaker
 Session = scoped_session(sessionmaker())
 
 
+def _fix_query_dict(query_dict):
+    if '_id' in query_dict:
+        query_dict['id'] = query_dict.pop('_id')
+
+
 class GMGTableBase(object):
     query = Session.query_property()
 
     @classmethod
     def find(cls, query_dict={}):
+        _fix_query_dict(query_dict)
         return cls.query.filter_by(**query_dict)
 
     @classmethod
     def find_one(cls, query_dict={}):
+        _fix_query_dict(query_dict)
         return cls.query.filter_by(**query_dict).first()
+
+    @classmethod
+    def one(cls, query_dict):
+        retval = cls.find_one(query_dict)
+        assert retval is not None
+        return retval