windowmain.cpp

#include "windowmain.h"
#include "ui_windowmain.h"
#include <QTimer>
#include <QTextBrowser>

#include <iostream>

/***********************************************************************************************************
 * We setup the associated UI and instantiate a database connection for the WinowMain model
 * We also set the style sheet of the QTableView embedded db's row highlight color.
***********************************************************************************************************/
WindowMain::WindowMain(QWidget *parent) :
    QWidget(parent),
    ui(new Ui::WindowMain)
{
    ui->setupUi(this);
    DatabaseConnection *dbc = new DatabaseConnection(this);
    db3 = dbc->establishConnection("three");

    ui->tableWidget_db_3->setStyleSheet("QTableView:item:selected {background-color: #F56525; color: #FFFFFF}\n"
                         "QTableView:item:selected:focus {background-color: #F56525;}");
    ui->tableWidget_db_4->setStyleSheet("QTableView:item:selected {background-color: #F56525; color: #FFFFFF}\n"
                         "QTableView:item:selected:focus {background-color: #F56525;}");



    //initialize calendar
    ui->scheddate->setText(
                QDate::currentDate().toString("dd.MM.yyyy")
                );
    ui->schedstart->setText(
                ui->calendarWidget1_2->getStart(QDate::currentDate())
                );
    ui->schedend->setText(
                ui->calendarWidget1_2->getEnd(QDate::currentDate())
                );

    /***************************************************************************************************************/
    /***************************************** QT CHART BEGIN ******************************************************/
    /***************************************************************************************************************/

    QSqlQuery query_teams(QSqlDatabase::database("three"));
    QString querytheteams;
    querytheteams = "SELECT DISTINCT Team FROM Project";
    query_teams.exec(querytheteams);

    //Fill a new dropdown
    while (query_teams.next()) {
        ui->comboBox->addItem(query_teams.value(0).toString());
    }

    /***************************************************************************************************************/
    /******************************************* QT CHART END ******************************************************/
    /***************************************************************************************************************/

    ui->oopsNoEmployee->hide();
    ui->oopsNoTeamSelected->hide();
    ui->oopsSearchFilter->hide();
    ui->oopsTooManyEmp->hide();
    ui->oopsTooManyTeams->hide();
    ui->oopsSearchEmpty->hide();
    ui->oopsTeamEmpty->hide();

    sound = new QSound(":/new/prefix1/pop.wav");
}




WindowMain::~WindowMain()
{
    delete ui;
}





/***********************************************************************************************************
 * This button is found in the Control tab.
 * When it is clicked, the user wants to view/edit the selected employee.
 * If there is in fact a selected row in the embedded database, a ControlTab_ViewEditEmployee object is
 * instantiated (which is the main UI for viewing/editing existing details of an employee).
 * If there is no epmloyee selected, a MessageBox will pop up in error.
***********************************************************************************************************/
void WindowMain::on_pushButton_vieweditemployee_2_clicked()
{
    QModelIndexList rowSelection = ui->tableWidget_db_3->selectionModel()->selectedRows();
    if (rowSelection.count() != 1){
        if (rowSelection.count() == 0) {
            ui->oopsNoEmployee->show();
            ui->oopsTooManyEmp->hide();
            ui->oopsNoTeamSelected->hide();
            ui->oopsTooManyTeams->hide();
            ui->oopsSearchFilter->hide();
            ui->oopsSearchEmpty->hide();
            ui->oopsTeamEmpty->hide();
        } else {
            ui->oopsTooManyEmp->show();
            ui->oopsNoTeamSelected->hide();
            ui->oopsTooManyTeams->hide();
            ui->oopsSearchFilter->hide();
            ui->oopsNoEmployee->hide();
            ui->oopsSearchEmpty->hide();
            ui->oopsTeamEmpty->hide();
        }
        sound->play();
    } else {

        QModelIndex index = ui -> tableWidget_db_3 -> selectionModel() -> currentIndex();

        QVector<QString> fields = { };
        for (int i = 0; i < ui -> tableWidget_db_3 -> columnCount(); i++) {
            QString str = ui -> tableWidget_db_3 -> model() -> index(index.row(), i).data().toString();
            fields.push_back(str);
            std::cout << fields.back().toStdString() << std::endl;
        }

        viewEditEmployeeWindow = new ControlTab_ViewEditEmployee();
        viewEditEmployeeWindow -> set_fields(fields, fields);
        viewEditEmployeeWindow -> set_view_position(userPosition);
        viewEditEmployeeWindow -> set_edit_mode_lock();
        viewEditEmployeeWindow -> grab_password();
        viewEditEmployeeWindow->set_db_table_refs(ui->tableWidget_db_3, ui->tableWidget_db_4);
        viewEditEmployeeWindow -> show();
    }

}





