X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=daklib%2Fpolicy.py;h=aeed9a2c19cbfb679274520e3418a73f42dc3722;hb=55513d80673e99884067b8820ca1a0b3e9aeeeef;hp=cdff56a7e0fe2a306b58b41fe3db7c922d4d36f4;hpb=a7f77393881783e65423f1f36778c67c52e5213d;p=dak.git

diff --git a/daklib/policy.py b/daklib/policy.py
index cdff56a7..aeed9a2c 100644
--- a/daklib/policy.py
+++ b/daklib/policy.py
@@ -279,13 +279,18 @@ class PolicyQueueUploadHandler(object):
                             ))
             components.add(component)
 
+        source = self.upload.source
         source_component = '(unknown)'
-        for component in ('main', 'contrib', 'non-free'):
+        for component, in self.session.query(Component.component_name).order_by(Component.ordering):
             if component in components:
                 source_component = component
                 break
+            else:
+                if source is not None:
+                    if self._source_override(component) is not None:
+                        source_component = component
+                        break
 
-        source = self.upload.source
         if source is not None:
             override = self._source_override(source_component)
             if override is None: