def get_money_annotations(text: str, float_digits=4) \
-> Generator[MoneyAnnotation, None, None]:
for match in CURRENCY_PTN_RE.finditer(text):
capture = match.capturesdict()
if not (capture['prefix']
or capture['postfix']) and not (capture['trigger_word']):
continue
prefix = capture['prefix']
postfix = capture['postfix']
amount = list(
get_amounts(capture['amount'][0], float_digits=float_digits))
if len(amount) != 1:
continue
if prefix:
prefix = prefix[0].lower()
currency_type = CURRENCY_SYMBOL_MAP.get(prefix)\
or CURRENCY_PREFIX_MAP.get(prefix)\
or prefix.upper()
elif postfix:
postfix = postfix[0].lower()
currency_type = CURRENCY_TOKEN_MAP.get(postfix) or (
capture['postfix'][0]).upper()
else:
currency_type = None
if not currency_type:
currency_type = DEFAULT_CURRENCY
text = capture['text'][0].strip(
string.punctuation.replace('$', '') + string.whitespace)
ant = MoneyAnnotation(coords=match.span(),
amount=amount[0],
text=text,
currency=currency_type)
yield ant
def get_money(text, return_sources=False, float_digits=4) -> Generator:
for match in CURRENCY_PTN_RE.finditer(text):
capture = match.capturesdict()
if not (capture['prefix'] or capture['postfix']):
continue
prefix = capture['prefix']
postfix = capture['postfix']
amount = list(
get_amounts(capture['amount'][0], float_digits=float_digits))
if len(amount) != 1:
continue
if prefix:
prefix = prefix[0].lower()
currency_type = CURRENCY_SYMBOL_MAP.get(prefix)\
or CURRENCY_PREFIX_MAP.get(prefix)\
or prefix.upper()
else:
postfix = postfix[0].lower()
currency_type = CURRENCY_TOKEN_MAP.get(postfix) or (
capture['postfix'][0]).upper()
item = (amount[0], currency_type)
if return_sources:
item += (capture['text'][0].strip(
string.punctuation.replace('$', '') + string.whitespace), )
yield item
