1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
#ifndef CSVIMPORT_H
#define CSVIMPORT_H
#include "csvimport_global.h"
#include "plugins/importplugin.h"
#include "plugins/genericplugin.h"
#include "config_builder.h"
#include "csvserializer.h"
CFG_CATEGORIES(CsvImportConfig,
CFG_CATEGORY(CsvImport,
CFG_ENTRY(bool, FirstRowAsColumns, false)
CFG_ENTRY(int, Separator, 0)
CFG_ENTRY(QString, CustomSeparator, QString())
CFG_ENTRY(bool, NullValues, false)
CFG_ENTRY(QString, NullValueString, QString())
)
)
class QFile;
class QTextStream;
class CSVIMPORTSHARED_EXPORT CsvImport : public GenericPlugin, public ImportPlugin
{
Q_OBJECT
SQLITESTUDIO_PLUGIN("csvimport.json")
public:
CsvImport();
QString getDataSourceTypeName() const;
ImportManager::StandardConfigFlags standardOptionsToEnable() const;
bool beforeImport(const ImportManager::StandardImportConfig& config);
void afterImport();
QList<ColumnDefinition> getColumns() const;
QList<QVariant> next();
CfgMain* getConfig();
QString getImportConfigFormName() const;
bool validateOptions();
QString getFileFilter() const;
bool init();
void deinit();
private:
bool extractColumns();
void defineCsvFormat();
QFile* file = nullptr;
QTextStream* stream = nullptr;
QStringList columnNames;
CsvFormat csvFormat;
CFG_LOCAL_PERSISTABLE(CsvImportConfig, cfg)
};
#endif // CSVIMPORT_H
|
