def add_annotation(self, match: MatchFound):
mlen = len(match.subset)
name = match.subset[self.court_name_column].values[0] \
if match.is_exact else \
match.court_name if match.court_name is not None else \
match.subset[self.court_name_column].values[0] if mlen > 0 else ''
court_type = match.subset[self.court_type_column].values[0] \
if match.is_exact else \
match.court_type if match.court_type is not None else \
match.subset[self.court_type_column].values[0] if mlen > 0 else ''
jurisdiction = match.subset[self.jurisdiction_column].values[0] \
if match.is_exact else \
match.jurisdiction if match.jurisdiction is not None else \
match.subset[self.jurisdiction_column].values[0] if mlen > 0 else ''
ant = CourtAnnotation(name=name,
coords=(match.entry_start, match.entry_end),
locale=self.locale,
text=match.text)
ant.jurisdiction = jurisdiction
ant.court_type = court_type
self.annotations.append(ant)
def test_repr(self):
ant = CourtAnnotation(name='GrossCourt',
coords=(1, 10),
text='Court Gross',
locale='EN')
s = ant.__repr__()
self.assertGreater(len(s), 0)
def get_court_annotations(
locale: str,
text: str,
court_config_list: List[DictionaryEntry],
priority: bool = False,
text_locales: List[str] = (),
simplified_normalization: bool = False
) -> Generator[CourtAnnotation, None, None]:
locale_obj = Locale(locale)
dic_entries = find_dict_entities(
text,
court_config_list,
default_language=locale_obj.language,
conflict_resolving_func=conflicts_take_first_by_id
if priority else None,
text_languages=[Locale(item).language for item in text_locales],
simplified_normalization=simplified_normalization)
for ent in dic_entries:
ant = CourtAnnotation(coords=ent.coords)
if ent.entity[0]:
toponim = ent.entity[0] # type: DictionaryEntry
ant.entity_id = toponim.id
ant.entity_category = toponim.category
ant.entity_priority = toponim.priority
ant.name_en = toponim.entity_name
ant.name = toponim.name
if toponim.extra_columns:
for extr_col in toponim.extra_columns:
setattr(ant, extr_col, toponim.extra_columns[extr_col])
if ent.entity[1]: # alias
ant.alias = ent.entity[1].alias
ant.locale = ent.entity[1].language
if not ant.locale:
ant.locale = locale_obj.language
yield ant