DataCleaner.hpp

Go to the documentation of this file.

#pragma once

#include "../utility/TableHelper.hpp"

#include <string>
#include <vector>


namespace ISM
{

class DataCleaner
{
    public:
        void cleanModels(const std::vector<std::string>& databaseFiles)
        {
            for (const std::string& databaseFile : databaseFiles)
            {
                TableHelper database(databaseFile);
                std::cout << "deleting model from " << databaseFile << std::endl;
                database.dropModelTables();
            }
        }

        void cleanRecords(const std::vector<std::string>& databaseFiles)
        {
            for (const std::string& databaseFile : databaseFiles)
            {
                TableHelper database(databaseFile);
                std::cout << "deleting record from " << databaseFile << std::endl;
                database.dropRecordTables();
            }
        }
};
}