Russian Qt Forum

#ifndef MAINWINDOW_H
#define MAINWINDOW_H
 
#include <QMainWindow>
#include<QtSql/QSqlDatabase>
#include<QtSql/QSqlQuery>
#include<QTreeWidgetItem>
#include<QSqlTableModel>
 
 
 
namespace Ui {
class MainWindow;
}
 
class MainWindow : public QMainWindow
{
    Q_OBJECT
 
public:
    explicit MainWindow(QWidget *parent = 0);
    ~MainWindow();
    QStringList columnList;
    int columnCount;
 
private:
    QSqlDatabase db;
    Ui::MainWindow *ui;
    QString dbName;
    QString tableName;
    QString fieldName;
    QString typeName;
    QTreeWidgetItem *itm;
    QSqlTableModel *model;
    QString conName;
    void showButtons(int);
 
private slots:
    void newDB();
    void openDB();
    void newTable();
    void openTable();
    void addRecord();
    void addField();
    void deleteRecord();
    void deleteFieldname();
    void removeDB();
 
};
 
#endif // MAINWINDOW_H

#include "mainwindow.h"
#include "ui_mainwindow.h"
#include<dbdialog.h>
#include<QFileDialog>
#include<QDebug>
#include<tabledialog.h>
#include<delegate.cpp>
#include<QSqlRecord>
#include<fielddialog.h>
#include<deletefield.h>
 
 
 
 
MainWindow::MainWindow(QWidget *parent) :
    QMainWindow(parent),
    ui(new Ui::MainWindow)
{
    ui->setupUi(this);
 
    itm=new QTreeWidgetItem;
 
    conName="new123";
 
    connect(ui->newDBaction,SIGNAL(triggered()),this,SLOT(newDB()));
    connect(ui->openAction,SIGNAL(triggered()),this,SLOT(openDB()));
    connect(ui->newTableacrion,SIGNAL(triggered()),this,SLOT(newTable()));
    connect(ui->openTableAction,SIGNAL(triggered()),this,SLOT(openTable()));
    connect(ui->addRecordaction,SIGNAL(triggered()),this,SLOT(addRecord()));
    connect(ui->addFieldAction,SIGNAL(triggered()),this,SLOT(addField()));
    connect(ui->deleteRecordaction,SIGNAL(triggered()),this,SLOT(deleteRecord()));
    connect(ui->deleteFieldAction,SIGNAL(triggered()),this,SLOT(deleteFieldname()));
    connect(ui->removeDBname,SIGNAL(triggered()),this,SLOT(removeDB()));
 
    NonEditTableColumnDelegate *delegate=new NonEditTableColumnDelegate(this);
    DateEditTableColumnDelegate *delegate1=new DateEditTableColumnDelegate(ui->tableView);
    ui->tableView->setItemDelegateForColumn(0,delegate);
    ui->tableView->setItemDelegateForColumn(1,delegate1);
    ui->tableView->setSortingEnabled(true);
 
 
 
 
}
 
MainWindow::~MainWindow()
{
    delete ui;
}
 
void MainWindow::newDB()
{
 
    ui->treeWidget->clear();
    dbdialog dialog;
    dialog.setModal(true);
    dialog.setWindowTitle("Создать новую БД");
    dialog.exec();
    dbName=dialog.name;
    db=QSqlDatabase::addDatabase("QSQLITE",conName);
    db.setDatabaseName(dbName+".sqlite");
    if(db.open())
    {
    qDebug()<<"Database open "+dbName;
    itm->setText(0,dbName);
    ui->treeWidget->addTopLevelItem(itm);
    }
    QSqlQuery query("CREATE TABLE "+dbName+"(ID integer PRIMARY KEY NOT NULL);");
    query.exec();
 
}
 
void MainWindow::openDB()
{
    ui->treeWidget->clear();
    QString fileName = QFileDialog::getOpenFileName(this,
            tr("Открыть БД"), "",
            tr("DB (*.sqlite);;All Files (*)"));
    dbName=fileName;
    db=QSqlDatabase::addDatabase("QSQLITE",conName);
    db.setDatabaseName(dbName);
 
    if(db.open())
    {
        QStringList list=db.tables();
        qDebug()<<"Database open "+dbName;
        itm->setText(0,list[0]);
        ui->treeWidget->addTopLevelItem(itm);
        int j=db.tables().count();
        for(int i=1;i<=j-1;i++)
        {
            QTreeWidgetItem*itm1=new QTreeWidgetItem;
            itm1->setText(0,list.at(i));
            itm->addChild(itm1);
        }
    }
 
}
 
