diff options
author | tpearson <tpearson@283d02a7-25f6-0310-bc7c-ecb5cbfe19da> | 2011-07-04 22:38:03 +0000 |
---|---|---|
committer | tpearson <tpearson@283d02a7-25f6-0310-bc7c-ecb5cbfe19da> | 2011-07-04 22:38:03 +0000 |
commit | dadc34655c3ab961b0b0b94a10eaaba710f0b5e8 (patch) | |
tree | 99e72842fe687baea16376a147619b6048d7e441 /kmymoney2/plugins/ofximport/ofximporterplugin.cpp | |
download | kmymoney-dadc34655c3ab961b0b0b94a10eaaba710f0b5e8.tar.gz kmymoney-dadc34655c3ab961b0b0b94a10eaaba710f0b5e8.zip |
Added kmymoney
git-svn-id: svn://anonsvn.kde.org/home/kde/branches/trinity/applications/kmymoney@1239792 283d02a7-25f6-0310-bc7c-ecb5cbfe19da
Diffstat (limited to 'kmymoney2/plugins/ofximport/ofximporterplugin.cpp')
-rw-r--r-- | kmymoney2/plugins/ofximport/ofximporterplugin.cpp | 688 |
1 files changed, 688 insertions, 0 deletions
diff --git a/kmymoney2/plugins/ofximport/ofximporterplugin.cpp b/kmymoney2/plugins/ofximport/ofximporterplugin.cpp new file mode 100644 index 0000000..21a6466 --- /dev/null +++ b/kmymoney2/plugins/ofximport/ofximporterplugin.cpp @@ -0,0 +1,688 @@ +/*************************************************************************** + ofxiimporterplugin.cpp + ------------------- + begin : Sat Jan 01 2005 + copyright : (C) 2005 by Ace Jones + email : Ace Jones <[email protected]> + ***************************************************************************/ + +/*************************************************************************** + * * + * This program is free software; you can redistribute it and/or modify * + * it under the terms of the GNU General Public License as published by * + * the Free Software Foundation; either version 2 of the License, or * + * (at your option) any later version. * + * * + ***************************************************************************/ + +// ---------------------------------------------------------------------------- +// QT Includes + +#include <qfile.h> +#include <qtextstream.h> +#include <qradiobutton.h> +#include <qspinbox.h> +#include <qdatetimeedit.h> + +// ---------------------------------------------------------------------------- +// KDE Includes + +#include <kgenericfactory.h> +#include <kdebug.h> +#include <kfile.h> +#include <kurl.h> +#include <kaction.h> +#include <kmessagebox.h> + +// ---------------------------------------------------------------------------- +// Project Includes + +#include "ofximporterplugin.h" +#include "konlinebankingstatus.h" +#include "konlinebankingsetupwizard.h" +#include "kofxdirectconnectdlg.h" + +K_EXPORT_COMPONENT_FACTORY( kmm_ofximport, + KGenericFactory<OfxImporterPlugin>( "kmm_ofximport" ) ) + +OfxImporterPlugin::OfxImporterPlugin(QObject *parent, const char *name, const QStringList&) : + KMyMoneyPlugin::Plugin( parent, name ), + KMyMoneyPlugin::ImporterPlugin(), + m_valid( false ) +{ + setInstance(KGenericFactory<OfxImporterPlugin>::instance()); + setXMLFile("kmm_ofximport.rc"); + createActions(); +} + +OfxImporterPlugin::~OfxImporterPlugin() +{ +} + +void OfxImporterPlugin::createActions(void) +{ + new KAction(i18n("OFX..."), "", 0, this, SLOT(slotImportFile()), actionCollection(), "file_import_ofx"); +} + +void OfxImporterPlugin::slotImportFile(void) +{ + KURL url = importInterface()->selectFile(i18n("OFX import file selection"), + "", + "*.ofx *.qfx *.ofc|OFX files (*.ofx, *.qfx, *.ofc)\n*.*|All files (*.*)", + static_cast<KFile::Mode>(KFile::File | KFile::ExistingOnly)); + if(url.isValid()) { + if ( isMyFormat(url.path()) ) { + slotImportFile(url.path()); + } else { + KMessageBox::error( 0, i18n("Unable to import %1 using the OFX importer plugin. This file is not the correct format.").arg(url.prettyURL(0, KURL::StripFileProtocol)), i18n("Incorrect format")); + } + + } +} + +QString OfxImporterPlugin::formatName(void) const +{ + return "OFX"; +} + +QString OfxImporterPlugin::formatFilenameFilter(void) const +{ + return "*.ofx *.qfx *.ofc"; +} + + +bool OfxImporterPlugin::isMyFormat( const QString& filename ) const +{ + // filename is considered an Ofx file if it contains + // the tag "<OFX>" or "<OFC>" in the first 20 lines + // which contain some data. + bool result = false; + + QFile f( filename ); + if ( f.open( IO_ReadOnly ) ) + { + QTextStream ts( &f ); + + int lineCount = 20; + while ( !ts.atEnd() && !result && lineCount != 0) + { + // get a line of data and remove all unnecessary whitepace chars + QString line = ts.readLine().simplifyWhiteSpace(); + if ( line.contains("<OFX>",false) + || line.contains("<OFC>",false) ) + result = true; + // count only lines that contains some non white space chars + if(!line.isEmpty()) + lineCount--; + } + f.close(); + } + + return result; +} + +bool OfxImporterPlugin::import( const QString& filename ) +{ + m_fatalerror = i18n("Unable to parse file"); + m_valid = false; + m_errors.clear(); + m_warnings.clear(); + m_infos.clear(); + + m_statementlist.clear(); + m_securitylist.clear(); + + QCString filename_deep( filename.utf8() ); + + LibofxContextPtr ctx = libofx_get_new_context(); + Q_CHECK_PTR(ctx); + + ofx_set_transaction_cb(ctx, ofxTransactionCallback, this); + ofx_set_statement_cb(ctx, ofxStatementCallback, this); + ofx_set_account_cb(ctx, ofxAccountCallback, this); + ofx_set_security_cb(ctx, ofxSecurityCallback, this); + ofx_set_status_cb(ctx, ofxStatusCallback, this); + libofx_proc_file(ctx, filename_deep, AUTODETECT); + libofx_free_context(ctx); + + if ( m_valid ) + { + m_fatalerror = QString(); + m_valid = storeStatements(m_statementlist); + } + return m_valid; +} + +QString OfxImporterPlugin::lastError(void) const +{ + if(m_errors.count() == 0) + return m_fatalerror; + return m_errors.join("<p>"); +} + +/* __________________________________________________________________________ + * AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA + * + * Static callbacks for LibOFX + * + * YYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYY + */ + +int OfxImporterPlugin::ofxTransactionCallback(struct OfxTransactionData data, void * pv) +{ +// kdDebug(2) << __func__ << endl; + + OfxImporterPlugin* pofx = reinterpret_cast<OfxImporterPlugin*>(pv); + MyMoneyStatement& s = pofx->back(); + + MyMoneyStatement::Transaction t; + + if(data.date_posted_valid==true) + { + QDateTime dt; + dt.setTime_t(data.date_posted, Qt::UTC); + t.m_datePosted = dt.date(); + } + else if(data.date_initiated_valid==true) + { + QDateTime dt; + dt.setTime_t(data.date_initiated, Qt::UTC); + t.m_datePosted = dt.date(); + } + + if(data.amount_valid==true) + { + t.m_amount = MyMoneyMoney(data.amount, 1000); + // if this is an investment statement, reverse the sign. not sure + // why this is needed, so I suppose it's a bit of a hack for the moment. + if (data.invtransactiontype_valid==true) + t.m_amount = -t.m_amount; + } + + if(data.check_number_valid==true) + { + t.m_strNumber = data.check_number; + } + + if(data.fi_id_valid==true) + { + t.m_strBankID = QString("ID ") + data.fi_id; + } + else if(data.reference_number_valid==true) + { + t.m_strBankID = QString("REF ") + data.reference_number; + } + // Decide whether to import NAME or PAYEEID if both are present in the download + if (pofx->m_preferName) { + if(data.name_valid==true) + { + t.m_strPayee = data.name; + } + else if(data.payee_id_valid==true) + { + t.m_strPayee = data.payee_id; + } + } + else { + if(data.payee_id_valid==true) + { + t.m_strPayee = data.payee_id; + } + else if(data.name_valid==true) + { + t.m_strPayee = data.name; + } + } + if(data.memo_valid==true){ + t.m_strMemo = data.memo; + } + + // If the payee or memo fields are blank, set them to + // the other one which is NOT blank. (acejones) + if ( t.m_strPayee.isEmpty() ) + { + // But we only create a payee for non-investment transactions (ipwizard) + if ( ! t.m_strMemo.isEmpty() && data.invtransactiontype_valid == false) + t.m_strPayee = t.m_strMemo; + } + else + { + if ( t.m_strMemo.isEmpty() ) + t.m_strMemo = t.m_strPayee; + } + + if(data.security_data_valid==true) + { + struct OfxSecurityData* secdata = data.security_data_ptr; + + if(secdata->ticker_valid==true){ + t.m_strSymbol = secdata->ticker; + } + + if(secdata->secname_valid==true){ + t.m_strSecurity = secdata->secname; + } + } + + t.m_shares = MyMoneyMoney(); + if(data.units_valid==true) + { + t.m_shares = MyMoneyMoney(data.units, 100000).reduce(); + } + + t.m_price = MyMoneyMoney(); + if(data.unitprice_valid == true) + { + t.m_price = MyMoneyMoney(data.unitprice, 100000).reduce(); + } + + t.m_fees = MyMoneyMoney(); + if(data.fees_valid==true) + { + t.m_fees += MyMoneyMoney(data.fees, 1000).reduce(); + } + + if(data.commission_valid==true) + { + t.m_fees += MyMoneyMoney(data.commission, 1000).reduce(); + } + + bool unhandledtype = false; + QString type; + + if(data.invtransactiontype_valid==true) + { + switch (data.invtransactiontype) + { + case OFX_BUYDEBT: + case OFX_BUYMF: + case OFX_BUYOPT: + case OFX_BUYOTHER: + case OFX_BUYSTOCK: + t.m_eAction = MyMoneyStatement::Transaction::eaBuy; + break; + case OFX_REINVEST: + t.m_eAction = MyMoneyStatement::Transaction::eaReinvestDividend; + break; + case OFX_SELLDEBT: + case OFX_SELLMF: + case OFX_SELLOPT: + case OFX_SELLOTHER: + case OFX_SELLSTOCK: + t.m_eAction = MyMoneyStatement::Transaction::eaSell; + break; + case OFX_INCOME: + t.m_eAction = MyMoneyStatement::Transaction::eaCashDividend; + // NOTE: With CashDividend, the amount of the dividend should + // be in data.amount. Since I've never seen an OFX file with + // cash dividends, this is an assumption on my part. (acejones) + break; + + // + // These types are all not handled. We will generate a warning for them. + // + case OFX_CLOSUREOPT: + unhandledtype = true; + type = "CLOSUREOPT (Close a position for an option)"; + break; + case OFX_INVEXPENSE: + unhandledtype = true; + type = "INVEXPENSE (Misc investment expense that is associated with a specific security)"; + break; + case OFX_JRNLFUND: + unhandledtype = true; + type = "JRNLFUND (Journaling cash holdings between subaccounts within the same investment account)"; + break; + case OFX_MARGININTEREST: + unhandledtype = true; + type = "MARGININTEREST (Margin interest expense)"; + break; + case OFX_RETOFCAP: + unhandledtype = true; + type = "RETOFCAP (Return of capital)"; + break; + case OFX_SPLIT: + unhandledtype = true; + type = "SPLIT (Stock or mutial fund split)"; + break; + case OFX_TRANSFER: + unhandledtype = true; + type = "TRANSFER (Transfer holdings in and out of the investment account)"; + break; + default: + unhandledtype = true; + type = QString("UNKNOWN %1").arg(data.invtransactiontype); + break; + } + } + else + t.m_eAction = MyMoneyStatement::Transaction::eaNone; + + // In the case of investment transactions, the 'total' is supposed to the total amount + // of the transaction. units * unitprice +/- commission. Easy, right? Sadly, it seems + // some ofx creators do not follow this in all circumstances. Therefore, we have to double- + // check the total here and adjust it if it's wrong. + +#if 0 + // Even more sadly, this logic is BROKEN. It consistently results in bogus total + // values, because of rounding errors in the price. A more through solution would + // be to test if the comission alone is causing a discrepency, and adjust in that case. + + if(data.invtransactiontype_valid==true && data.unitprice_valid) + { + double proper_total = t.m_dShares * data.unitprice + t.m_moneyFees; + if ( proper_total != t.m_moneyAmount ) + { + pofx->addWarning(QString("Transaction %1 has an incorrect total of %2. Using calculated total of %3 instead.").arg(t.m_strBankID).arg(t.m_moneyAmount).arg(proper_total)); + t.m_moneyAmount = proper_total; + } + } +#endif + + if ( unhandledtype ) + pofx->addWarning(QString("Transaction %1 has an unsupported type (%2).").arg(t.m_strBankID,type)); + else + s.m_listTransactions += t; + +// kdDebug(2) << __func__ << "return 0 " << endl; + + return 0; +} + +int OfxImporterPlugin::ofxStatementCallback(struct OfxStatementData data, void* pv) +{ +// kdDebug(2) << __func__ << endl; + + OfxImporterPlugin* pofx = reinterpret_cast<OfxImporterPlugin*>(pv); + MyMoneyStatement& s = pofx->back(); + + pofx->setValid(); + + if(data.currency_valid==true) + { + s.m_strCurrency = data.currency; + } + if(data.account_id_valid==true) + { + s.m_strAccountNumber = data.account_id; + } + + if(data.date_start_valid==true) + { + QDateTime dt; + dt.setTime_t(data.date_start, Qt::UTC); + s.m_dateBegin = dt.date(); + } + + if(data.date_end_valid==true) + { + QDateTime dt; + dt.setTime_t(data.date_end, Qt::UTC); + s.m_dateEnd = dt.date(); + } + + if(data.ledger_balance_valid==true) + { + s.m_closingBalance = MyMoneyMoney(data.ledger_balance); + } + +// kdDebug(2) << __func__ << " return 0" << endl; + + return 0; +} + +int OfxImporterPlugin::ofxAccountCallback(struct OfxAccountData data, void * pv) +{ +// kdDebug(2) << __func__ << endl; + + OfxImporterPlugin* pofx = reinterpret_cast<OfxImporterPlugin*>(pv); + pofx->addnew(); + MyMoneyStatement& s = pofx->back(); + + // Having any account at all makes an ofx statement valid + pofx->m_valid = true; + + if(data.account_id_valid==true) + { + s.m_strAccountName = data.account_name; + s.m_strAccountNumber = data.account_id; + } + if(data.bank_id_valid == true) + { + s.m_strRoutingNumber = data.bank_id; + } + if(data.broker_id_valid == true) + { + s.m_strRoutingNumber = data.broker_id; + } + if(data.currency_valid==true) + { + s.m_strCurrency = data.currency; + } + + if(data.account_type_valid==true) + { + switch(data.account_type) + { + case OfxAccountData::OFX_CHECKING : s.m_eType = MyMoneyStatement::etCheckings; + break; + case OfxAccountData::OFX_SAVINGS : s.m_eType = MyMoneyStatement::etSavings; + break; + case OfxAccountData::OFX_MONEYMRKT : s.m_eType = MyMoneyStatement::etInvestment; + break; + case OfxAccountData::OFX_CREDITLINE : s.m_eType = MyMoneyStatement::etCreditCard; + break; + case OfxAccountData::OFX_CMA : s.m_eType = MyMoneyStatement::etCreditCard; + break; + case OfxAccountData::OFX_CREDITCARD : s.m_eType = MyMoneyStatement::etCreditCard; + break; + case OfxAccountData::OFX_INVESTMENT : s.m_eType = MyMoneyStatement::etInvestment; + break; + } + } + + // ask KMyMoney for an account id + s.m_accountId = pofx->account("kmmofx-acc-ref", QString("%1-%2").arg(s.m_strRoutingNumber, s.m_strAccountNumber)).id(); + + // copy over the securities + s.m_listSecurities = pofx->m_securitylist; + +// kdDebug(2) << __func__ << " return 0" << endl; + + return 0; +} + +int OfxImporterPlugin::ofxSecurityCallback(struct OfxSecurityData data, void* pv) +{ + // kdDebug(2) << __func__ << endl; + + OfxImporterPlugin* pofx = reinterpret_cast<OfxImporterPlugin*>(pv); + MyMoneyStatement::Security sec; + + if(data.unique_id_valid==true){ + sec.m_strId = data.unique_id; + } + if(data.secname_valid==true){ + sec.m_strName = data.secname; + } + if(data.ticker_valid==true){ + sec.m_strSymbol = data.ticker; + } + + pofx->m_securitylist += sec; + + return 0; +} + +int OfxImporterPlugin::ofxStatusCallback(struct OfxStatusData data, void * pv) +{ +// kdDebug(2) << __func__ << endl; + + OfxImporterPlugin* pofx = reinterpret_cast<OfxImporterPlugin*>(pv); + QString message; + + // if we got this far, we know we were able to parse the file. + // so if it fails after here it can only because there were no actual + // accounts in the file! + pofx->m_fatalerror = "No accounts found."; + + if(data.ofx_element_name_valid==true) + message.prepend(QString("%1: ").arg(data.ofx_element_name)); + + if(data.code_valid==true) + message += QString("%1 (Code %2): %3").arg(data.name).arg(data.code).arg(data.description); + + if(data.server_message_valid==true) + message += QString(" (%1)").arg(data.server_message); + + if(data.severity_valid==true){ + switch(data.severity){ + case OfxStatusData::INFO: + pofx->addInfo( message ); + break; + case OfxStatusData::ERROR: + pofx->addError( message ); + break; + case OfxStatusData::WARN: + pofx->addWarning( message ); + break; + default: + pofx->addWarning( message ); + pofx->addWarning( "Previous message was an unknown type. 'WARNING' was assumed."); + break; + } + } + +// kdDebug(2) << __func__ << " return 0 " << endl; + + return 0; +} + +bool OfxImporterPlugin::importStatement(const MyMoneyStatement& s) +{ + qDebug("OfxImporterPlugin::importStatement start"); + return statementInterface()->import(s); +} + +const MyMoneyAccount& OfxImporterPlugin::account(const QString& key, const QString& value) const +{ + return statementInterface()->account(key, value); +} + +void OfxImporterPlugin::protocols(QStringList& protocolList) const +{ + protocolList.clear(); + protocolList << "OFX"; +} + +QWidget* OfxImporterPlugin::accountConfigTab(const MyMoneyAccount& acc, QString& name) +{ + name = i18n("Online settings"); + m_statusDlg = new KOnlineBankingStatus(acc, 0, 0); + return m_statusDlg; +} + +MyMoneyKeyValueContainer OfxImporterPlugin::onlineBankingSettings(const MyMoneyKeyValueContainer& current) +{ + MyMoneyKeyValueContainer kvp(current); + // keep the provider name in sync with the one found in kmm_ofximport.desktop + kvp["provider"] = "KMyMoney OFX"; + if(m_statusDlg) { + kvp.deletePair("appId"); + kvp.deletePair("kmmofx-headerVersion"); + if(!m_statusDlg->appId().isEmpty()) + kvp.setValue("appId", m_statusDlg->appId()); + kvp.setValue("kmmofx-headerVersion", m_statusDlg->headerVersion()); + kvp.setValue("kmmofx-numRequestDays", QString::number(m_statusDlg->m_numdaysSpin->value())); + kvp.setValue("kmmofx-todayMinus", QString::number(m_statusDlg->m_todayRB->isChecked())); + kvp.setValue("kmmofx-lastUpdate", QString::number(m_statusDlg->m_lastUpdateRB->isChecked())); + kvp.setValue("kmmofx-pickDate", QString::number(m_statusDlg->m_pickDateRB->isChecked())); + kvp.setValue("kmmofx-specificDate", m_statusDlg->m_specificDate->date().toString()); + kvp.setValue("kmmofx-preferPayeeid", QString::number(m_statusDlg->m_payeeidRB->isChecked())); + kvp.setValue("kmmofx-preferName", QString::number(m_statusDlg->m_nameRB->isChecked())); + } + return kvp; +} + +bool OfxImporterPlugin::mapAccount(const MyMoneyAccount& acc, MyMoneyKeyValueContainer& settings) +{ + Q_UNUSED(acc); + + bool rc = false; + KOnlineBankingSetupWizard wiz(0, "onlinebankingsetup"); + if(wiz.isInit()) { + if(wiz.exec() == QDialog::Accepted) { + rc = wiz.chosenSettings( settings ); + } + } + + return rc; +} + +bool OfxImporterPlugin::updateAccount(const MyMoneyAccount& acc, bool moreAccounts) +{ + Q_UNUSED(moreAccounts); + + try { + if(!acc.id().isEmpty()) { + // Save the value of preferName to be used by ofxTransactionCallback + m_preferName = acc.onlineBankingSettings().value("kmmofx-preferName").toInt() != 0; + KOfxDirectConnectDlg dlg(acc); + + connect(&dlg, SIGNAL(statementReady(const QString&)), + this, SLOT(slotImportFile(const QString&))); + + dlg.init(); + dlg.exec(); + } + } catch (MyMoneyException *e) { + KMessageBox::information(0 ,i18n("Error connecting to bank: %1").arg(e->what())); + delete e; + } + + return false; +} + +void OfxImporterPlugin::slotImportFile(const QString& url) +{ + + if(!import(url)) { + KMessageBox::error( 0, QString("<qt>%1</qt>").arg(i18n("Unable to import %1 using the OFX importer plugin. The plugin returned the following error:<p>%2").arg(url, lastError())), i18n("Importing error")); + } +} + +bool OfxImporterPlugin::storeStatements(QValueList<MyMoneyStatement>& statements) +{ + bool hasstatements = (statements.count() > 0); + bool ok = true; + bool abort = false; + + // FIXME Deal with warnings/errors coming back from plugins + /*if ( ofx.errors().count() ) + { + if ( KMessageBox::warningContinueCancelList(this,i18n("The following errors were returned from your bank"),ofx.errors(),i18n("OFX Errors")) == KMessageBox::Cancel ) + abort = true; + } + + if ( ofx.warnings().count() ) + { + if ( KMessageBox::warningContinueCancelList(this,i18n("The following warnings were returned from your bank"),ofx.warnings(),i18n("OFX Warnings"),KStdGuiItem::cont(),"ofxwarnings") == KMessageBox::Cancel ) + abort = true; + }*/ + + qDebug("OfxImporterPlugin::storeStatements() with %d statements called", static_cast<int>(statements.count())); + QValueList<MyMoneyStatement>::const_iterator it_s = statements.begin(); + while ( it_s != statements.end() && !abort ) { + ok = ok && importStatement((*it_s)); + ++it_s; + } + + if ( hasstatements && !ok ) { + KMessageBox::error( 0, i18n("Importing process terminated unexpectedly."), i18n("Failed to import all statements.")); + } + + return ( !hasstatements || ok ); +} + +#include "ofximporterplugin.moc" +// vim:cin:si:ai:et:ts=2:sw=2: |