|
|
|
|
|
class StringStrip: |
|
|
"""Removing the leading and/or the trailing characters based on the string |
|
|
argument passed. |
|
|
|
|
|
Args: |
|
|
strip (bool): Whether remove characters from both left and right of |
|
|
the string. Default: True. |
|
|
strip_pos (str): Which position for removing, can be one of |
|
|
('both', 'left', 'right'), Default: 'both'. |
|
|
strip_str (str|None): A string specifying the set of characters |
|
|
to be removed from the left and right part of the string. |
|
|
If None, all leading and trailing whitespaces |
|
|
are removed from the string. Default: None. |
|
|
""" |
|
|
|
|
|
def __init__(self, strip=True, strip_pos='both', strip_str=None): |
|
|
assert isinstance(strip, bool) |
|
|
assert strip_pos in ('both', 'left', 'right') |
|
|
assert strip_str is None or isinstance(strip_str, str) |
|
|
|
|
|
self.strip = strip |
|
|
self.strip_pos = strip_pos |
|
|
self.strip_str = strip_str |
|
|
|
|
|
def __call__(self, in_str): |
|
|
|
|
|
if not self.strip: |
|
|
return in_str |
|
|
|
|
|
if self.strip_pos == 'left': |
|
|
return in_str.lstrip(self.strip_str) |
|
|
elif self.strip_pos == 'right': |
|
|
return in_str.rstrip(self.strip_str) |
|
|
else: |
|
|
return in_str.strip(self.strip_str) |
|
|
|
