Spaces:
Runtime error
Runtime error
File size: 1,331 Bytes
e198e1c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
# Copyright (c) OpenMMLab. All rights reserved.
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)
|