qgsdatabasetablecombobox.h

Go to the documentation of this file.

/***************************************************************************
   qgsdatabasetablecombobox.h
    --------------------------------
   Date                 : March 2020
   Copyright            : (C) 2020 Nyall Dawson
   Email                : nyall dot dawson at gmail dot com
***************************************************************************
*                                                                         *
*   This program is free software; you can redistribute it and/or modify  *
*   it under the terms of the GNU General Public License as published by  *
*   the Free Software Foundation; either version 2 of the License, or     *
*   (at your option) any later version.                                   *
*                                                                         *
***************************************************************************/
 
#ifndef QGSDATABASETABLECOMBOBOX_H
#define QGSDATABASETABLECOMBOBOX_H
 
#include "qgis_gui.h"
#include "qgis_sip.h"
 
#include <QComboBox>
#include <QSortFilterProxyModel>
 
class QgsDatabaseTableModel;
class QgsAbstractDatabaseProviderConnection;

#ifndef SIP_RUN
class GUI_EXPORT QgsDatabaseTableComboBoxSortModel : public QSortFilterProxyModel
{
    Q_OBJECT
  public:
    explicit QgsDatabaseTableComboBoxSortModel( QObject *parent = nullptr );
 
  protected:
    bool lessThan( const QModelIndex &source_left, const QModelIndex &source_right ) const override;
};
#endif

class GUI_EXPORT QgsDatabaseTableComboBox : public QWidget
{
    Q_OBJECT
 
  public:
    explicit QgsDatabaseTableComboBox( const QString &provider, const QString &connection, const QString &schema = QString(), QWidget *parent SIP_TRANSFERTHIS = nullptr );

    explicit QgsDatabaseTableComboBox( QgsAbstractDatabaseProviderConnection *connection SIP_TRANSFER, const QString &schema = QString(), QWidget *parent SIP_TRANSFERTHIS = nullptr );

    void setAllowEmptyTable( bool allowEmpty );

    bool allowEmptyTable() const;

    QString currentTable() const;

    QString currentSchema() const;

    QComboBox *comboBox() { return mComboBox; }
 
  public slots:

    void setTable( const QString &table, const QString &schema = QString() );

    void setConnectionName( const QString &connection, const QString &provider = QString() );

    void setSchema( const QString &schema );

    void refreshTables();
 
  signals:
    void tableChanged( const QString &table, const QString &schema = QString() );
 
  private slots:
    void indexChanged( int i );
    void rowsChanged();
 
  private:
    void init();
 
    bool mAllowEmpty = false;
    QString mProvider;
    QString mConnection;
    QString mSchema;
    QgsDatabaseTableModel *mModel = nullptr;
    QSortFilterProxyModel *mSortModel = nullptr;
    QComboBox *mComboBox = nullptr;
};
 
#endif // QGSDATABASETABLECOMBOBOX_H