Add option to forbid source and/or binary uploads to a suite

[dak.git] / daklib / checks.py

diff --git a/daklib/checks.py b/daklib/checks.py
index 2eec769def845f125f40882163541d918efbb318..e17a191e39472b112e50f24a00442da161c65970 100644 (file)
--- a/daklib/checks.py
+++ b/daklib/checks.py
@@ -719,6 +719,7 @@ class NoSourceOnlyCheck(Check):
            and 'all' not in changes.architectures \
            and 'experimental' not in changes.distributions \
            and 'unstable' not in changes.distributions \
+           and 'sid' not in changes.distributions \
            and changes.source.package_list.has_arch_indep_packages():
             raise Reject('Uploads not including architecture-independent packages are not allowed.')
 
@@ -798,6 +799,14 @@ class SourceFormatCheck(Check):
         if query.first() is None:
             raise Reject('source format {0} is not allowed in suite {1}'.format(source_format, suite.suite_name))
 
+class SuiteCheck(Check):
+    def per_suite_check(self, upload, suite):
+        if not suite.accept_source_uploads and upload.changes.source is not None:
+            raise Reject('The suite "{0}" does not accept source uploads.'.format(suite.suite_name))
+        if not suite.accept_binary_uploads and len(upload.changes.binaries) != 0:
+            raise Reject('The suite "{0}" does not accept binary uploads.'.format(suite.suite_name))
+        return True
+
 class SuiteArchitectureCheck(Check):
     def per_suite_check(self, upload, suite):
         session = upload.session