[dak/master] dak/admin.py: new suites need an archive

[Ansgar Burchardt <ansgar@debian.org>]

---
 dak/admin.py |    5 +++++
 1 files changed, 5 insertions(+), 0 deletions(-)

diff --git a/dak/admin.py b/dak/admin.py
index 23b4c2d..378a878 100755
--- 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()
-- 
1.7.2.5