From 02e6f293488332f269a1b4fec31041eb43358260 Mon Sep 17 00:00:00 2001 From: "Jip J. Dekker" Date: Wed, 23 Apr 2014 15:27:14 +0200 Subject: [PATCH] Include and Exclude fix after testing --- sourceloader.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/sourceloader.py b/sourceloader.py index 68c1832..bb0d81a 100644 --- a/sourceloader.py +++ b/sourceloader.py @@ -21,15 +21,16 @@ class SourceLoader: known_parser.add(cls) def include(self, source_names): - new = [] + print source_names + new = set() for name in source_names: - new.extend([src for src in self.sources if re.match(name, src.__class__.__name__)]) - self.sources = new + new.update([src for src in self.sources if re.match(name, src.__class__.__name__)]) + self.sources = list(new) def exclude(self, source_names): exclude = [] for name in source_names: - exclude.extend([src for src in self.sources if re.match(source_names,src.__class__.__name__)]) + exclude.extend([src for src in self.sources if re.match(name, src.__class__.__name__)]) self.sources = [src for src in self.sources if src not in exclude] def __str__(self): @@ -37,5 +38,5 @@ class SourceLoader: for src in self.sources: string += "Source: " + src.__class__.__name__ string += " - " - string += "URI: " + src.website + "\n" + string += "URI: " + src.website + "\n" return string \ No newline at end of file