void MainWindow::newTable()
{
    tabledialog dialog;
    dialog.setModal(true);
    dialog.setWindowTitle("Создание таблицы");
    dialog.exec();
    QSqlQuery query;
    tableName=dialog.name;
    qDebug()<<"tableName"<<tableName;
    if(query.exec("CREATE TABLE "+tableName+" (ID integer PRIMARY KEY NOT NULL);"))
    {
        QTreeWidgetItem *itm0=new QTreeWidgetItem;
        itm0->setText(0,tableName);
        itm->addChild(itm0);
    }
    qDebug()<<db.tables();
 
 
}
 
void MainWindow::openTable()
{
    qDebug()<<"openTable";
    tableName=ui->treeWidget->currentItem()->text(0);
    model = new QSqlTableModel(this);
    model->setTable(tableName);
    model->setEditStrategy(QSqlTableModel::OnFieldChange);
    model->select();
    ui->tableView->setModel(model);
    ui->tableView->resizeColumnsToContents();
 
}
 
void MainWindow::addRecord()
{
    QSqlQuery query;
    QSqlRecord recordCount;
    QSqlRecord rec;
    query.exec("select count(*) as pc_count from "+tableName);
    query.next();
    recordCount=query.record();
    int cSize=query.value(recordCount.indexOf("pc_count")).toInt();
    QModelIndex index = model->index(cSize-1, 0);
    QString text = index.data(Qt::DisplayRole).toString();
    QString str=QString::number(cSize);
    int i=text.toInt();
    i++;
    str=QString::number(i);
    rec=model->record();
    rec.setValue("id",str);
    model->insertRecord(-1,rec);
}
 
void MainWindow::addField()
{
    fielddialog dialog;
    dialog.setModal(true);
    dialog.setWindowTitle("Создание нового поля");
    dialog.exec();
    QSqlQuery query;
    QString fieldName;
    QString typeName;
    fieldName=dialog.fieldName;
    typeName=dialog.fieldType;
    query.exec("alter table "+tableName+" add "+fieldName+" "+typeName);
    qDebug()<<fieldName;
    qDebug()<<typeName;
    model->setTable(tableName);
    model->select();
    ui->tableView->setModel(model);
 
}
 
void MainWindow::deleteRecord()
{
    QSqlQuery query;
    QString str;
    int id;
    QModelIndexList indexes =   ui->tableView->selectionModel()->selection().indexes();
    foreach(QModelIndex index, indexes) {
            QSqlRecord record = model->record(index.row());
            id = record.value("ID").toInt();
            qDebug()<<id;
        }
    str=QString::number(id);
    query.exec("delete from "+tableName+" where ID="+str);
    model->setTable(tableName);
    model->select();
    ui->tableView->setModel(model);
}
 
void MainWindow::deleteFieldname()
{
    QSqlRecord record1;
    record1=db.record(tableName);
    int i=0;
    QString x=record1.fieldName(i);
    columnList.insert(i,x);
    i++;
    while(x!="")
    {
        x=record1.fieldName(i);
        columnList.insert(i,x);
        if(x=="")
            break;
        i++;
    }
    columnCount=i;
    columnList.removeLast();
    QSqlQuery query;
    deleteField dialog(this,columnList,columnCount);
    dialog.setWindowTitle("Удалить поле");
    dialog.setModal(true);
    dialog.exec();
    qDebug()<<dialog.columnName;
    QString first="create table newtable5 as select ";
    for(int i=0;i<columnCount;i++)
    {
        if(columnList[i]!=dialog.columnName)
        first=first+columnList[i]+",";
    }
    first.remove(first.length()-1,1);
    first=first+" from "+tableName+";";
    query.exec(first);
    query.exec("DROP TABLE IF EXISTS "+tableName+";");
    query.exec("ALTER TABLE newtable5 RENAME TO "+tableName+";");
    model->setTable(tableName);
    model->select();
    ui->tableView->setModel(model);
}
 
void MainWindow::removeDB()
{
    if(db.isOpen()==true)
    {
        ui->treeWidget->clear();
 
        db.close();
        db.removeDatabase(db.connectionName());
        QSqlDatabase::removeDatabase(db.connectionName());
    }
}

conName="new123";
 ...
    db=QSqlDatabase::addDatabase("QSQLITE",conName);