#ifndef DATABASE_H
#define DATABASE_H
#include <QDir>
#include <QtSql/QSql>
#include <QtSql/QSqlDatabase>
#include <QtSql/QSqlQuery>
#include <string>
#include <vector>

using std::vector;

class DataBase {
  public:
    DataBase();
    ~DataBase();
    vector<vector<QString>> getData(std::string statement, int cols);
    vector<vector<QString>> getData2(std::string statement, int cols);
    void setData(std::string statement);
    QStringList getDataQStringList(std::string statement);

  private:
    QSqlDatabase * db;
};

#endif // DATABASE_H