diff --git a/filter_plugins/custom.py b/filter_plugins/custom.py index 82b0efa..ad22133 100644 --- a/filter_plugins/custom.py +++ b/filter_plugins/custom.py @@ -2,6 +2,7 @@ __author__ = 'dale mcdiarmid' import re import os.path +from six import string_types def modify_list(values=[], pattern='', replacement='', ignorecase=False): ''' Perform a `re.sub` on every item in the list''' @@ -13,7 +14,7 @@ def modify_list(values=[], pattern='', replacement='', ignorecase=False): return [_re.sub(replacement, value) for value in values] def append_to_list(values=[], suffix=''): - if isinstance(values, basestring): + if isinstance(values, string_types): values = values.split(',') return [str(value+suffix) for value in values] @@ -46,5 +47,4 @@ class FilterModule(object): 'array_to_str':array_to_str, 'extract_role_users':extract_role_users, 'filter_reserved':filter_reserved, - 'filename':filename} - + 'filename':filename} \ No newline at end of file