still WIP
							parent
							
								
									333fea4d18
								
							
						
					
					
						commit
						31fc1c369a
					
				|  | @ -245,8 +245,13 @@ class FilterSet(object): | |||
|         """ """ | ||||
|         return len(self._filters) | ||||
| 
 | ||||
|     def add(self, filters): | ||||
|     def add(self, filters=None): | ||||
|         """ """ | ||||
|         if not filters: | ||||
|             # so add() can be called blindly, useful for | ||||
|             # DataStore/Environment usage of filter operations | ||||
|             return | ||||
| 
 | ||||
|         if not isinstance(filters, FilterSet) and not isinstance(filters, list): | ||||
|             filters = [filters] | ||||
| 
 | ||||
|  | @ -256,8 +261,13 @@ class FilterSet(object): | |||
| 
 | ||||
|         return | ||||
| 
 | ||||
|     def remove(self, filters): | ||||
|     def remove(self, filters=None): | ||||
|         """ """ | ||||
|         if not filters: | ||||
|             # so remove() can be called blindly, useful for | ||||
|             # DataStore/Environemnt usage of filter ops | ||||
|             return | ||||
| 
 | ||||
|         if not isinstance(filters, FilterSet) and not isinstance(filters, list): | ||||
|             filters = [filters] | ||||
| 
 | ||||
|  |  | |||
|  | @ -158,7 +158,7 @@ class Environment(DataStoreMixin): | |||
|     set_default_object_marking_refs.__doc__ = ObjectFactory.set_default_object_marking_refs.__doc__ | ||||
| 
 | ||||
|     def add_filters(self, *args, **kwargs): | ||||
|         return self.source.filters.update(*args, **kwargs) | ||||
|         return self.source.filters.add(*args, **kwargs) | ||||
| 
 | ||||
|     def add_filter(self, *args, **kwargs): | ||||
|         return self.source.filters.add(*args, **kwargs) | ||||
|  |  | |||
|  | @ -461,7 +461,6 @@ def test_apply_common_filters(): | |||
|     # BUG: This test is brokem , weird behavior, the file obj | ||||
|     # in stix_objs is being parsed into real python-stix2 obj even though | ||||
|     # it never goes through parse() --> BAD <_< | ||||
|     print(stix_objs) | ||||
|     resp = list(apply_common_filters(stix_objs, [filters[14]])) | ||||
|     assert resp[0]["id"] == stix_objs[14]["id"] | ||||
|     assert len(resp) == 1 | ||||
|  |  | |||
		Loading…
	
		Reference in New Issue
	
	 =
						=