def extra_info(self, df, fmt_tag):
df['label'] = df.index.map(self.label_map)
df['link'] = df.index.map(self.link_map)
df['fmt_tag'] = fmt_tag
# save and order index
ordered_index = utils.order_paths(df.index)
return df, ordered_index
def calcs(self):
table_data = []
for table_path in ['assets', 'liabilities', 'equity']:
header = {'fmt_tag': 'header', 'label': self.label_map(table_path)}
table_data.append(header)
table_paths = utils.order_paths(accountifie.gl.models.get_child_paths(table_path))
path_totals = pd.Series(index=self.column_order)
path_totals = 0
for path in table_paths:
if table_path == 'equity':
bals = self.query_manager.pd_path_balances(self.company_id, self.columns, [path])
else:
bals = self.query_manager.path_drilldown(self.company_id, self.columns, path)
if bals is not None:
if table_path == 'assets':
for col in bals.columns:
bals = bals * (-1.0)
bals['fmt_tag'] = 'item'
bals['label'] = bals.index.map(self.label_map)
bals['link'] = bals.index.map(self.link_map)
totals = bals[self.column_order].sum(axis=0)
path_totals = path_totals + totals
totals['fmt_tag'] = 'minor_total'
totals['label'] = 'Total %s' % self.label_map(path)
totals['link'] = ''
bals.loc['Total %s' % self.label_map(path)] = totals
bals['index'] = bals.index
table_data += bals.to_dict(orient='records')
path_totals['fmt_tag'] = 'major_total'
path_totals['label'] = 'Total %s' % self.label_map(table_path)
path_totals['link'] = ''
path_totals['index'] = table_path
table_data.append(path_totals.to_dict())
for row in table_data:
if row['fmt_tag'] != 'header':
for col in self.column_order:
row[col] = {'text': row[col], 'link': row['link']}
return table_data