/****************************************************************************************************************************
 * Details from the employee database are loaded into the QTableWidget that acts as the embedded MYSQL database for employees.
 * To get this data a query is created that gets all the employees from the Employee table in the db, and loads them one by
 * one into a newly created row for the QTableWidget employee embedded db.
*****************************************************************************************************************************/
void WindowMain::resetEmployeeTable(){

    QSqlQuery query_getEmployeeRow(QSqlDatabase::database("three"));
    QString employeeRowSqlString;
    employeeRowSqlString = "SELECT EmployeeID, Name_Last, Name_First, Salary, SSN, Position_Code, Username, Password, Team FROM Employee";
    query_getEmployeeRow.exec(employeeRowSqlString);

    ui->tableWidget_db_3->setRowCount(0);
    int col = 0;

    while (query_getEmployeeRow.next()) {
        ui->tableWidget_db_3->insertRow ( ui->tableWidget_db_3->rowCount() );
        int row = ui->tableWidget_db_3->rowCount()-1;
        while (col < 9){
            if (col == 7){
                QString password = query_getEmployeeRow.value(col).toString();
                QString encryptedPassword = "";
                for (auto letter : password){
                    encryptedPassword.append("*");
                }
                ui->tableWidget_db_3->setItem( row, col, new QTableWidgetItem(encryptedPassword));
            }
            else{
                ui->tableWidget_db_3->setItem( row, col, new QTableWidgetItem(query_getEmployeeRow.value(col).toString()));
            }
            col++;
        }
        col = 0;
    }
}





/****************************************************************************************************************************
 * See details of resetEmployeeTable. All the same applies but to the Team table in the db rather than the Employee table.
*****************************************************************************************************************************/
void WindowMain::resetTeamTable(){
    // Populates the Team embedded db
    QSqlQuery query_numTeams(QSqlDatabase::database("three"));
    QSqlQuery query_getTeamRow(QSqlDatabase::database("three"));
    QString teamRowSqlString;
    query_numTeams.exec("SELECT COUNT(*) FROM Team");
    teamRowSqlString = "SELECT TeamID, TeamName, ProjectsAssigned, ProjectsCompleted FROM Team";
    query_getTeamRow.exec(teamRowSqlString);
    ui->tableWidget_db_4->setRowCount(0);
    int col = 0;

    while (query_getTeamRow.next()) {
        ui->tableWidget_db_4->insertRow ( ui->tableWidget_db_4->rowCount() );
        int row = ui->tableWidget_db_4->rowCount()-1;
        while (col < 4){
            ui->tableWidget_db_4->setItem( row, col, new QTableWidgetItem(query_getTeamRow.value(col).toString()));
            col++;
        }
        col = 0;
    }
}



/***********************************************************************************************************
 * When the WindowMain object is first viewed, it opens on the MyEDI tab. When the tab is changed,
 * we reset the embedded db tables so duplicated rows do not occur.
***********************************************************************************************************/
void WindowMain::on_tabWidget_currentChanged(int index)
{
    ui->oopsNoEmployee->hide();
    ui->oopsNoTeamSelected->hide();
    ui->oopsSearchFilter->hide();
    ui->oopsTooManyEmp->hide();
    ui->oopsTooManyTeams->hide();
    ui->oopsSearchEmpty->hide();
    ui->oopsTeamEmpty->hide();
    resetEmployeeTable();
    resetTeamTable();
}





/***********************************************************************************************************
 * When log out button pushed from WindowMain, we see a popup window saying "Goodbye!".
***********************************************************************************************************/
void WindowMain::on_pushButton_LogOut_clicked()
{
    QMessageBox::information(
        this,
        tr("EDI: Employee Data Interface"), // Window text
        tr("Goodbye!") );                   // Text displayed inside of window
    this -> close();
}





