mirror/sqlitebrowser
1
0
Fork 0
mirror of https://github.com/sqlitebrowser/sqlitebrowser.git synced 2026-01-20 11:00:44 -06:00
Code Issues Packages Projects Releases Wiki Activity

exportcsvdialog: Escape especial characters in headers and values

Browse Source 
When exporting to CSV, escape the chosen separator, the quote
character and the accepted line endings (CR, LF and CRLF).

See issue #219.
This commit is contained in:
Samir Aguiar
2015-03-09 21:26:04 +01:00
parent 6d1a874f80
commit 80edba2cae
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/ExportCsvDialog.cpp
View File

@@ -69,6 +69,9 @@ bool ExportCsvDialog::exportQuery(const QString& sQuery, const QString& sFilenam
QChar sepChar = currentSeparatorChar();
QString newlineChar = "\r\n";
// chars that require escaping
std::string special_chars = newlineChar.toStdString() + sepChar.toAscii() + quoteChar.toAscii();
// Open file
QFile file(sFilename);
if(file.open(QIODevice::WriteOnly))
@@ -88,7 +91,7 @@ bool ExportCsvDialog::exportQuery(const QString& sQuery, const QString& sFilenam
for (int i = 0; i < columns; ++i)
{
QString content = QString::fromUtf8(sqlite3_column_name(stmt, i));
if(content.contains(quoteChar) || content.contains(newlineChar))
if(content.toStdString().find_first_of(special_chars) != std::string::npos)
stream << quoteChar << content.replace(quoteChar, quotequoteChar) << quoteChar;
else
stream << content;
@@ -108,7 +111,7 @@ bool ExportCsvDialog::exportQuery(const QString& sQuery, const QString& sFilenam
QString content = QString::fromUtf8(
(const char*)sqlite3_column_blob(stmt, i),
sqlite3_column_bytes(stmt, i));
if(content.contains(quoteChar) || content.contains(sepChar) || content.contains('\n'))
if(content.toStdString().find_first_of(special_chars) != std::string::npos)
stream << quoteChar << content.replace(quoteChar, quotequoteChar) << quoteChar;
else
stream << content;