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 b187f0305997245bbe4b0bfb5262f8024c333d08..e17a191e39472b112e50f24a00442da161c65970 100644 (file)
--- a/daklib/checks.py
+++ b/daklib/checks.py
@@ -799,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