/***********************************************************************************************************
 * Sets the "Welcome <user>" message at the top of the WindowMain UI.
***********************************************************************************************************/
void WindowMain::setWelcomeName(QString userFirstName){
    this->userFirstName = userFirstName;
    ui->label_welcomeUser->setText("Welcome, " + userFirstName + ".");
}





/***********************************************************************************************************
 * Stores a copy of the user's position.
***********************************************************************************************************/
void WindowMain::setPosition(QString userPos) {
    userPosition = userPos;
}





/***********************************************************************************************************
 * Disables tabs indexed 1 and 2, or the Control tab and the Analyze tab when a users position code matches
 * that of a basic level employee "E".
***********************************************************************************************************/
void WindowMain::setDisabledFeatures(QString userPositionCode){
    if (userPositionCode == "E"){
        ui->tabWidget->setTabEnabled(1, false);
        ui->tabWidget->setTabEnabled(2, false);
    }
}





/***********************************************************************************************************
 * Searches for an employee in the embedded employee db based on the radio button/filter that is selected.
***********************************************************************************************************/
void WindowMain::on_pushButton_SearchEmployee_clicked(){

    QString lineEdit_searchEmployee = ui->lineEdit_searchEmployee->text();
    QSqlQuery query_searchEmployee(QSqlDatabase::database("three"));
    QString stringSearchEmployee = "";

    if (lineEdit_searchEmployee != ""){ // The user entered SOMETHING in the search field. It is not empty.
        if(ui->radioButton_ID->isChecked()){
            query_searchEmployee.prepare("SELECT * FROM Employee WHERE EmployeeID = :employeeID");
            query_searchEmployee.bindValue(":employeeID", lineEdit_searchEmployee);
            query_searchEmployee.exec();
        }
        else if(ui->radioButton_FirstName->isChecked()){
            query_searchEmployee.prepare("SELECT * FROM Employee WHERE Name_First = :employeeName_First");
            query_searchEmployee.bindValue(":employeeName_First", lineEdit_searchEmployee);
            query_searchEmployee.exec();
        }
        else if(ui->radioButton_LastName->isChecked()){
            query_searchEmployee.prepare("SELECT * FROM Employee WHERE Name_Last = :employeeName_Last");
            query_searchEmployee.bindValue(":employeeName_Last", lineEdit_searchEmployee);
            query_searchEmployee.exec();
        }
        else if(ui->radioButton_SSN->isChecked()){
            query_searchEmployee.prepare("SELECT * FROM Employee WHERE SSN = :employee_SSN");
            query_searchEmployee.bindValue(":employee_SSN", lineEdit_searchEmployee);
            query_searchEmployee.exec();
        }
        else if (ui->radioButton_Position->isChecked()){
            query_searchEmployee.prepare("SELECT * FROM Employee WHERE Position_Code = :employeePosition_Code");
            query_searchEmployee.bindValue(":employeePosition_Code", lineEdit_searchEmployee);
            query_searchEmployee.exec();
        }
        else if(ui->radioButton_Salary->isChecked()){
            query_searchEmployee.prepare("SELECT * FROM Employee WHERE Salary = :employeeSalary");
            query_searchEmployee.bindValue(":employeeSalary", lineEdit_searchEmployee);
            query_searchEmployee.exec();
        }
        else if(ui->radioButton_Username->isChecked()){
            query_searchEmployee.prepare("SELECT * FROM Employee WHERE Username = :employeeUsername");
            query_searchEmployee.bindValue(":employeeUsername", lineEdit_searchEmployee);
            query_searchEmployee.exec();
        }
        else if(ui->radioButton_Team->isChecked()){
            query_searchEmployee.prepare("SELECT * FROM Employee WHERE Team = :employeeTeam");
            query_searchEmployee.bindValue(":employeeTeam", lineEdit_searchEmployee);
            query_searchEmployee.exec();
        }
        else{ // No Radio button was selected
            ui->oopsSearchFilter->show();
            sound->play();
            ui->oopsNoEmployee->hide();
            ui->oopsNoTeamSelected->hide();
            ui->oopsTooManyEmp->hide();
            ui->oopsTooManyTeams->hide();
            ui->oopsSearchEmpty->hide();
            ui->oopsTeamEmpty->hide();
        }


        // The search of the embedded db table returns at least one employe, based on the radio button/filter selected.
        if (query_searchEmployee.size() > 0){
            ui->tableWidget_db_3->setRowCount(0);
            int col = 0;
            while (query_searchEmployee.next()) {
                ui->tableWidget_db_3->insertRow ( ui->tableWidget_db_3->rowCount() );
                int row = ui->tableWidget_db_3->rowCount()-1;
                ui->tableWidget_db_3->setItem( row, 0, new QTableWidgetItem(query_searchEmployee.value(0).toString()));
                ui->tableWidget_db_3->setItem( row, 1, new QTableWidgetItem(query_searchEmployee.value(1).toString()));
                ui->tableWidget_db_3->setItem( row, 2, new QTableWidgetItem(query_searchEmployee.value(2).toString()));
                ui->tableWidget_db_3->setItem( row, 3, new QTableWidgetItem(query_searchEmployee.value(3).toString()));
                ui->tableWidget_db_3->setItem( row, 4, new QTableWidgetItem(query_searchEmployee.value(4).toString()));
                ui->tableWidget_db_3->setItem( row, 5, new QTableWidgetItem(query_searchEmployee.value(5).toString()));
                ui->tableWidget_db_3->setItem( row, 6, new QTableWidgetItem(query_searchEmployee.value(6).toString()));
                ui->tableWidget_db_3->setItem( row, 7, new QTableWidgetItem(query_searchEmployee.value(7).toString()));

                QString password = query_searchEmployee.value(7).toString();
                QString encryptedPassword = "";
                for (auto letter : password){
                    encryptedPassword.append("*");
                }
                ui->tableWidget_db_3->setItem( row, 7, new QTableWidgetItem(encryptedPassword));
                col++;
            }
        }
        else{
            ui->tableWidget_db_3->setRowCount(0);
        }
    }
    else{
        ui->oopsSearchEmpty->show();
        sound->play();
        ui->oopsNoEmployee->hide();
        ui->oopsNoTeamSelected->hide();
        ui->oopsSearchFilter->hide();
        ui->oopsTooManyEmp->hide();
        ui->oopsTooManyTeams->hide();
        ui->oopsTeamEmpty->hide();
    }

}





