# coding: utf-8
"""Utilities for working with strings and text."""
from __future__ import annotations
[docs]
def tonumber(s: str) -> float:
"""
Convert string to number, raise ValueError if s cannot be converted.
"""
# Duck test.
try:
stnum = s.upper().replace("D", "E") # D-01 is not recognized by python: Replace it with E.
# stnum = strip_punct(stnum) # Remove punctuation chars.
return float(stnum) # Try to convert.
except ValueError:
raise
except Exception:
raise RuntimeError("Don't know how to handle type %s: %s" % (type(s), str(s)))
[docs]
def nums_and_text(line: str) -> tuple:
"""
Split line into (numbers, text).
"""
tokens = line.split()
text = ""
numbers = []
for tok in tokens:
try:
numbers.append(tonumber(tok))
except ValueError:
text += " " + tok
return numbers, text
[docs]
def rreplace(s, old, new, occurrence):
"""
replace old with new in string but, instead of starting from the beginning
as replace does, starting from the end.
>>> s = '1232425'
>>> assert rreplace(s, '2', ' ', 2) == '123 4 5'
>>> assert rreplace(s, '2', ' ', 3) == '1 3 4 5'
>>> assert rreplace(s, '2', ' ', 4) == '1 3 4 5'
>>> assert rreplace(s, '2', ' ', 0) == '1232425'
"""
# Based on:
# https://stackoverflow.com/questions/2556108/rreplace-how-to-replace-the-last-occurrence-of-an-expression-in-a-string
li = s.rsplit(old, occurrence)
return new.join(li)
[docs]
def rm_multiple_spaces(string: str) -> str:
"""remove multiple spaces in a string."""
import re
return re.sub(' +', ' ', string)