def optimize():
in_txs = request.json
in_txs = [
tx for tx in in_txs if len(tx['creditor']) != 0
and len(tx['debtor']) != 0 and len(tx['amount']) != 0
]
ledger = build_ledger_from_transactions(in_txs)
_, optimized_ledger = optimize_debts_from_ledger(ledger)
out_txs = build_transactions_from_ledger(optimized_ledger)
return jsonify(transactions=out_txs)
def optimize():
in_txs = request.json
in_txs = [
tx for tx in in_txs if
len(tx['creditor']) != 0 and
len(tx['debtor']) != 0 and
len(tx['amount']) != 0
]
ledger = build_ledger_from_transactions(in_txs)
_, optimized_ledger = optimize_debts_from_ledger(ledger)
out_txs = build_transactions_from_ledger(optimized_ledger)
return jsonify(transactions=out_txs)
def optimize(
yml_file,
draw,
show_balances,
basic,
):
if not yml_file:
yml_file = stdin
ledger = yaml.load(yml_file)
optimized_graph, optimized_ledger = optimize_debts_from_ledger(ledger, basic=basic)
yaml.dump(optimized_ledger, stdout, default_flow_style=False)
if draw:
import matplotlib.pyplot as plt
pos = nx.spring_layout(optimized_graph)
nx.draw(
optimized_graph,
pos=pos,
node_color='white',
node_size=3000,
with_labels=True
)
edge_labels = {
(u, v): format_amount(data['amount'])
for u, v, data in optimized_graph.edges_iter(data=True)
}
nx.draw_networkx_edge_labels(
optimized_graph,
pos=pos,
edge_labels=edge_labels
)
if show_balances:
node_labels = {
node: format_amount(data['orig_balance'])
for node, data in optimized_graph.nodes_iter(data=True)
}
elevated_pos = {
key: (x, y + 5 / 100)
for key, (x, y) in pos.items()
}
nx.draw_networkx_labels(
optimized_graph,
pos=elevated_pos,
labels=node_labels,
font_weight='bold'
)
plt.show()
def optimize(
yml_file,
draw,
show_balances,
basic,
):
if not yml_file:
yml_file = stdin
ledger = yaml.load(yml_file)
optimized_graph, optimized_ledger = optimize_debts_from_ledger(ledger,
basic=basic)
yaml.dump(optimized_ledger, stdout, default_flow_style=False)
if draw:
import matplotlib.pyplot as plt
pos = nx.spring_layout(optimized_graph)
nx.draw(optimized_graph,
pos=pos,
node_color='white',
node_size=3000,
with_labels=True)
edge_labels = {(u, v): format_amount(data['amount'])
for u, v, data in optimized_graph.edges_iter(data=True)}
nx.draw_networkx_edge_labels(optimized_graph,
pos=pos,
edge_labels=edge_labels)
if show_balances:
node_labels = {
node: format_amount(data['orig_balance'])
for node, data in optimized_graph.nodes_iter(data=True)
}
elevated_pos = {
key: (x, y + 5 / 100)
for key, (x, y) in pos.items()
}
nx.draw_networkx_labels(optimized_graph,
pos=elevated_pos,
labels=node_labels,
font_weight='bold')
plt.show()
