def test_add_shouldAdd(self):
component_under_test = Depotwerte()
component_under_test.add(name='1name', isin='some isin')
assert len(component_under_test.content) == 1
assert component_under_test.content.Name[0] == '1name'
assert component_under_test.content.ISIN[0] == 'some isin'
def test_add_shouldAdd():
component_under_test = Depotwerte()
component_under_test.add(name='1name',
isin='some isin',
typ=component_under_test.TYP_ETF)
assert len(component_under_test.content) == 1
assert component_under_test.content.Name[0] == '1name'
assert component_under_test.content.ISIN[0] == 'some isin'
assert component_under_test.content.Typ[0] == component_under_test.TYP_ETF
def test_parse_and_migrate_with_existing_data_should_do_nothing():
depotwerte = Depotwerte()
raw_table = pd.DataFrame([["demoname", "demoisin", Depotwerte.TYP_FOND]],
columns=Depotwerte.TABLE_HEADER)
depotwerte.parse_and_migrate(raw_table)
assert check_equal(depotwerte.content.columns, Depotwerte.TABLE_HEADER)
assert len(depotwerte.content) == 1
assert depotwerte.content.Name[0] == 'demoname'
assert depotwerte.content.ISIN[0] == 'demoisin'
assert depotwerte.content.Typ[0] == Depotwerte.TYP_FOND
def test_parse_and_migrate_without_type_shoud_set_type_etf():
depotwerte = Depotwerte()
raw_table = pd.DataFrame([["demoname", "demoisin"]],
columns=['Name', 'ISIN'])
depotwerte.parse_and_migrate(raw_table)
assert check_equal(depotwerte.content.columns, Depotwerte.TABLE_HEADER)
assert len(depotwerte.content) == 1
assert depotwerte.content.Name[0] == 'demoname'
assert depotwerte.content.ISIN[0] == 'demoisin'
assert depotwerte.content.Typ[0] == Depotwerte.TYP_ETF
def __init__(self, name, ausgeschlossene_kategorien=set()):
self.name = name
self.dauerauftraege = Dauerauftraege()
self.gemeinsamebuchungen = Gemeinsamebuchungen()
self.einzelbuchungen = Einzelbuchungen()
self.sparbuchungen = Sparbuchungen()
self.sparkontos = Kontos()
self.depotwerte = Depotwerte()
self.order = Order()
self.orderdauerauftrag = OrderDauerauftrag()
self.depotauszuege = Depotauszuege()
self.einzelbuchungen.ausgeschlossene_kategorien = ausgeschlossene_kategorien
self.tainted = 0
def test_get_valid_isins(self):
component_under_test = Depotwerte()
component_under_test.add('invalid isin', '-')
component_under_test.add('valid isin', 'isin56789012')
assert component_under_test.get_valid_isins() == ['isin56789012']
def test_get_isin_nach_typ():
depotwerte = Depotwerte()
depotwerte.add('Name_Etf1', 'ISIN_Etf1', depotwerte.TYP_ETF)
depotwerte.add('Name_Etf2', 'ISIN_Etf2', depotwerte.TYP_ETF)
depotwerte.add('Name_Fond', 'ISIN_Fond', depotwerte.TYP_FOND)
result = depotwerte.get_isin_nach_typ()
assert len(result.keys()) == 2
assert result[depotwerte.TYP_ETF] == ['ISIN_Etf1', 'ISIN_Etf2']
assert result[depotwerte.TYP_FOND] == ['ISIN_Fond']
def test_get_valid_isins():
component_under_test = Depotwerte()
component_under_test.add(name='invalid isin',
isin='-',
typ=component_under_test.TYP_ETF)
component_under_test.add(name='valid isin',
isin='isin56789012',
typ=component_under_test.TYP_ETF)
assert component_under_test.get_valid_isins() == ['isin56789012']
def test_get_depotwerte():
component_under_test = Depotwerte()
component_under_test.add(name='0name',
isin='0isin',
typ=component_under_test.TYP_ETF)
component_under_test.add(name='1name',
isin='1isin',
typ=component_under_test.TYP_ETF)
assert component_under_test.get_depotwerte() == ['0isin', '1isin']
def test_get_depotwerte(self):
component_under_test = Depotwerte()
component_under_test.add(
'0name',
'0isin',
)
component_under_test.add(
'1name',
'1isin',
)
assert component_under_test.get_depotwerte() == ['0isin', '1isin']
def test_get_isin_nach_typ_with_empty_db_should_return_empty():
depotwerte = Depotwerte()
assert not depotwerte.get_isin_nach_typ()
class Database:
def __init__(self, name, ausgeschlossene_kategorien=set()):
self.name = name
self.dauerauftraege = Dauerauftraege()
self.gemeinsamebuchungen = Gemeinsamebuchungen()
self.einzelbuchungen = Einzelbuchungen()
self.sparbuchungen = Sparbuchungen()
self.sparkontos = Kontos()
self.depotwerte = Depotwerte()
self.order = Order()
self.orderdauerauftrag = OrderDauerauftrag()
self.depotauszuege = Depotauszuege()
self.einzelbuchungen.ausgeschlossene_kategorien = ausgeschlossene_kategorien
self.tainted = 0
def taint(self):
self.tainted = self.tainted + 1
def is_tainted(self):
return self.taint_number() != 0
def taint_number(self):
return self.tainted + \
self.dauerauftraege.taint_number() + \
self.einzelbuchungen.taint_number() + \
self.gemeinsamebuchungen.taint_number() + \
self.sparbuchungen.taint_number() + \
self.sparkontos.taint_number() + \
self.depotwerte.taint_number() + \
self.order.taint_number() +\
self.depotauszuege.taint_number() +\
self.orderdauerauftrag.taint_number()
def refresh(self):
print('DATABASE: Erneuere Datenbestand')
alle_dauerauftragsbuchungen = self.dauerauftraege.get_all_einzelbuchungen_until_today(
)
self.einzelbuchungen.append_row(alle_dauerauftragsbuchungen)
anteil_gemeinsamer_buchungen = self.gemeinsamebuchungen.anteil_gemeinsamer_buchungen(
)
self.einzelbuchungen.append_row(anteil_gemeinsamer_buchungen)
anteil_sparbuchungen = self.sparbuchungen.get_dynamische_einzelbuchungen(
)
self.einzelbuchungen.append_row(anteil_sparbuchungen)
anteil_orderdauerauftrag = self.orderdauerauftrag.get_all_order_until_today(
)
self.order.append_row(anteil_orderdauerauftrag)
anteil_order = self.order.get_dynamische_einzelbuchungen()
self.einzelbuchungen.append_row(anteil_order)
print('DATABASE: Datenbestand erneuert')
def _write_trenner(self, abrechnunsdatei):
return abrechnunsdatei.write("".rjust(40, "#") + "\n ")
def abrechnen(self,
mindate,
maxdate,
set_ergebnis=None,
verhaeltnis=50,
set_self_kategorie=None,
set_other_kategorie=None):
selector = self.gemeinsamebuchungen.select().select_range(
mindate, maxdate)
name_self = persisted_state.database_instance().name
name_partner = viewcore.name_of_partner()
gemeinsame_buchungen_content = selector.content
select_partner = selector.fuer(name_partner)
select_self = selector.fuer(name_self)
summe_partner = select_partner.sum()
summe_self = select_self.sum()
ausgaben_gesamt = selector.sum()
abrechnunsdatei = StringWriter()
zeitraum = datum_to_german(mindate) + '-' + datum_to_german(maxdate)
abrechnunsdatei.write_line('Abrechnung vom ' +
datum_to_german(time.today()) + ' (' +
zeitraum + ')')
self._write_trenner(abrechnunsdatei)
abrechnunsdatei.write_line('Ergebnis:')
abrechnunsdatei.write_line(set_ergebnis)
abrechnunsdatei.write_empty_line()
self._write_large_table_row(abrechnunsdatei,
'Ausgaben von ' + name_partner,
summe_partner)
self._write_large_table_row(abrechnunsdatei,
'Ausgaben von ' + name_self, summe_self)
abrechnunsdatei.write_line("".ljust(38, "-"))
self._write_large_table_row(abrechnunsdatei, "Gesamt", ausgaben_gesamt)
if verhaeltnis == 50:
self.write_into_file(abrechnunsdatei,
selector.faktor(0.5).to_list(),
'Gesamtausgaben pro Person ')
self.write_into_file(abrechnunsdatei, select_partner.to_list(),
'Ausgaben von ' + name_partner)
self.write_into_file(abrechnunsdatei, select_self.to_list(),
'Ausgaben von ' + name_self)
ausgaben_fuer_partner = DataFrame()
faktor_partner = self._faktor_other(verhaeltnis)
ausgaben_fuer_self = DataFrame()
faktor_self = self._faktor_self(verhaeltnis)
summe_halb = selector.faktor(0.5).sum()
if set_self_kategorie:
faktor_self = 0.5
if set_other_kategorie:
faktor_partner = 0.5
for _, row in gemeinsame_buchungen_content.iterrows():
buchung_partner = self._berechne_abbuchung(
row['Datum'], row['Kategorie'], row['Name'],
("%.2f" % (row['Wert'] * faktor_partner)))
buchung_partner.Dynamisch = False
ausgaben_fuer_partner = pd.concat(
[ausgaben_fuer_partner, buchung_partner])
buchung_self = self._berechne_abbuchung(
row['Datum'], row['Kategorie'], row['Name'],
("%.2f" % (row['Wert'] * faktor_self)))
buchung_self.Dynamisch = False
ausgaben_fuer_self = pd.concat([ausgaben_fuer_self, buchung_self])
if set_self_kategorie:
extra_wert = (ausgaben_gesamt *
self._faktor_self(verhaeltnis)) - summe_halb
extra_ausgleichs_buchung = self._berechne_abbuchung(
maxdate, set_self_kategorie, set_self_kategorie,
("%.2f" % extra_wert))
extra_ausgleichs_buchung.Dynamisch = False
ausgaben_fuer_self = pd.concat(
[ausgaben_fuer_self, extra_ausgleichs_buchung])
if set_other_kategorie:
extra_wert = (ausgaben_gesamt *
self._faktor_other(verhaeltnis)) - summe_halb
extra_ausgleichs_buchung = self._berechne_abbuchung(
maxdate, set_other_kategorie, set_other_kategorie,
("%.2f" % extra_wert))
extra_ausgleichs_buchung.Dynamisch = False
ausgaben_fuer_partner = pd.concat(
[ausgaben_fuer_partner, extra_ausgleichs_buchung])
report = TextReportWriter().generate_report(
ausgaben_fuer_partner, abrechnunsdatei.to_string())
self.einzelbuchungen.append_row(ausgaben_fuer_self)
self.einzelbuchungen.taint()
self.gemeinsamebuchungen.drop(
gemeinsame_buchungen_content.index.tolist())
self.taint()
write_abrechnung("Abrechnung_" + str(time.now()), report)
return report
def _faktor_self(self, verhaeltnis):
return verhaeltnis / 100
def _faktor_other(self, verhaeltnis):
return self._faktor_self(100 - verhaeltnis)
def _write_large_table_row(self, abrechnunsdatei, name, summe):
abrechnunsdatei.write_line(
name.ljust(30, " ") + str("%.2f" % summe).rjust(7, " "))
def write_into_file(self, abrechnunsdatei, ausgaben, title):
abrechnunsdatei.write_empty_line(count=2)
self._write_trenner(abrechnunsdatei)
abrechnunsdatei.write_line(title)
self._write_trenner(abrechnunsdatei)
self._write_tabelle(abrechnunsdatei, ausgaben)
def _write_tabelle(self, writer, tabelle):
writer.write_line(
self._to_left("Datum", 10) + self._to_left(" Kategorie", 14) +
self._to_left("Name", 21) + self._to_right("Wert", 7))
for row in tabelle:
writer.write_line(
datum_to_german(row['Datum']) + " " +
row['Kategorie'].ljust(len("Kategorie "), " ") + " " +
row['Name'].ljust(20, " ") + " " +
str("%.2f" % (row['Wert'])).rjust(7, " "))
def _to_left(self, target_string, size):
return target_string.ljust(size, ' ')
def _to_right(self, target_string, size):
return target_string.rjust(size, ' ')
def _berechne_abbuchung(self, laufdatum, kategorie, name, wert):
return DataFrame([[laufdatum, kategorie, name, wert, True]],
columns=('Datum', 'Kategorie', 'Name', 'Wert',
'Dynamisch'))
def _row_to_dict(self, columns, index, row_data):
row = {'index': index}
for key in columns:
row[key] = row_data[key]
return row
def frame_to_list_of_dicts(self, dataframe):
result_list = []
for index, row_data in dataframe.iterrows():
row = self._row_to_dict(dataframe.columns, index, row_data)
result_list.append(row)
return result_list
def test_edit_shouldEdit(self):
component_under_test = Depotwerte()
component_under_test.add(
'0name',
'0isin',
)
component_under_test.add(
'1name',
'1isin',
)
component_under_test.add(
'2name',
'2isin',
)
assert len(component_under_test.content) == 3
element_before = component_under_test.get(1)
assert element_before == {'index': 1, 'Name': '1name', 'ISIN': '1isin'}
component_under_test.edit(index=1, name='13name', isin='13isin')
assert len(component_under_test.content) == 3
element_after = component_under_test.get(1)
assert element_after == {
'index': 1,
'Name': '13name',
'ISIN': '13isin'
}