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