/***********************************************************************************************************
 * The user wants to reset the employee search. The table will be repopulated with all employee data.
***********************************************************************************************************/
void WindowMain::on_pushButton_Reset_clicked()
{
    resetEmployeeTable();
}





/***********************************************************************************************************
 * Searches for a team in the embedded team db. Search is performed on TeamID or TeamName.
***********************************************************************************************************/
void WindowMain::on_pushButton_SearchTeam_clicked()
{
    QString lineEdit_searchTeam = ui->lineEdit_searchTeam->text();
    QSqlQuery query_searchTeam(QSqlDatabase::database("three"));
    QString stringSearchTeam = "";

    if (lineEdit_searchTeam != ""){
        query_searchTeam.prepare("SELECT * FROM Team WHERE TeamID = :teamSearch OR TeamName = :teamSearch");
        query_searchTeam.bindValue(":teamSearch", lineEdit_searchTeam);
        query_searchTeam.exec();

        if (query_searchTeam.size() > 0){

            ui->tableWidget_db_4->setRowCount(0);
            int col = 0;

            while (query_searchTeam.next()) {
                ui->tableWidget_db_4->insertRow ( ui->tableWidget_db_4->rowCount() );
                int row = ui->tableWidget_db_4->rowCount()-1;
                ui->tableWidget_db_4->setItem( row, 0, new QTableWidgetItem(query_searchTeam.value(0).toString()));
                ui->tableWidget_db_4->setItem( row, 1, new QTableWidgetItem(query_searchTeam.value(1).toString()));
                ui->tableWidget_db_4->setItem( row, 2, new QTableWidgetItem(query_searchTeam.value(2).toString()));
                ui->tableWidget_db_4->setItem( row, 3, new QTableWidgetItem(query_searchTeam.value(3).toString()));
                col++;
            }
        }
        else{
            ui->tableWidget_db_4->setRowCount(0);
        }
    }
    else{
        ui->oopsTeamEmpty->show();
        sound->play();
        ui->oopsNoEmployee->hide();
        ui->oopsNoTeamSelected->hide();
        ui->oopsSearchFilter->hide();
        ui->oopsTooManyEmp->hide();
        ui->oopsTooManyTeams->hide();
        ui->oopsSearchEmpty->hide();
    }

}




