[dak/master] Refactor add_dsc_to_db().

[Torsten Werner <twerner@debian.org>]

Signed-off-by: Torsten Werner <twerner@debian.org>
---
 daklib/dbconn.py |    9 +++------
 1 files changed, 3 insertions(+), 6 deletions(-)

diff --git a/daklib/dbconn.py b/daklib/dbconn.py
index 6f598bb..611b30a 100755
--- a/daklib/dbconn.py
+++ b/daklib/dbconn.py
@@ -2268,12 +2268,9 @@ def add_dsc_to_db(u, filename, session=None):
     session.add(source)
     session.flush()
 
-    for suite_name in u.pkg.changes["distribution"].keys():
-        sa = SrcAssociation()
-        sa.source_id = source.source_id
-        sa.suite_id = get_suite(suite_name).suite_id
-        session.add(sa)
-
+    suite_names = u.pkg.changes["distribution"].keys()
+    source.suites = session.query(Suite). \
+        filter(Suite.suite_name.in_(suite_names)).all()
     session.flush()
 
     # Add the source files to the DB (files and dsc_files)
-- 
1.5.6.5