eelke/pgLab
1
0
Fork 0
Code Issues 28 Pull requests Projects Releases 1 Packages Wiki Activity Actions

Completed the implementation of the CsvWriter for use with Copy and Export.

Browse source
This commit is contained in:
Eelke Klein 2017-02-04 11:53:20 +01:00
parent 7a22b4cbea
commit 798f8f51b9
2 changed files with 54 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

58
csvwriter.cpp
View file

@ -1,30 +1,64 @@
#include "csvwriter.h"
CsvWriter::CsvWriter()
{
{}
CsvWriter::CsvWriter(QTextStream *output)
: m_output(output)
{}
void CsvWriter::setDestination(QTextStream *output)
{
m_output = output;
m_column = 0;
}
void CsvWriter::setSeperator(QChar ch)
{
m_seperator = ch;
}
//int complexField = (strchr(field, csvWriter->delimiter_) || strchr(field, '\n') || strchr(field, '\"')) ? 1 : 0;
void CsvWriter::setQuote(QChar ch)
{
m_quote = ch;
}
void CsvWriter::writeField(QString field) {
// if field contains any of seperator quote or newline then it needs to be quoted
void CsvWriter::writeField(QString field)
{
QTextStream &out = *m_output;
if (m_column > 0) {
out << m_seperator;
}
// if field contains any of seperator, quote or newline then it needs to be quoted
// when quoted quotes need to be doubled to escape them
bool needs_quotes = false;
int new_len = field.length();
for (auto ch : field) {
if (ch == m_quote) {
++new_len;
}
if (ch == '\n' || ch == m_seperator || ch == m_quote) {
needs_quotes = true;
break;
}
}
if (needs_quotes) {
new_len += 2;
out << m_quote;
for (auto ch : field) {
if (ch == m_quote)
out << m_quote;
out << ch;
}
out << m_quote;
}
else {
out << field;
}
++m_column;
}
void CsvWriter::nextRow()
{
QTextStream &out = *m_output;
out << '\n';
m_column = 0;
}