/***********************************************************************************************************
 * The user wants to reset the team search. The table will be repopulated with all employee data.
***********************************************************************************************************/
void WindowMain::on_pushButton_TeamReset_clicked()
{
    resetTeamTable();
}





/*************************************************************************************
 * The user selected to create a New Team.
**************************************************************************************/
void WindowMain::on_pushButton_newTeam_clicked()
{
    newTeam = new NewTeam();
    newTeam->set_db_table_refs(ui->tableWidget_db_3, ui->tableWidget_db_4);
    newTeam->show();
}





/***********************************************************************************************************
 * The user selected to View/Edit Team for the currently selected team row in the embedded team db.
***********************************************************************************************************/
void WindowMain::on_pushButton_viewEditTeam_clicked()
{
    QModelIndexList rowSelection = ui->tableWidget_db_4->selectionModel()->selectedRows();
    if (rowSelection.count() != 1){
        if (rowSelection.count() == 0){
            ui->oopsNoTeamSelected->show();
            ui->oopsNoEmployee->hide();
            ui->oopsSearchFilter->hide();
            ui->oopsTooManyEmp->hide();
            ui->oopsTooManyTeams->hide();
            ui->oopsSearchEmpty->hide();
            ui->oopsTeamEmpty->hide();
        }
        else{
            ui->oopsTooManyTeams->show();
            ui->oopsNoEmployee->hide();
            ui->oopsNoTeamSelected->hide();
            ui->oopsSearchFilter->hide();
            ui->oopsTooManyEmp->hide();
            ui->oopsSearchEmpty->hide();
            ui->oopsTeamEmpty->hide();
        }
        sound->play();
    } else {
        QModelIndex index = ui -> tableWidget_db_4 -> selectionModel() -> currentIndex();
        QVector<QString> fields = { };
        for (int i = 0; i < ui -> tableWidget_db_4 -> columnCount(); i++) {
            QString str = ui -> tableWidget_db_4 -> model() -> index(index.row(), i).data().toString();
            fields.push_back(str);
        }
        viewEditTeamWindow = new ControlTab_ViewEditTeamWindow();
        viewEditTeamWindow -> set_fields(fields);
        viewEditTeamWindow->set_db_table_refs(ui->tableWidget_db_3, ui->tableWidget_db_4);
        viewEditTeamWindow -> show();
    }
}





/***********************************************************************************************************
 * process(ControlTab_ViewEditTeamWindow *viewEditTeamWindow): resets the team table.
***********************************************************************************************************/
void WindowMain::process(ControlTab_ViewEditTeamWindow *viewEditTeamWindow) {
      resetTeamTable();
}






/***********************************************************************************************************
 * Hides the Edi helper popup when no employee was selected, but the user clicked "View/Edit Employee"
***********************************************************************************************************/
void WindowMain::on_pushButton_noEmployee_clicked()
{
    ui->oopsNoEmployee->hide();
}






/***********************************************************************************************************
 * Hides the Edi helper popup when no employee was selected, but the user clicked "View/Edit Employee"
***********************************************************************************************************/
void WindowMain::on_pushButton_noTeamSelected_clicked()
{
    ui->oopsNoTeamSelected->hide();
}





/*********************************************************************************************************
 * User pressed Search for an employee
*********************************************************************************************************/
void WindowMain::on_pushButton_SearchField_clicked()
{
    ui->oopsSearchFilter->hide();
}





/*********************************************************************************************************
 * User pressed View/Edit Employee but too many rows were selected
*********************************************************************************************************/
void WindowMain::on_pushButton_tooManyEmp_clicked()
{
    ui->oopsTooManyEmp->hide();
}





