class TestCalculator(unittest.TestCase):
def setUp(self):
csv_importer = Importer(FIXTURES)
transactions_df = csv_importer.import_csv()
self.calculator = Calculator(
transactions_df
)
self.calculator.calculate()
def test_init(self):
assert isinstance(self.calculator, Calculator)
def test_get_gains(self):
gains_ref = ('{"index":{"0":{"gain":{"2019":82.8801608579,"2020":49.4627604496,"2018":0},"queue":[]},"1":{"gain":{"2020":248.97,"2018":0,"2019":0},"queue":[]},"2":{"gain":{"2019":41.6666666667,"2020":-294.0870149755,"2018":0},"queue":[]},"3":"Grand Totals"},"2018":{"0":0.0,"1":0.0,"2":0.0,"3":0.0},"2019":{"0":82.8801608579,"1":0.0,"2":41.6666666667,"3":124.5468275246},"2020":{"0":49.4627604496,"1":248.97,"2":-294.0870149755,"3":4.3457454741},"Total":{"0":132.3429213075,"1":248.97,"2":-252.4203483088,"3":128.8925729987}}')
assert self.calculator.get_gains().to_json() == gains_ref
class Main(object):
def __init__(self):
self.builder = Gtk.Builder()
self.builder.add_from_file("main.glade")
self.builder.connect_signals(self)
self.main_box = self.builder.get_object("main_box")
self.window = self.builder.get_object("main_window")
self.add_dialog = self.builder.get_object("add_dialog")
self.gains_dialog = self.builder.get_object("gains_dialog")
self.cgt_dialog = self.builder.get_object("cgt_dialog")
self.cgt_results = self.builder.get_object("cgt_label")
self.main_menu = self.builder.get_object("main_menu")
self.menu_button = self.builder.get_object("menu_button")
self.header = self.builder.get_object("header")
self.gains_view = self.builder.get_object("gains_view")
self.portfolio_view = self.builder.get_object("portfolio_view")
self.transactions_view = self.builder.get_object("transactions_view")
self.transactions_store = self.builder.get_object("transactions_store")
self.portfolio_store = self.builder.get_object("portfolio_store")
self._transactions_df = pd.DataFrame()
self.calculator = Calculator(self.transactions_df)
if os.path.isfile(STORE_FILE):
self.load_csv()
self.main_menu.show_all()
self.window.show_all()
@property
def transactions_df(self):
if self._transactions_df.empty:
if os.path.exists(STORE_FILE):
self._transactions_df = pd.read_csv(
STORE_FILE,
parse_dates=[0],
infer_datetime_format=True
)
return self._transactions_df
@transactions_df.setter
def transactions_df(self, dataframe):
self._transactions_df = dataframe
dataframe.to_csv(STORE_FILE, index=False)
def on_destroy(self, *args):
Gtk.main_quit()
def on_calculate_clicked(self, button):
self.calculator.calculate()
self.gains = gains = self.calculator.get_gains()
store = Gtk.ListStore(str, *([int] * (len(gains.columns) - 1)))
for _, row in gains.iterrows():
row["index"] = str(row["index"]) # convert Queue to str
store.append(list(row))
self.gains_view.set_model(store)
renderer = Gtk.CellRendererText()
# populate columns if not already
if not self.gains_view.get_columns():
self.gains_view.append_column(
Gtk.TreeViewColumn("Product", renderer, text=0),
)
for index, year in enumerate(gains.columns[1:-1]):
self.gains_view.append_column(
Gtk.TreeViewColumn(year, renderer, text=index+1),
)
self.gains_view.append_column(
Gtk.TreeViewColumn("Total", renderer, text=len(gains.columns)-1),
)
self.gains_dialog.run()
def on_close_clicked(self, button, *args):
self.gains_dialog.hide()
def on_cgt_close_clicked(self, button, *args):
self.cgt_dialog.hide()
def on_add_clicked(self, button):
#self.progress_bar.set_fraction(self.progress_bar.get_fraction() + 0.01)
self.add_dialog.run()
def on_remove_clicked(self, button):
# TODO
# get the selection
# button only available for transactions
# confirmation dialog
# delete
pass
def on_calculate_cgt_clicked(self, button, *args):
# calculates actual CGT
cgt = self.calculator.get_cgt(self.gains)
result = "\n".join(["{}: €{}".format(k, v) for k, v in cgt.items()])
self.cgt_results.set_text(result)
self.cgt_dialog.run()
def on_cancel_clicked(self, button, *args):
self.add_dialog.hide()
def on_add_transaction_clicked(self, button):
product_name = self.builder.get_object("add_product_name").get_text()
isin = self.builder.get_object("add_isin").get_text()
price = float(self.builder.get_object("add_price").get_text())
number_of_items = int(self.builder.get_object("add_number_of_items").get_text())
exchange_rate = float(self.builder.get_object("add_exchange_rate").get_text())
fees = (float(self.builder.get_object("add_fees").get_text())) * -1
stock_exchange = self.builder.get_object("add_stock_exchange").get_text()
date_time = pd.to_datetime(
self.builder.get_object("add_date_time").get_text(),
infer_datetime_format=True
)
self.transactions_df = self.transactions_df.append(
{
"Date_Time": date_time,
"Product": product_name,
"ISIN": isin,
"Price": price,
"Quantity": number_of_items,
"Reference exchange": stock_exchange,
"Exchange rate": exchange_rate,
"Local value": (price * number_of_items) * exchange_rate * -1,
"Value": price * number_of_items * -1,
"Transaction": fees,
"Total": price * number_of_items * -1,
}, ignore_index=True)
self.load_csv()
self.add_dialog.hide()
def on_import_clicked(self, button):
dialog = Gtk.FileChooserDialog(
title="Please choose a file",
parent=self.window,
action=Gtk.FileChooserAction.OPEN,
)
dialog.add_buttons(
Gtk.STOCK_CANCEL,
Gtk.ResponseType.CANCEL,
Gtk.STOCK_OPEN,
Gtk.ResponseType.OK
)
filter_text = Gtk.FileFilter()
filter_text.set_name("CSV files")
filter_text.add_mime_type("text/csv")
dialog.add_filter(filter_text)
response = dialog.run()
if response == Gtk.ResponseType.OK:
csv_file = dialog.get_filename()
self.load_csv(csv_file)
dialog.hide()
def load_csv(self, filename=None):
if filename:
csv_importer = Importer(filename)
self.transactions_df = csv_importer.import_csv()
portfolio = Portfolio(self.transactions_df)
portfolio_data = portfolio.plain_view()
self.populate(
self.transactions_view,
self.transactions_df,
self.transactions_store,
[
"Date_Time", # move to settings
"Product",
"Reference exchange",
"Quantity",
"Price",
"Local value",
"Value",
"Exchange rate",
"Transaction",
"Total",
],
)
self.populate(
self.portfolio_view,
portfolio_data,
self.portfolio_store,
)
def populate(self, view, data, store, cols=None):
"""Populates given view with the data"""
# TODO convert stores to properties to properly update
if cols:
data = data[cols]
store.clear()
for _, row in data.iterrows():
row[0] = str(row[0]) # convert datetime to str
store.append(list(row))
view.set_model(store)
if not view.get_columns():
renderer = Gtk.CellRendererText()
for index, column in enumerate(data.columns):
view.append_column(
Gtk.TreeViewColumn(column, renderer, text=index),
)