sqlobject-discuss

[SQLObject] suggestion: sqlbuilder magic query object accept foreign key

[Karl C. <qu...@no...>]

I request/suggest syntax sugar so that you can do this:
  k = OtherTable.get(123)
  MyTable.select(MyTable.q.other == k)
Instead of only:
  MyTable.select(MyTable.q.otherID == k.id)


Second, I am trying to add a hook so that I can do this:
  MyTable.select(OtherTable.q.foo == 123)
and have it automatically add an ID match to get:
  MyTable.select(AND( OtherTable.q.foo == 123,
                      MyTable.q.otherID == OtherTable.q.id))

any suggestions on how to do this?


-- 
Karl 2004-06-21 02:48

[SQLObject] Re: suggestion: sqlbuilder magic query object accept foreign key

[Karl C. <qu...@no...>]

I ended up writing this for myself; if anyone is interested here's
the patch.  Now you can do:
  PhoneNumber.selectBy(user=myuser) 
in addition to
  PhoneNumber.selectBy(userID=myuser.id)



Index: dbconnection.py
===================================================================
--- dbconnection.py	(revision 1186)
+++ dbconnection.py	(working copy)
@@ -418,12 +418,20 @@
                     self.sqlrepr(firstValue),
                     self.sqlrepr(secondValue)))
 
+    def _SO_kwpair2dbpair(self, soClass, key, value):
+        dbcolumn = soClass._SO_columnDict.get(key)
+        if dbcolumn:
+            return (dbcolumn.dbName, self.sqlrepr(value))
+        dbcolumnid = soClass._SO_columnDict.get(key+'ID')
+        if dbcolumnid:
+            return (dbcolumnid.dbName, self.sqlrepr(value.id))
+        raise Exception, 'Invalid keyword "%s" to selectBy' %(key)
+
     def _SO_columnClause(self, soClass, kw):
         return ' AND '.join(['%s = %s' %
-                         (soClass._SO_columnDict[key].dbName,
-                          self.sqlrepr(value))
-                         for key, value
-                         in kw.items()])
+                             self._SO_kwpair2dbpair(soClass, key, value)
+                             for key, value
+                             in kw.items()])
 
     def sqlrepr(self, v):
         return sqlrepr(v, self.dbName)


-- 
Karl 2004-06-24 04:51