Include and Exclude fix after testing
This commit is contained in:
parent
b44ab49255
commit
02e6f29348
@ -21,15 +21,16 @@ class SourceLoader:
|
|||||||
known_parser.add(cls)
|
known_parser.add(cls)
|
||||||
|
|
||||||
def include(self, source_names):
|
def include(self, source_names):
|
||||||
new = []
|
print source_names
|
||||||
|
new = set()
|
||||||
for name in source_names:
|
for name in source_names:
|
||||||
new.extend([src for src in self.sources if re.match(name, src.__class__.__name__)])
|
new.update([src for src in self.sources if re.match(name, src.__class__.__name__)])
|
||||||
self.sources = new
|
self.sources = list(new)
|
||||||
|
|
||||||
def exclude(self, source_names):
|
def exclude(self, source_names):
|
||||||
exclude = []
|
exclude = []
|
||||||
for name in source_names:
|
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]
|
self.sources = [src for src in self.sources if src not in exclude]
|
||||||
|
|
||||||
def __str__(self):
|
def __str__(self):
|
||||||
|
Reference in New Issue
Block a user