bool ScientistRepository::addScientist(Scientist scientist)
{
ofstream file;
file.open(fileName.c_str(), std::ios::app);
if (file.is_open())
{
string name = scientist.getName();
enum sexType sex = scientist.getSex();
int yearBorn = scientist.getYearBorn();
int yearDied = scientist.getYearDied();
file << name << constants::FILE_DELIMETER
<< sex << constants::FILE_DELIMETER
<< yearBorn << constants::FILE_DELIMETER;
if (yearDied != constants::YEAR_DIED_DEFAULT_VALUE)
{
file << yearDied;
}
file << '\n';
}
else
{
return false;
}
file.close();
return true;
}
bool ScientistRepository::addScientist(Scientist scientist)
{
db.open();
if (!db.isOpen())
{
return false;
}
QSqlQuery query(db);
stringstream sqlQuery;
sqlQuery << "INSERT INTO Scientists (name, Gender, BirthYear, DeathYear) VALUES ("
<< "'" << scientist.getName() << "', "
<< scientist.getSex() << ", "
<< scientist.getYearBorn() << ", "
<< scientist.getYearDied()
<< ")";
if (!query.exec(QString::fromStdString(sqlQuery.str())))
{
return false;
}
db.close();
return true;
}
bool ScientistRepository::editScientist(Scientist scientist, int id)
{
db.open();
if (!db.isOpen())
{
return false;
}
QSqlQuery query(db);
stringstream sqlQuery;
sqlQuery << "UPDATE Scientists SET "
<< "name = " << "'" << scientist.getName() << "', "
<< "sex = " << scientist.getSex() << ", "
<< "yearBorn = " << scientist.getYearBorn() << ", "
<< "yearDied = " << scientist.getYearDied() << ", "
<< "info = " << "'" << scientist.getInfo() << "' "
<< "WHERE id = " << id;
if (!query.exec(QString::fromStdString(sqlQuery.str())))
{
return false;
}
db.close();
return true;
}
bool ScientistRepository::addScientist(Scientist scientist)
{
if(db.open())
{
QSqlQuery query(db);
string tableCreation ="CREATE TABLE IF NOT EXISTS scientists(sci_id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL , name VARCHAR, sex INTEGER, yearBorn INTEGER, yearDied INTEGER)";
query.exec(QString(tableCreation.c_str()));
string sSex = utils::intToString(scientist.getSex());
string sYearBorn = utils::intToString(scientist.getYearBorn());
string queryInsert = "INSERT INTO scientists(name,sex,yearBorn)VALUES('"+scientist.getName()+"','"+sSex+"','"+sYearBorn+"')";
if(query.exec(QString(queryInsert.c_str()))) //exec returns true if it is successful
{
if(scientist.getYearDied() != constants::YEAR_NOT_ENTERED_DEFAULT_VALUE)
{
string sYearDied = utils::intToString(scientist.getYearDied());
queryInsert = "UPDATE scientists SET yearDied = '" + sYearDied + "' WHERE sci_id = (SELECT MAX(sci_id) FROM scientists)";
query.exec(QString(queryInsert.c_str()));
}
}
else
{ //first query failed
db.close();
return false;
}
db.close();
}
else //file open failed
return false;
return true;
}
void MainWindow::displayScientist(vector<Scientist> scientists) {
ui->table_scientists->clearContents();
ui->table_scientists->setRowCount(scientists.size());
for (unsigned int i = 0; i < scientists.size(); i++) {
Scientist currentScientist = scientists.at(i);
QString FirstName = QString::fromStdString(currentScientist.getFirstName());
QString LastName = QString::fromStdString(currentScientist.getLastName());
QString Sex = QString::fromStdString(currentScientist.getSex());
QString YearOfBirth = QString::fromStdString(currentScientist.getYearOfBirth());
QString YearOfDeath = QString::fromStdString(currentScientist.getYearOfDeath());
ui->table_scientists->setItem(i, 0, new QTableWidgetItem(FirstName));
ui->table_scientists->setItem(i, 1, new QTableWidgetItem(LastName));
ui->table_scientists->setItem(i, 2, new QTableWidgetItem(Sex));
ui->table_scientists->setItem(i, 3, new QTableWidgetItem(YearOfBirth));
ui->table_scientists->setItem(i, 4, new QTableWidgetItem(YearOfDeath));
}
currentlyDisplayedScientists = scientists;
}
bool ScientistRepository::removeScientist(Scientist scientist)
{
db.open();
QSqlQuery query(db);
stringstream sqlQuery;
//All scientists with the exact same information as the input scientist will be deleted from the database. The primary key is irrelevant.
sqlQuery << "DELETE FROM scientists WHERE name = '" << scientist.getName() << "' AND sex = " << scientist.getSex() << " AND yearBorn = " << scientist.getYearBorn();
if (scientist.getYearDied() == constants::YEAR_NOT_ENTERED_DEFAULT_VALUE)
{ //No year of death entered, search empty fields
sqlQuery << " AND yearDied IS NULL";
}
else
{ //year of death entered, normal search
sqlQuery << " AND yearDied = " << scientist.getYearDied();
}
bool success = query.exec(QString::fromStdString(sqlQuery.str()));
db.close();
return success;
}
