override

[dak.git] / dak / stats.py

diff --git a/dak/stats.py b/dak/stats.py
index 20a02b55cac3c0d4345efe5f8d4017ed5a2f02e2..849c36b27cb6081265a23db0a9d751f757b11dca 100755 (executable)
--- a/dak/stats.py
+++ b/dak/stats.py
@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 
-# Various statistical pr0nography fun and games
+""" Various statistical pr0nography fun and games """
 # Copyright (C) 2000, 2001, 2002, 2003, 2006  James Troup <james@nocrew.org>
 
 # This program is free software; you can redistribute it and/or modify
@@ -33,6 +33,7 @@
 import pg, sys
 import apt_pkg
 from daklib import utils
+from daklib import database
 
 ################################################################################
 
@@ -179,7 +180,7 @@ SELECT suite, count(suite) FROM src_associations GROUP BY suite;""")
     for suite in suite_list:
         suite_id = suite_ids[suite]
         suite_arches[suite_id] = {}
-        for arch in Cnf.ValueList("Suite::%s::Architectures" % (suite)):
+        for arch in database.get_suite_architectures(suite_id):
             suite_arches[suite_id][arch] = ""
         suite_id_list.append(suite_id)
     output_list = [ output_format(i) for i in suite_list ]
@@ -234,6 +235,7 @@ def main ():
     mode = args[0].lower()
 
     projectB = pg.connect(Cnf["DB::Name"], Cnf["DB::Host"], int(Cnf["DB::Port"]))
+    database.init(Cnf, projectB)
 
     if mode == "arch-space":
         per_arch_space_use()