def batch_deactivate_people(name_list, **kwargs):
update_queue = []
for name in name_list:
node = utils.load_people_node(name, is_id=False)
old_status = node.active
node.active = False
if node.parent is not None:
pnode = utils.load_people_node(node.parent)
pnode.active_children = [
cid for cid in pnode.active_children if cid != node.id
]
pnode.dump()
node.dump()
update_queue.append(node.id)
print('id: {}, name: {} was {}, now deactive.'.format(
node.id, node.name, 'active' if old_status else 'deactive'))
# update metadata
print('Updating metadata ...')
for nid in update_queue:
node = utils.load_people_node(nid)
node.update_upward()
node.dump()
def batch_new_people(filename, **kwargs):
"""
file format: csv
name parent incash outcash
"""
df = pd.read_csv(filename,
dtype={
'name': str,
'parent': str,
'incash': float,
'outcash': float
})
df['name'] = df['name'].fillna('')
df['parent'] = df['parent'].fillna('')
df.loc[df['parent'] == '', 'parent'] = None
df['incash'] = df['incash'].fillna(0.)
df['outcash'] = df['outcash'].fillna(0.)
update_queue = []
for idx, row in df.iterrows():
row_dict = row.to_dict()
if utils.ID_MAP.get(row_dict['name'], None) is not None:
raise ValueError(
'Duplicated username found in id: {}, name: {}'.format(
utils.ID_MAP[row_dict['name']], row_dict['name']))
node = People(**row_dict)
pname = row_dict.get('parent', None)
if pname is None:
utils.add_root(node.id)
else:
node.parent = utils.ID_MAP[pname]
pnode = utils.load_people_node(pname, is_id=False)
pnode.children.append(node.id)
pnode.active_children.append(node.id)
pnode.dump()
node.dump()
utils.ID_MAP[node.name] = node.id
update_queue.append(node.id)
print('id: {}, {} added.'.format(
'WD00000'[:-len(str(node.id))] + str(node.id), row_dict))
# update metadata
print('Updating metadata ...')
for nid in update_queue:
node = utils.load_people_node(nid)
node.update_upward()
node.dump()
export_all(people=True)
def recurse_payback(node, pbamount, _expense=True):
if not _expense:
node._outcash_cache += (pb['l1'] + pb['l2']) * pbamount
node.dump()
if node.parent is not None:
l1_node = utils.load_people_node(node.parent)
l1_node._incash_cache += pb['l1'] * pbamount
if l1_node.parent is not None:
l2_node = utils.load_people_node(l1_node.parent)
if len(l2_node.active_children
) >= utils.L2_THRESHOLD_NUM_PEOPLE:
l2_node._incash_cache += pb['l2'] * pbamount
l2_node.dump()
l1_node.dump()
recurse_payback(l1_node, l1_node._incash_cache, _expense=False)
def recurse_cleanup(node):
node.expense_cache += node._expense_cache
node.incash_cache += node._incash_cache
node.outcash_cache += node._outcash_cache
node._expense_cache = 0
node._incash_cache = 0
node._outcash_cache = 0
node.dump()
if node.parent is not None:
pnode = utils.load_people_node(node.parent)
recurse_cleanup(pnode)
def apply():
with open(utils.ID_NUM_FILE, 'r') as f:
last = int(f.read())
for i in range(0, last):
node = utils.load_people_node(i)
node.expense += node.expense_cache
node.incash += node.incash_cache
node.outcash += node.outcash_cache
node.expense_cache = 0
node.incash_cache = 0
node.outcash_cache = 0
node.dump()
export_all()
def reset_all(cache=False):
with open(utils.ID_NUM_FILE, 'r') as f:
i = int(f.read())
for nid in range(i):
node = utils.load_people_node(nid)
if not cache:
node.expense = 0.
node.incash = 0.
node.outcash = 0.
node.expense_cache = 0.
node.incash_cache = 0.
node.outcash_cache = 0.
node._expense_cache = 0.
node._incash_cache = 0.
node._outcash_cache = 0.
node.dump()
def _payment(payer, amount, ptype):
payer_node = utils.load_people_node(payer, is_id=False)
pb = utils.PERCENTAGE[ptype]
payer_node._expense_cache += amount
def recurse_payback(node, pbamount, _expense=True):
if not _expense:
node._outcash_cache += (pb['l1'] + pb['l2']) * pbamount
node.dump()
if node.parent is not None:
l1_node = utils.load_people_node(node.parent)
l1_node._incash_cache += pb['l1'] * pbamount
if l1_node.parent is not None:
l2_node = utils.load_people_node(l1_node.parent)
if len(l2_node.active_children
) >= utils.L2_THRESHOLD_NUM_PEOPLE:
l2_node._incash_cache += pb['l2'] * pbamount
l2_node.dump()
l1_node.dump()
recurse_payback(l1_node, l1_node._incash_cache, _expense=False)
recurse_payback(payer_node, amount)
def recurse_cleanup(node):
node.expense_cache += node._expense_cache
node.incash_cache += node._incash_cache
node.outcash_cache += node._outcash_cache
node._expense_cache = 0
node._incash_cache = 0
node._outcash_cache = 0
node.dump()
if node.parent is not None:
pnode = utils.load_people_node(node.parent)
recurse_cleanup(pnode)
recurse_cleanup(payer_node)
def update_upward(self):
self.num_children = self._get_num_children()
self.dump()
if self.parent is not None:
pnode = utils.load_people_node(self.parent)
pnode.update_upward()
def _get_num_children(self):
result = len(self.active_children)
for cid in self.children:
cnode = utils.load_people_node(cid)
result += cnode._get_num_children()
return result
def export_all(people=False, cache=False):
with open(utils.ID_NUM_FILE, 'r') as f:
i = int(f.read())
df_dict = {
'id': [],
'name': [],
'parent1_id': [],
'parent1_name': [],
'parent2_id': [],
'parent2_name': [],
'netcash': [],
'expense': [],
'incash': [],
'outcash': []
}
for nid in range(i):
node = utils.load_people_node(nid)
nid_str = 'WD00000'[:-len(str(nid))] + str(nid)
if node.active:
df_dict['id'].append(nid_str)
df_dict['name'].append(node.name)
if node.parent is not None:
pnode = utils.load_people_node(node.parent)
df_dict['parent1_id'].append('WD00000'[:-len(str(pnode.id))] +
str(pnode.id))
df_dict['parent1_name'].append(pnode.name)
if pnode.parent is not None:
ppnode = utils.load_people_node(pnode.parent)
df_dict['parent2_id'].append(
'WD00000'[:-len(str(ppnode.id))] + str(ppnode.id))
df_dict['parent2_name'].append(ppnode.name)
else:
df_dict['parent2_id'].append('')
df_dict['parent2_name'].append('')
else:
df_dict['parent1_id'].append('')
df_dict['parent1_name'].append('')
df_dict['parent2_id'].append('')
df_dict['parent2_name'].append('')
if cache:
df_dict['expense'].append(node.expense_cache)
df_dict['incash'].append(node.incash_cache)
df_dict['outcash'].append(node.outcash_cache)
df_dict['netcash'].append(node.incash_cache -
node.outcash_cache)
else:
df_dict['expense'].append(node.expense_cache)
df_dict['incash'].append(node.incash)
df_dict['outcash'].append(node.outcash)
df_dict['netcash'].append(node.incash - node.outcash)
df = pd.DataFrame(df_dict)
if people:
filename = 'output_id.csv'
with open(filename, 'w') as f:
df[[
'id', 'name', 'parent1_id', 'parent1_name', 'parent2_id',
'parent2_name'
]].to_csv(f, index=False, line_terminator='\n')
else:
filename = 'output_cache.csv' if cache else 'output.csv'
with open(filename, 'w') as f:
df[['id', 'name', 'netcash', 'expense', 'incash',
'outcash']].to_csv(f, index=False, line_terminator='\n')
