dak/admin.py: new suites need an archive

diff --git a/dak/admin.py b/dak/admin.py
index 23b4c2d8e1a05ff17e1d7f8aecb51dab9e75c24c..378a878a7d35c89a9f252ae0abbc60316f2d5d0c 100755 (executable)
--- a/dak/admin.py
+++ b/dak/admin.py
@@ -225,6 +225,11 @@ def __suite_add(d, args, addallarches=False):
             signingkey = get_field('signingkey')
             if signingkey is not None:
                 suite.signingkeys = [signingkey.upper()]
+            archive_name = get_field('archive')
+            if archive_name is not None:
+                suite.archive = get_archive(archive_name, s)
+            else:
+                suite.archive = s.query(Archive).filter(~Archive.archive_name.in_(['build-queues', 'new', 'policy'])).one()
             suite.srcformats = s.query(SrcFormat).all()
             s.add(suite)
             s.flush()