NPO-Accounting/import2ledger
6
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

hooks.add_entity: Don't try to rearrange non-Latin names.

Browse source
This commit is contained in:
Brett Smith 2017-12-27 10:45:18 -05:00
parent ba81dc9847
commit e758db31ad
1 changed files with 12 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
import2ledger/hooks/add_entity.py
View file

@ -41,20 +41,24 @@ class AddEntityHook:
if word: if word:
yield word yield word
def _str2entity(self, s, trim_re): def _str2entity(self, s, trim_re, name_shifts):
parts = list(self._entity_parts(s, trim_re)) parts = list(self._entity_parts(s, trim_re))
if not parts: if not parts:
return '' return ''
parts.insert(0, parts.pop()) for _ in range(name_shifts):
parts.insert(0, parts.pop())
if parts[-1].lower() in self.NAME_PREFIXES: if parts[-1].lower() in self.NAME_PREFIXES:
parts.insert(0, parts.pop()) parts.insert(0, parts.pop())
return '-'.join(parts) return '-'.join(parts)
def _name2entity(self, name, name_shifts):
name = self._remove_parens(name)
name = self._destroke(name)
entity = self._str2entity(name, self.NONASCII_RE, name_shifts)
if not entity:
entity = self._str2entity(name, self.NONALNUM_RE, 0)
return entity
def run(self, data): def run(self, data):
if ('payee' in data) and ('entity' not in data): if ('payee' in data) and ('entity' not in data):
payee = self._remove_parens(data['payee']) data['entity'] = self._name2entity(data['payee'], 1)
payee = self._destroke(payee)
entity = self._str2entity(payee, self.NONASCII_RE)
if not entity:
entity = self._str2entity(payee, self.NONALNUM_RE)
data['entity'] = entity