Changeset 178 in tracdarcs for tracdarcs/dbutil.py

Timestamp:

05/12/10 18:40:53 (2 years ago)

Author:

lele@…

Hash name:

20100512164053-97f81-e69d930ef723d57a771c19c6a23bc5ed9781b30a

Message:

Use a parametrized query

File:

• tracdarcs/dbutil.py (modified) (2 diffs)

tracdarcs/dbutil.py

@@ -35 +35 @@
 IS_TRAC_0_12_OR_BETTER = not not re.match(r'^0\.1[2-9]', tracVersion)
 
-def query_nodes_for_revision(repo_id, rev=None):
+def query_nodes_for_revision(repo_id, rev=None, cond=None):
     '''
     Returns a query which represents the nodes available as of
@@ -44 +44 @@
     if rev is None:
         revfilter = "dn.remove_rev IS NULL"
+        revargs = []
     else:
-        revfilter = ("dnc2.rev <= %d"
-                     " AND ((dn.remove_rev IS NULL) "
-                     "      OR (dn.remove_rev > %d))" % (rev,rev))
+        revfilter = ("dnc2.rev <= %s"
+                     " AND ((dn.remove_rev IS NULL)"
+                     " OR (dn.remove_rev > %s))")
+        revargs = [rev, rev]
+    if cond is None:
+        cond = ''
+    else:
+        cond = 'AND %s' % cond
     q = """SELECT dnc.node_id, dnc.rev, dnc.path, dnc.parent_id
            FROM darcs_node_changes AS dnc
-           WHERE dnc.repo_id = %s
+           WHERE dnc.repo_id = %%s
              AND dnc.rev = (SELECT max(dnc2.rev)
-                            FROM darcs_node_changes AS dnc2, darcs_nodes AS dn
+                            FROM darcs_node_changes AS dnc2
+                              JOIN darcs_nodes AS dn
+                                   ON dn.repo_id = dnc2.repo_id
+                                      AND dn.node_id = dnc2.node_id
                             WHERE dnc2.repo_id = dnc.repo_id
                               AND dnc2.node_id = dnc.node_id
-                              AND dn.repo_id = dnc2.repo_id
-                              AND dn.node_id = dnc2.node_id
-                              AND %s)""" % (repo_id, revfilter)
-    return q
+                              AND %s)
+             %s""" % (revfilter, cond)
+    return q, [repo_id] + revargs
 
 def get_node_type(db, repo_id, node_id):