+Logger = None;
+
+#######################################################################################
+
+def get_id (package, version, architecture):
+ if architecture == "source":
+ q = projectB.query("SELECT id FROM source WHERE source = '%s' AND version = '%s'" % (package, version))
+ else:
+ q = projectB.query("SELECT b.id FROM binaries b, architecture a WHERE b.package = '%s' AND b.version = '%s' AND (a.arch_string = '%s' OR a.arch_string = 'all') AND b.architecture = a.id" % (package, version, architecture))
+
+ ql = q.getresult();
+ if ql == []:
+ utils.warn("Couldn't find '%s~%s~%s'." % (package, version, architecture));
+ return None;
+ if len(ql) > 1:
+ utils.warn("Found more than one match for '%s~%s~%s'." % (package, version, architecture));
+ return None;
+ id = ql[0][0];
+ return id;
+
+#######################################################################################
+
+def set_suite (file, suite_id):
+ lines = file.readlines();
+
+ projectB.query("BEGIN WORK");
+
+ # Build up a dictionary of what is currently in the suite
+ current = {};
+ q = projectB.query("SELECT b.package, b.version, a.arch_string, ba.id FROM binaries b, bin_associations ba, architecture a WHERE ba.suite = %s AND ba.bin = b.id AND b.architecture = a.id" % (suite_id));
+ ql = q.getresult();
+ for i in ql:
+ key = string.join(i[:3]);
+ current[key] = i[3];
+ q = projectB.query("SELECT s.source, s.version, sa.id FROM source s, src_associations sa WHERE sa.suite = %s AND sa.source = s.id" % (suite_id));
+ ql = q.getresult();
+ for i in ql:
+ key = string.join(i[:2]) + " source";
+ current[key] = i[2];
+
+ # Build up a dictionary of what should be in the suite
+ desired = {};
+ for line in lines:
+ split_line = string.split(string.strip(line[:-1]));
+ if len(split_line) != 3:
+ utils.warn("'%s' does not break into 'package version architecture'." % (line[:-1]));
+ continue;
+ key = string.join(split_line);
+ desired[key] = "";
+
+ # Check to see which packages need removed and remove them
+ for key in current.keys():
+ if not desired.has_key(key):
+ (package, version, architecture) = string.split(key);
+ id = current[key];
+ if architecture == "source":
+ q = projectB.query("DELETE FROM src_associations WHERE id = %s" % (id));
+ else:
+ q = projectB.query("DELETE FROM bin_associations WHERE id = %s" % (id));
+ Logger.log(["removed",key,id]);
+
+ # Check to see which packages need added and add them
+ for key in desired.keys():
+ if not current.has_key(key):
+ (package, version, architecture) = string.split(key);
+ id = get_id (package, version, architecture);
+ if not id:
+ continue;
+ if architecture == "source":
+ q = projectB.query("INSERT INTO src_associations (suite, source) VALUES (%s, %s)" % (suite_id, id));
+ else:
+ q = projectB.query("INSERT INTO bin_associations (suite, bin) VALUES (%s, %s)" % (suite_id, id));
+ Logger.log(["added",key,id]);
+
+ projectB.query("COMMIT WORK");