diff --git a/src/main/java/de/deadlocker8/budgetmaster/database/Database.java b/src/main/java/de/deadlocker8/budgetmaster/database/Database.java index d839e1db6f2ec0501633a208a2e8c27330ad4728..15ead6533389c1255a8ee1378ca04be99627cd6c 100644 --- a/src/main/java/de/deadlocker8/budgetmaster/database/Database.java +++ b/src/main/java/de/deadlocker8/budgetmaster/database/Database.java @@ -2,6 +2,7 @@ package de.deadlocker8.budgetmaster.database; import com.google.gson.annotations.Expose; import de.deadlocker8.budgetmaster.accounts.Account; +import de.deadlocker8.budgetmaster.accounts.AccountType; import de.deadlocker8.budgetmaster.categories.Category; import de.deadlocker8.budgetmaster.charts.Chart; import de.deadlocker8.budgetmaster.images.Image; @@ -12,6 +13,7 @@ import de.deadlocker8.budgetmaster.transactions.Transaction; import java.util.LinkedHashMap; import java.util.List; import java.util.Map; +import java.util.stream.Collectors; public class Database { @@ -88,7 +90,12 @@ public class Database final Map<EntityType, Integer> numberOfEntitiesByType = new LinkedHashMap<>(); numberOfEntitiesByType.put(EntityType.CATEGORY, categories.size()); - numberOfEntitiesByType.put(EntityType.ACCOUNT, accounts.size()); + + final List<Account> customAccounts = accounts.stream() + .filter(account -> account.getType() == AccountType.CUSTOM) + .collect(Collectors.toList()); + numberOfEntitiesByType.put(EntityType.ACCOUNT, customAccounts.size()); + numberOfEntitiesByType.put(EntityType.TRANSACTION, transactions.size()); numberOfEntitiesByType.put(EntityType.TEMPLATE, templates.size()); numberOfEntitiesByType.put(EntityType.IMAGE, images.size());