hooks.add_entity: Add a corp_entity from a corporation variable.

This commit is contained in:
Brett Smith 2017-12-27 11:51:14 -05:00
parent 82e4bf5bd8
commit f2c869d783

View file

@ -68,3 +68,5 @@ class AddEntityHook:
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):
data['entity'] = self._name2entity(data['payee'], 1) data['entity'] = self._name2entity(data['payee'], 1)
if ('corporation' in data) and ('corp_entity' not in data):
data['corp_entity'] = self._name2entity(data['corporation'], 0)