/*********************************************************************************************************
 * User pressed View/Edit Team but too many rows were selected
*********************************************************************************************************/
void WindowMain::on_pushButton_tooManyTeams_clicked()
{
    ui->oopsTooManyTeams->hide();
}





/*********************************************************************************************************
 * Calendar stuff in MyEdi Tab - Evan
*********************************************************************************************************/

void WindowMain::on_calendarWidget1_2_clicked(const QDate &date)
{
    ui->scheddate->setText(
                date.toString("dd.MM.yyyy")
                );
    ui->schedstart->setText(
                ui->calendarWidget1_2->getStart(date)
                );
    ui->schedend->setText(
                ui->calendarWidget1_2->getEnd(date)
                );
}


// the event which updates the calendar
void WindowMain::on_refreshcalbutton_clicked()
{
    ui->calendarWidget1_2->refresh();
}






/*********************************************************************************************************
 * User pressed Search for an employee but did not enter anything in the employee search field
*********************************************************************************************************/
void WindowMain::on_pushButton_SearchEmpty_clicked()
{
    ui->oopsSearchEmpty->hide();
}







/*********************************************************************************************************
 * User pressed Search for a team but did not enter anything in the team search field
*********************************************************************************************************/
void WindowMain::on_pushButton_TeamEmpty_clicked()
{
    ui->oopsTeamEmpty->hide();
}






/*********************************************************************************************************
 * Setting the MyEdi calendar to highlight the shift dates for the currently logged in user.
*********************************************************************************************************/
void WindowMain::setEmployee(QString employeeID){
    curEmployee = employeeID;
    ui->calendarWidget1_2->setEmployee(curEmployee);

    QSqlQuery query(QSqlDatabase::database("three"));
        QString firstName, lastName, salary, teamName, numProjectsCompleted, numProjectsAssigned, assignedProjList, completedProjList;
        QVector<QString> completedProjects, assignedProjects;

        query.prepare("SELECT E.Name_First, E.Name_Last, E.Salary, T.TeamName FROM Employee E JOIN Team T ON E.Team = T.TeamID WHERE E.EmployeeID = :curID");
        query.bindValue(":curID", curEmployee);
        query.exec();

        while(query.next()){
            firstName = query.value(0).toString();
            lastName = query.value(1).toString();
            salary = query.value(2).toString();
            teamName = query.value(3).toString();
        }
        query.clear();


        query.prepare("SELECT P.ProjectName FROM Employee E JOIN Team T ON E.Team = T.TeamID JOIN Project P ON P.Team = T.TeamID WHERE E.EmployeeID = :curID AND P.ProjectStatus = 'Completed'");
        query.bindValue(":curID", curEmployee);
        query.exec();

        int completedProject = 0;
        while(query.next()){
            completedProjects.push_back(query.value(0).toString());
            completedProject++;
        }
        query.clear();


        query.prepare("SELECT P.ProjectName FROM Employee E JOIN Team T ON E.Team = T.TeamID JOIN Project P ON P.Team = T.TeamID WHERE E.EmployeeID = :curID AND P.ProjectStatus = 'Pending'");
        query.bindValue(":curID", curEmployee);
        query.exec();

        int assgnProject = 0;
        while(query.next()){
            assignedProjects.push_back(query.value(0).toString());
            assgnProject++;
        }
        query.clear();

        while(! assignedProjects.empty()){
            assignedProjList += assignedProjects.front() + "\n";
            assignedProjects.pop_front();
        }

        while(! completedProjects.empty()){
            completedProjList += completedProjects.front() + "\n";
            completedProjects.pop_front();
        }

        ui->employeename->setText(firstName + " " + lastName);
        ui->salary->setText(salary);
        ui->employeeteam->setText(teamName);
        ui->allprojpush->setText(QVariant(completedProject).toString());
        ui->curprojpush->setText(QVariant(assgnProject).toString());

        ui->textBrowser1->append(assignedProjList);
        ui->textBrowser2->append(completedProjList);

        QTextCursor cursor = ui -> textBrowser1 -> textCursor();
        cursor.setPosition(0);
        ui -> textBrowser1 -> setTextCursor(cursor);

        cursor = ui -> textBrowser2 -> textCursor();
        cursor.setPosition(0);
        ui -> textBrowser2 -> setTextCursor(cursor);

}






