add ordinal case
This commit is contained in:
parent
3970e82434
commit
00fe0a2d9d
1 changed files with 38 additions and 14 deletions
|
|
@ -2768,6 +2768,9 @@ _STR2INT_MAP = {
|
|||
"seventy": 70, "eighty": 80, "ninety": 90,
|
||||
"hundred": 100, "thousand": 1000,
|
||||
}
|
||||
_STR2INT_ADJS = {
|
||||
"first": 1, "second": 2, "third": 3,
|
||||
}
|
||||
def str2int(number):
|
||||
"""
|
||||
Converts a string to an integer.
|
||||
|
|
@ -2779,9 +2782,14 @@ def str2int(number):
|
|||
int: The string represented as an integer.
|
||||
"""
|
||||
number = str(number)
|
||||
original_input = number
|
||||
try:
|
||||
# it's a digit already
|
||||
return int(number)
|
||||
except:
|
||||
# if it's an ordinal number such as "1st", it'll convert to int with the last two characters chopped off
|
||||
try:
|
||||
return int(number[:-2])
|
||||
except:
|
||||
pass
|
||||
|
||||
|
|
@ -2789,7 +2797,20 @@ def str2int(number):
|
|||
# it's a single number, return it
|
||||
return i
|
||||
|
||||
number = number.replace(" and "," ")
|
||||
# convert sound changes for generic ordinal numbers
|
||||
if number[-2:] == "th":
|
||||
# remove "th"
|
||||
number = number[:-2]
|
||||
if number[-1] == "f":
|
||||
# e.g. twelfth, fifth
|
||||
number = number[:-1] + "ve"
|
||||
elif number[-2:] == "ie":
|
||||
# e.g. twentieth, fortieth
|
||||
number = number[:-2] + "y"
|
||||
# custom case for ninth
|
||||
elif number[-3:] == "nin"
|
||||
number += "e"
|
||||
|
||||
# split number words by spaces, hyphens and commas, to accommodate multiple styles
|
||||
numbers = [ word.lower() for word in re.split(r'[-\s\,]',number) if word ]
|
||||
sums = []
|
||||
|
|
@ -2808,8 +2829,11 @@ def str2int(number):
|
|||
else:
|
||||
sums.append(i)
|
||||
else:
|
||||
elif i := STR2INT_ADJS.get(word):
|
||||
# it's a special adj word; ordinal case will never be a multiplier
|
||||
sums.append(i)
|
||||
else:
|
||||
# invalid number-word, return None to error
|
||||
return None
|
||||
# invalid number-word, raise ValueError
|
||||
raise ValueError(f"String {original_input} cannot be converted to int.")
|
||||
return sum(sums)
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue