from faker import Faker

from src.models import ComdirectCreditCard, ComdirectDepot, ComdirectGiro
from src.settings.constants import Currency
from src.settings.models import AccountInfo

fake = Faker()

BANK_ACCOUNTS = {
    "comdirect": {
        "Girokonto": AccountInfo(
            iban=fake.iban(),
            ingestion_function=lambda x: ComdirectGiro.model_validate(x),
            currency=Currency.EUR,
        ),
        "Visa-Karte (Kreditkarte)": AccountInfo(
            iban=fake.iban(),
            ingestion_function=lambda x: ComdirectCreditCard.model_validate(x),
            currency=Currency.EUR,
        ),
        "Tagesgeld PLUS-Konto": AccountInfo(
            iban=fake.iban(),
            ingestion_function=lambda x: ComdirectGiro.model_validate(x),
            currency=Currency.EUR,
        ),
        "Depot": AccountInfo(
            iban=fake.iban(),
            ingestion_function=lambda x: ComdirectDepot.model_validate(x),
            currency=Currency.EUR,
        ),
    }
}