From 4abdee1dc1891ad212a48040d89ebc5b9dcc6088 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Adri=C3=A1n=20Oliva?= Date: Sun, 7 May 2023 17:56:57 -0600 Subject: Balance now has colors! Moved color functions to its own file. Now the colored output can be easily edited in general. --- utils/balance.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) (limited to 'utils/balance.py') diff --git a/utils/balance.py b/utils/balance.py index 64fa3e3..c13dee8 100644 --- a/utils/balance.py +++ b/utils/balance.py @@ -2,6 +2,7 @@ from typing import List from utils.read_file import entry from utils.register import currencies, complete_prices +from utils.colored_output import * class tree: @@ -53,18 +54,18 @@ class tree: result = '' if not ignore_level: for price in self.value: - result += f'{price:>20}\n' + result += price_f(f'{price:>20}') + '\n' result = result.rstrip() result += ' ' + ' '*(2 * level) if len(self.children) == 1: - result += self.name + ':' + result += account_f(self.name + ':') for child in self.children: result += child.__str__(level + 1, ignore_level = True) else: - result += self.name + '\n' + result += account_f(self.name) + '\n' for child in self.children: result += child.__str__(level + 1) @@ -96,7 +97,7 @@ class accounts_tree: result += '--------------------\n' for price in self.my_accounts.value: - result += f'{price:>20}\n' + result += price_f(f'{price:>20}') + '\n' return result.rstrip() -- cgit v1.2.3