/***********************************************************************************************************
 * The user selected the add employee button.
***********************************************************************************************************/
void WindowMain::on_pushButton_newEmployee_clicked() {
    newEmployeeWindow = new ControlTab_NewEmployee();
    newEmployeeWindow -> populate_position_comboBox(userPosition);
    newEmployeeWindow -> show();

}



void WindowMain::on_comboBox_currentIndexChanged(const QString &arg1)
{
//    QString Team;
    if (ui->comboBox->currentText() == "01"){
            Team = "01";
            DrawChart( Team, false );

        }
    else if (ui->comboBox->currentText() == "02"){
            Team = "02";
            DrawChart( Team, false );
        }
    else if (ui->comboBox->currentText() == "03"){
            Team = "03";
            DrawChart( Team, false );
        }
}


void WindowMain::DrawChart(QString Team, bool last7) {
    series = new QLineSeries();
    QValueAxis *axisX = new QValueAxis;
    QValueAxis *axisY = new QValueAxis;

    // Change color of the plotted line
    QPen pen(QRgb(0xfdb157));
    pen.setWidth(5);
    series->setPen(pen);

    // Change color of axis lines
    QPen axisPen(QRgb(0xd18952));
    axisPen.setWidth(2);
    axisX->setLinePen(axisPen);
    axisX->setTickCount(10);
    axisX->setMinorTickCount(1);

    axisY->setLinePen(axisPen);
    axisY->setTickCount(10);
    axisY->setMinorTickCount(1);


    // Turn off grid lines
    axisX->setGridLineVisible(true);
    axisY->setGridLineVisible(false);
    axisY->setShadesPen(Qt::NoPen);
    axisY->setShadesBrush(QBrush(QColor(0x99, 0xcc, 0xcc, 0x55)));
    axisY->setShadesVisible(true);

    //Make the SQL Query for all days
    QSqlQuery query_HoursDays(QSqlDatabase::database("three"));
    QString queryHoursDaysTable;
    queryHoursDaysTable = "SELECT Days, SUM(NumHours) FROM Project  WHERE Team = '" + Team + "' and ProjectStatus = 'Completed' GROUP BY Days ORDER BY Days ASC";
    query_HoursDays.exec(queryHoursDaysTable);

    //Make the SQL Query for Last 7 Days
    QSqlQuery query_HoursDays7(QSqlDatabase::database("three"));
    QString queryHoursDaysTable7;
    queryHoursDaysTable7 = "SELECT Days, SUM(NumHours) FROM Project  WHERE Team = '" + Team + "' and ProjectStatus = 'Completed' GROUP BY Days ORDER BY Days DESC LIMIT 7";
    query_HoursDays7.exec(queryHoursDaysTable7);

    if(last7 == false){
        series->append(0,0);
        while(query_HoursDays.next()) {
            series->append(query_HoursDays.value(0).toInt(),query_HoursDays.value(1).toInt());
        }
    }
    else {
        while(query_HoursDays7.next()) {
            series ->append(query_HoursDays7.value(0).toInt(),query_HoursDays7.value(1).toInt());
        }
        axisX->setTickCount(7);
        axisY->setRange(0, 40);
        axisY->setTickCount(8);
    }

    QChart *chart = new QChart();
    chart->setTitle("Team " + Team + " Productivity");
    chart->legend()->hide();
    chart->addSeries(series);
    axisY->setTitleText("Project Hours Completed"); // yaxis
    axisX->setTitleText("Day # ");                  // xaxis

    chart->addAxis(axisX, Qt::AlignBottom);
    chart->addAxis(axisY, Qt::AlignLeft);
    series->attachAxis(axisX);
    series->attachAxis(axisY);

    chartView = new QChartView(chart);
    chartView->setRenderHint(QPainter::Antialiasing);
    chartView->setParent(ui->productivityLineGraph);
    chartView->show();
}

void WindowMain::on_lastWeek_toggled(bool checked)
{
    if (checked == true) {
        last7 = true;
        DrawChart( Team , checked);
        QMessageBox::information(this,"","Last Week");
    }
    else if (checked == false) {
        last7 = false;
        DrawChart( Team , checked);
        QMessageBox::information(this,"","All Time");
    }
}