X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=dak%2Fmake_pkg_file_mapping.py;h=c457820fc0faf53d1e0e53dcbb3bc80e80832313;hb=9f523eb071132709bcb68234f9a8bd7c22d3565c;hp=60c39d3ec68dcb25a577a63feb6e944c60ebdd8a;hpb=cd5b29ddfd8de263c085f494b9573d683913f6f3;p=dak.git

diff --git a/dak/make_pkg_file_mapping.py b/dak/make_pkg_file_mapping.py
index 60c39d3e..c457820f 100755
--- a/dak/make_pkg_file_mapping.py
+++ b/dak/make_pkg_file_mapping.py
@@ -31,15 +31,7 @@ and binary package version it has in a standard rfc2822-like format.
 
 ################################################################################
 
-import os
-import pg
-import sys
-from daklib import database
-from daklib import utils
-
-################################################################################
-
-projectB = None #: database connection, pgobject
+from daklib.dbconn import *
 
 ################################################################################
 
@@ -74,14 +66,16 @@ def build_mapping():
     ORDER BY source, version, package, bin_version
     """
 
-    for row in projectB.query(query_sources).getresult():
+    session = DBConn().session()
+
+    for row in session.execute(query_sources).fetchall():
         (source, version, path) = row
         print "Path: %s"%path
         print "Source: %s"%source
         print "Source-Version: %s"%version
         print
 
-    for row in projectB.query(query_binaries).getresult():
+    for row in session.execute(query_binaries).fetchall():
         (source, version, arch, path, bin, binv) = row
         print "Path: %s"%path
         print "Source: %s"%source
@@ -94,10 +88,7 @@ def build_mapping():
 ################################################################################
 
 def main():
-    global projectB
-
-    Cnf = utils.get_conf()
-    projectB = pg.connect(Cnf["DB::Name"], Cnf["DB::Host"], int(Cnf["DB::Port"]))
+    DBConn()
     build_mapping()
 
 #########################################################################################