: QDialog(parent),

ui(new Ui::ExportCsvDialog),

pdb(db),

m_sQuery(query)

{

// Create UI

ui->setupUi(this);

// If a SQL query was specified hide the table combo box. If not fill it with tables to export

if(query.isEmpty())

{

// Get list of tables to export

objectMap objects = pdb->getBrowsableObjects();

for(objectMap::ConstIterator i=objects.begin();i!=objects.end();++i)

ui->comboTable->addItem(QIcon(QString(":icons/%1").arg(i.value().gettype())), i.value().getname());

// Sort list of tables and select the table specified in the selection parameter or alternatively the first one

ui->comboTable->model()->sort(0);

if(selection.isEmpty())

ui->comboTable->setCurrentIndex(0);

else

ui->comboTable->setCurrentIndex(ui->comboTable->findText(selection));

} else {

// Hide table combo box

ui->labelTable->setVisible(false);

ui->comboTable->setVisible(false);

}

{

// Get filename

QString fileName = QFileDialog::getSaveFileName(

this,

tr("Choose a filename to export data"),

PreferencesDialog::getSettingsValue("db", "defaultlocation").toString(),

tr("Text files(*.csv *.txt)"));

// Only if the user hasn't clicked the cancel button

if(fileName.size() > 0)

{

unsigned int first_column;

// Get data from selected table

SqliteTableModel tableModel(this, pdb);

if(m_sQuery.isEmpty())

{

tableModel.setTable(ui->comboTable->currentText());

first_column = 1;

} else {

tableModel.setQuery(m_sQuery);

first_column = 0;

}

while(tableModel.canFetchMore())

tableModel.fetchMore();

// Prepare the quote and separating characters

QString quoteChar = ui->comboQuoteCharacter->currentText();

QString quotequoteChar = quoteChar + quoteChar;

QString sepChar = ui->comboFieldSeparator->currentText();

if(sepChar == tr("Tab")) sepChar = "\t";

QString newlineChar = "\n";

// Open file

QFile file(fileName);

if(file.open(QIODevice::WriteOnly))

{

// Open text stream to the file

QTextStream stream(&file);

// Put field names in first row if user wants to have them

if(ui->checkHeader->isChecked())

{

for(int i=first_column; i < tableModel.columnCount(); ++i)

{

stream << quoteChar << tableModel.headerData(i, Qt::Horizontal).toString() << quoteChar;

if(i < tableModel.columnCount() - 1)

stream << sepChar;

else

stream << newlineChar;

}

}

// Get and write actual data

for(int i=0;i < tableModel.totalRowCount(); ++i)

{

for(int j=first_column; j < tableModel.columnCount(); ++j)

{

QString content = tableModel.data(tableModel.index(i, j)).toString();

stream << quoteChar << content.replace(quoteChar, quotequoteChar) << quoteChar;

if(j < tableModel.columnCount() - 1)

stream << sepChar;

else

stream << newlineChar;

}

}

// Done writing the file

file.close();

QMessageBox::information(this, QApplication::applicationName(), tr("Export completed."));

QDialog::accept();

} else {

QMessageBox::warning(this, QApplication::applicationName(), tr("Could not open output file."));

}

}