Revert "Merge commit 'stew/content_generation' into merge"

[dak.git] / daklib / database.py

diff --git a/daklib/database.py b/daklib/database.py
index 90e16d5e002f4352026c752b14065aaf46a1c6f0..3fbd2a504992bae6b9555af6a4bb56812aa7acfb 100755 (executable)
--- a/daklib/database.py
+++ b/daklib/database.py
@@ -3,10 +3,11 @@
 """ DB access functions
 @group readonly: get_suite_id, get_section_id, get_priority_id, get_override_type_id,
                  get_architecture_id, get_archive_id, get_component_id, get_location_id,
-                 get_source_id, get_suite_version, get_files_id, get_maintainer, get_suites
+                 get_source_id, get_suite_version, get_files_id, get_maintainer, get_suites,
+                 get_suite_architectures
 @group read/write: get_or_set*, set_files_id
 
-@contact: Debian FTPMaster <ftpmaster@debian.org>
+@contact: Debian FTP Master <ftpmaster@debian.org>
 @copyright: 2000, 2001, 2002, 2003, 2004, 2006  James Troup <james@nocrew.org>
 @copyright: 2009  Joerg Jaspert <joerg@debian.org>
 @license: GNU General Public License version 2 or later
@@ -269,7 +270,7 @@ def get_component_id (component):
     Results are kept in a cache during runtime to minimize database queries.
 
     @type component: string
-    @param component: The name of the override type
+    @param component: The name of the component
 
     @rtype: int
     @return: the database id for the given component
@@ -409,6 +410,33 @@ def get_suite_version(source, suite):
 
     return version
 
+def get_suite_architectures(suite):
+    """
+    Returns list of architectures for C{suite}.
+
+    @type suite: string, int
+    @param suite: the suite name or the suite_id
+
+    @rtype: list
+    @return: the list of architectures for I{suite}
+    """
+
+    suite_id = None
+    if type(suite) == str:
+        suite_id = get_suite_id(suite)
+    elif type(suite) == int:
+        suite_id = suite
+    else:
+        return None
+
+    sql = """ SELECT a.arch_string FROM suite_architectures sa
+              JOIN architecture a ON (a.id = sa.architecture)
+              WHERE suite='%s' """ % (suite_id)
+
+    q = projectB.query(sql)
+    return map(lambda x: x[0], q.getresult())
+
+
 ################################################################################
 
 def get_or_set_maintainer_id (maintainer):