17 lines
570 B
Python
17 lines
570 B
Python
import sys
|
|
from typing import TypeVar
|
|
|
|
T = TypeVar("T", str, bytes, bytearray)
|
|
|
|
if sys.version_info.minor > 8: # PY39
|
|
def remove_suffix(string: T, suffix: T) -> T:
|
|
return string.removesuffix(suffix)
|
|
|
|
def remove_prefix(string: T, prefix: T) -> T:
|
|
return string.removeprefix(prefix)
|
|
else:
|
|
def remove_suffix(string: T, suffix: T) -> T:
|
|
return string[:-len(suffix)] if string.endswith(suffix) else string
|
|
|
|
def remove_prefix(string: T, prefix: T) -> T:
|
|
return string[len(prefix):] if string.startswith(prefix) else string
|