[FEATURE] Import layers to data providers in background

Flips the drag and drop import of layers to browser data sources
to use task manager, allowing these slow imports to occur
in the background.
This commit is contained in:
Nyall Dawson 2017-05-07 09:48:57 +10:00
parent 384369c3c4
commit 64b612f2a8
12 changed files with 354 additions and 208 deletions

View File

@ -130,6 +130,69 @@ class QgsVectorLayerExporter : QgsFeatureSink
QgsVectorLayerExporter( const QgsVectorLayerExporter &rh );
};
class QgsVectorLayerExporterTask : QgsTask
{
%Docstring
QgsTask task which performs a QgsVectorLayerExporter layer export operation as a background
task. This can be used to export a vector layer out to a provider without blocking the
QGIS interface.
.. versionadded:: 3.0
.. seealso:: QgsVectorFileWriterTask
.. seealso:: QgsRasterFileWriterTask
%End
%TypeHeaderCode
#include "qgsvectorlayerexporter.h"
%End
public:
QgsVectorLayerExporterTask( QgsVectorLayer *layer,
const QString &uri,
const QString &providerKey,
const QgsCoordinateReferenceSystem &destinationCrs,
QMap<QString, QVariant> *options = 0 );
%Docstring
Constructor for QgsVectorLayerExporterTask. Takes a source ``layer``, destination ``uri``
and ``providerKey``, and various export related parameters such as destination CRS
and export ``options``.
%End
static QgsVectorLayerExporterTask *withLayerOwnership( QgsVectorLayer *layer /Transfer/,
const QString &uri,
const QString &providerKey,
const QgsCoordinateReferenceSystem &destinationCrs,
QMap<QString, QVariant> *options = 0 ) /Factory/;
%Docstring
Creates a new QgsVectorLayerExporterTask which has ownership over a source ``layer``.
When the export task has completed (successfully or otherwise) ``layer`` will be
deleted. The destination ``uri`` and ``providerKey``, and various export related parameters such as destination CRS
and export ``options`` must be specified.
:rtype: QgsVectorLayerExporterTask
%End
virtual void cancel();
signals:
void exportComplete();
%Docstring
Emitted when exporting the layer is successfully completed.
%End
void errorOccurred( int error, const QString &errorMessage );
%Docstring
Emitted when an error occurs which prevented the layer being exported (or if
the task is canceled). The export ``error`` and ``errorMessage`` will be reported.
%End
protected:
virtual bool run();
virtual void finished( bool result );
};
/************************************************************************
* This file has been generated automatically from *
* *

View File

@ -16,6 +16,7 @@ class QgsRasterFileWriterTask : QgsTask
task. This can be used to save a raster layer out to a file without blocking the
QGIS interface.
.. seealso:: QgsVectorFileWriterTask
.. seealso:: QgsVectorFileExporterTask
.. versionadded:: 3.0
%End

View File

@ -570,6 +570,7 @@ SET(QGIS_CORE_MOC_HDRS
qgsvectorlayereditbuffer.h
qgsvectorlayereditpassthrough.h
qgsvectorlayer.h
qgsvectorlayerexporter.h
qgsvectorlayerfeaturecounter.h
qgsvectorlayerjoinbuffer.h
qgsvectorlayertools.h

View File

@ -30,6 +30,7 @@
* task. This can be used to save a vector layer out to a file without blocking the
* QGIS interface.
* \since QGIS 3.0
* \see QgsVectorLayerExporterTask
* \see QgsRasterFileWriterTask
*/
class CORE_EXPORT QgsVectorFileWriterTask : public QgsTask

View File

@ -463,3 +463,60 @@ QgsVectorLayerExporter::exportLayer( QgsVectorLayer *layer,
return NoError;
}
//
// QgsVectorLayerExporterTask
//
QgsVectorLayerExporterTask::QgsVectorLayerExporterTask( QgsVectorLayer *layer, const QString &uri, const QString &providerKey, const QgsCoordinateReferenceSystem &destinationCrs, QMap<QString, QVariant> *options )
: QgsTask( tr( "Exporting %1" ).arg( layer->name() ), QgsTask::CanCancel )
, mLayer( layer )
, mDestUri( uri )
, mDestProviderKey( providerKey )
, mDestCrs( destinationCrs )
, mOptions( options ? * options : QMap<QString, QVariant>() )
, mOwnedFeedback( new QgsFeedback() )
{
if ( mLayer )
setDependentLayers( QList< QgsMapLayer * >() << mLayer );
}
QgsVectorLayerExporterTask *QgsVectorLayerExporterTask::withLayerOwnership( QgsVectorLayer *layer, const QString &uri, const QString &providerKey, const QgsCoordinateReferenceSystem &destinationCrs, QMap<QString, QVariant> *options )
{
std::unique_ptr< QgsVectorLayerExporterTask > newTask( new QgsVectorLayerExporterTask( layer, uri, providerKey, destinationCrs, options ) );
newTask->mOwnsLayer = true;
return newTask.release();
}
void QgsVectorLayerExporterTask::cancel()
{
mOwnedFeedback->cancel();
QgsTask::cancel();
}
bool QgsVectorLayerExporterTask::run()
{
if ( !mLayer )
return false;
connect( mOwnedFeedback.get(), &QgsFeedback::progressChanged, this, &QgsVectorLayerExporterTask::setProgress );
mError = QgsVectorLayerExporter::exportLayer(
mLayer.data(), mDestUri, mDestProviderKey, mDestCrs, false, &mErrorMessage, false,
&mOptions );
if ( mOwnsLayer )
delete mLayer;
return mError == QgsVectorLayerExporter::NoError;
}
void QgsVectorLayerExporterTask::finished( bool result )
{
if ( result )
emit exportComplete();
else
emit errorOccurred( mError, mErrorMessage );
}

View File

@ -22,10 +22,12 @@
#include "qgis.h"
#include "qgsfeature.h"
#include "qgsfeaturesink.h"
#include "qgstaskmanager.h"
#include "qgsfeedback.h"
#include "qgsvectorlayer.h"
class QProgressDialog;
class QgsVectorDataProvider;
class QgsVectorLayer;
class QgsFields;
/**
@ -168,4 +170,81 @@ class CORE_EXPORT QgsVectorLayerExporter : public QgsFeatureSink
};
/**
* \class QgsVectorLayerExporterTask
* \ingroup core
* QgsTask task which performs a QgsVectorLayerExporter layer export operation as a background
* task. This can be used to export a vector layer out to a provider without blocking the
* QGIS interface.
* \since QGIS 3.0
* \see QgsVectorFileWriterTask
* \see QgsRasterFileWriterTask
*/
class CORE_EXPORT QgsVectorLayerExporterTask : public QgsTask
{
Q_OBJECT
public:
/**
* Constructor for QgsVectorLayerExporterTask. Takes a source \a layer, destination \a uri
* and \a providerKey, and various export related parameters such as destination CRS
* and export \a options.
*/
QgsVectorLayerExporterTask( QgsVectorLayer *layer,
const QString &uri,
const QString &providerKey,
const QgsCoordinateReferenceSystem &destinationCrs,
QMap<QString, QVariant> *options = nullptr );
/**
* Creates a new QgsVectorLayerExporterTask which has ownership over a source \a layer.
* When the export task has completed (successfully or otherwise) \a layer will be
* deleted. The destination \a uri and \a providerKey, and various export related parameters such as destination CRS
* and export \a options must be specified.
*/
static QgsVectorLayerExporterTask *withLayerOwnership( QgsVectorLayer *layer SIP_TRANSFER,
const QString &uri,
const QString &providerKey,
const QgsCoordinateReferenceSystem &destinationCrs,
QMap<QString, QVariant> *options = nullptr ) SIP_FACTORY;
virtual void cancel() override;
signals:
/**
* Emitted when exporting the layer is successfully completed.
*/
void exportComplete();
/**
* Emitted when an error occurs which prevented the layer being exported (or if
* the task is canceled). The export \a error and \a errorMessage will be reported.
*/
void errorOccurred( int error, const QString &errorMessage );
protected:
virtual bool run() override;
virtual void finished( bool result ) override;
private:
QPointer< QgsVectorLayer > mLayer = nullptr;
bool mOwnsLayer = false;
QString mDestUri;
QString mDestProviderKey;
QgsCoordinateReferenceSystem mDestCrs;
QMap<QString, QVariant> mOptions;
std::unique_ptr< QgsFeedback > mOwnedFeedback;
QgsVectorLayerExporter::ExportError mError = QgsVectorLayerExporter::NoError;
QString mErrorMessage;
};
#endif // QGSVECTORLAYEREXPORTER_H

View File

@ -31,6 +31,7 @@
* task. This can be used to save a raster layer out to a file without blocking the
* QGIS interface.
* \see QgsVectorFileWriterTask
* \see QgsVectorFileExporterTask
* \since QGIS 3.0
*/
class CORE_EXPORT QgsRasterFileWriterTask : public QgsTask

View File

@ -24,6 +24,7 @@
#include "qgsvectorlayerexporter.h"
#include "qgsvectorlayer.h"
#include "qgssettings.h"
#include "qgsmessageoutput.h"
#include <QMessageBox>
#include <QProgressDialog>
@ -316,16 +317,8 @@ bool QgsDb2ConnectionItem::handleDrop( const QMimeData *data, const QString &toS
return false;
// TODO: probably should show a GUI with settings etc
qApp->setOverrideCursor( Qt::WaitCursor );
QProgressDialog *progress = new QProgressDialog( tr( "Copying features..." ), tr( "Abort" ), 0, 0, nullptr );
progress->setWindowTitle( tr( "Import layer" ) );
progress->setWindowModality( Qt::WindowModal );
progress->show();
QStringList importResults;
bool hasError = false;
bool canceled = false;
QgsMimeDataUtils::UriList lst = QgsMimeDataUtils::decodeUriList( data );
Q_FOREACH ( const QgsMimeDataUtils::Uri &u, lst )
@ -357,60 +350,51 @@ bool QgsDb2ConnectionItem::handleDrop( const QMimeData *data, const QString &toS
if ( srcLayer->geometryType() != QgsWkbTypes::NullGeometry )
uri += QLatin1String( " (geom)" );
QgsVectorLayerExporter::ExportError err;
QString importError;
err = QgsVectorLayerExporter::exportLayer( srcLayer, uri, QStringLiteral( "DB2" ), srcLayer->crs(), false, &importError, false, nullptr, progress );
if ( err == QgsVectorLayerExporter::NoError )
std::unique_ptr< QgsVectorLayerExporterTask > exportTask( QgsVectorLayerExporterTask::withLayerOwnership( srcLayer, uri, QStringLiteral( "DB2" ), srcLayer->crs() ) );
// when export is successful:
connect( exportTask.get(), &QgsVectorLayerExporterTask::exportComplete, this, [ = ]()
{
importResults.append( tr( "%1: OK!" ).arg( u.name ) );
QgsDebugMsg( "import successful" );
}
else
{
if ( err == QgsVectorLayerExporter::ErrUserCanceled )
{
canceled = true;
QgsDebugMsg( "import canceled" );
}
// this is gross - TODO - find a way to get access to messageBar from data items
QMessageBox::information( nullptr, tr( "Import to DB2 database" ), tr( "Import was successful." ) );
if ( state() == Populated )
refresh();
else
populate();
} );
// when an error occurs:
connect( exportTask.get(), &QgsVectorLayerExporterTask::errorOccurred, this, [ = ]( int error, const QString & errorMessage )
{
if ( error != QgsVectorLayerExporter::ErrUserCanceled )
{
QString errMsg = QStringLiteral( "%1: %2" ).arg( u.name, importError );
QgsDebugMsg( "import failed: " + errMsg );
importResults.append( errMsg );
hasError = true;
QgsMessageOutput *output = QgsMessageOutput::createMessageOutput();
output->setTitle( tr( "Import to DB2 database" ) );
output->setMessage( tr( "Failed to import some layers!\n\n" ) + errorMessage, QgsMessageOutput::MessageText );
output->showMessage();
}
}
if ( state() == Populated )
refresh();
else
populate();
} );
QgsApplication::taskManager()->addTask( exportTask.release() );
}
else
{
importResults.append( tr( "%1: OK!" ).arg( u.name ) );
importResults.append( tr( "%1: Not a valid layer!" ).arg( u.name ) );
hasError = true;
}
delete srcLayer;
}
delete progress;
qApp->restoreOverrideCursor();
if ( canceled )
if ( hasError )
{
QMessageBox::information( nullptr, tr( "Import to DB2 database" ), tr( "Import canceled." ) );
refresh();
QgsMessageOutput *output = QgsMessageOutput::createMessageOutput();
output->setTitle( tr( "Import to DB2 database" ) );
output->setMessage( tr( "Failed to import some layers!\n\n" ) + importResults.join( QStringLiteral( "\n" ) ), QgsMessageOutput::MessageText );
output->showMessage();
}
else if ( hasError )
{
QMessageBox::warning( nullptr, tr( "Import to DB2 database" ), tr( "Failed to import some layers!\n\n" ) + importResults.join( QStringLiteral( "\n" ) ) );
}
else
{
QMessageBox::information( nullptr, tr( "Import to DB2 database" ), tr( "Import was successful." ) );
}
if ( state() == Populated )
refresh();
else
populate();
return true;
}

View File

@ -26,6 +26,7 @@
#include "qgsdatasourceuri.h"
#include "qgsmssqlprovider.h"
#include "qgssettings.h"
#include "qgsmessageoutput.h"
#include <QMessageBox>
#include <QSqlDatabase>
@ -383,16 +384,8 @@ bool QgsMssqlConnectionItem::handleDrop( const QMimeData *data, const QString &t
return false;
// TODO: probably should show a GUI with settings etc
qApp->setOverrideCursor( Qt::WaitCursor );
QProgressDialog *progress = new QProgressDialog( tr( "Copying features..." ), tr( "Abort" ), 0, 0, nullptr );
progress->setWindowTitle( tr( "Import layer" ) );
progress->setWindowModality( Qt::WindowModal );
progress->show();
QStringList importResults;
bool hasError = false;
bool canceled = false;
QgsMimeDataUtils::UriList lst = QgsMimeDataUtils::decodeUriList( data );
Q_FOREACH ( const QgsMimeDataUtils::Uri &u, lst )
@ -423,49 +416,49 @@ bool QgsMssqlConnectionItem::handleDrop( const QMimeData *data, const QString &t
if ( srcLayer->geometryType() != QgsWkbTypes::NullGeometry )
uri += QLatin1String( " (geom)" );
QgsVectorLayerExporter::ExportError err;
QString importError;
err = QgsVectorLayerExporter::exportLayer( srcLayer, uri, QStringLiteral( "mssql" ), srcLayer->crs(), false, &importError, false, nullptr, progress );
if ( err == QgsVectorLayerExporter::NoError )
importResults.append( tr( "%1: OK!" ).arg( u.name ) );
else if ( err == QgsVectorLayerExporter::ErrUserCanceled )
canceled = true;
else
std::unique_ptr< QgsVectorLayerExporterTask > exportTask( QgsVectorLayerExporterTask::withLayerOwnership( srcLayer, uri, QStringLiteral( "mssql" ), srcLayer->crs() ) );
// when export is successful:
connect( exportTask.get(), &QgsVectorLayerExporterTask::exportComplete, this, [ = ]()
{
importResults.append( QStringLiteral( "%1: %2" ).arg( u.name, importError ) );
hasError = true;
}
// this is gross - TODO - find a way to get access to messageBar from data items
QMessageBox::information( nullptr, tr( "Import to MSSQL database" ), tr( "Import was successful." ) );
if ( state() == Populated )
refresh();
else
populate();
} );
// when an error occurs:
connect( exportTask.get(), &QgsVectorLayerExporterTask::errorOccurred, this, [ = ]( int error, const QString & errorMessage )
{
if ( error != QgsVectorLayerExporter::ErrUserCanceled )
{
QgsMessageOutput *output = QgsMessageOutput::createMessageOutput();
output->setTitle( tr( "Import to MSSQL database" ) );
output->setMessage( tr( "Failed to import some layers!\n\n" ) + errorMessage, QgsMessageOutput::MessageText );
output->showMessage();
}
if ( state() == Populated )
refresh();
else
populate();
} );
}
else
{
importResults.append( tr( "%1: OK!" ).arg( u.name ) );
importResults.append( tr( "%1: Not a valid layer!" ).arg( u.name ) );
hasError = true;
}
delete srcLayer;
}
delete progress;
qApp->restoreOverrideCursor();
if ( canceled )
if ( hasError )
{
QMessageBox::information( nullptr, tr( "Import to MSSQL database" ), tr( "Import canceled." ) );
refresh();
QgsMessageOutput *output = QgsMessageOutput::createMessageOutput();
output->setTitle( tr( "Import to MSSQL database" ) );
output->setMessage( tr( "Failed to import some layers!\n\n" ) + importResults.join( QStringLiteral( "\n" ) ), QgsMessageOutput::MessageText );
output->showMessage();
}
else if ( hasError )
{
QMessageBox::warning( nullptr, tr( "Import to MSSQL database" ), tr( "Failed to import some layers!\n\n" ) + importResults.join( QStringLiteral( "\n" ) ) );
}
else
{
QMessageBox::information( nullptr, tr( "Import to MSSQL database" ), tr( "Import was successful." ) );
}
if ( state() == Populated )
refresh();
else
populate();
return true;
}

View File

@ -230,16 +230,8 @@ bool QgsOracleConnectionItem::handleDrop( const QMimeData *data, Qt::DropAction
// TODO: probably should show a GUI with settings etc
QgsDataSourceUri uri = QgsOracleConn::connUri( mName );
qApp->setOverrideCursor( Qt::WaitCursor );
QProgressDialog *progress = new QProgressDialog( tr( "Copying features..." ), tr( "Abort" ), 0, 0, nullptr );
progress->setWindowTitle( tr( "Import layer" ) );
progress->setWindowModality( Qt::WindowModal );
progress->show();
QStringList importResults;
bool hasError = false;
bool canceled = false;
QgsMimeDataUtils::UriList lst = QgsMimeDataUtils::decodeUriList( data );
Q_FOREACH ( const QgsMimeDataUtils::Uri &u, lst )
@ -264,54 +256,52 @@ bool QgsOracleConnectionItem::handleDrop( const QMimeData *data, Qt::DropAction
uri.setSrid( authid.mid( 5 ) );
}
QgsDebugMsgLevel( "URI " + uri.uri(), 3 );
QgsVectorLayerExporter::ExportError err;
QString importError;
err = QgsVectorLayerExporter::exportLayer( srcLayer, uri.uri(), "oracle", srcLayer->crs(), false, &importError, false, nullptr, progress );
if ( err == QgsVectorLayerExporter::NoError )
importResults.append( tr( "%1: OK!" ).arg( u.name ) );
else if ( err == QgsVectorLayerExporter::ErrUserCanceled )
canceled = true;
else
std::unique_ptr< QgsVectorLayerExporterTask > exportTask( QgsVectorLayerExporterTask::withLayerOwnership( srcLayer, uri.uri(), QStringLiteral( "oracle" ), srcLayer->crs() ) );
// when export is successful:
connect( exportTask.get(), &QgsVectorLayerExporterTask::exportComplete, this, [ = ]()
{
importResults.append( QString( "%1: %2" ).arg( u.name ).arg( importError ) );
hasError = true;
}
// this is gross - TODO - find a way to get access to messageBar from data items
QMessageBox::information( nullptr, tr( "Import to Oracle database" ), tr( "Import was successful." ) );
if ( state() == Populated )
refresh();
else
populate();
} );
// when an error occurs:
connect( exportTask.get(), &QgsVectorLayerExporterTask::errorOccurred, this, [ = ]( int error, const QString & errorMessage )
{
if ( error != QgsVectorLayerExporter::ErrUserCanceled )
{
QgsMessageOutput *output = QgsMessageOutput::createMessageOutput();
output->setTitle( tr( "Import to Oracle database" ) );
output->setMessage( tr( "Failed to import some layers!\n\n" ) + errorMessage, QgsMessageOutput::MessageText );
output->showMessage();
}
if ( state() == Populated )
refresh();
else
populate();
} );
QgsApplication::taskManager()->addTask( exportTask.release() );
}
else
{
importResults.append( tr( "%1: OK!" ).arg( u.name ) );
importResults.append( tr( "%1: Not a valid layer!" ).arg( u.name ) );
hasError = true;
}
delete srcLayer;
}
delete progress;
qApp->restoreOverrideCursor();
if ( canceled )
{
QMessageBox::information( nullptr, tr( "Import to Oracle database" ), tr( "Import canceled." ) );
refresh();
}
else if ( hasError )
if ( hasError )
{
QgsMessageOutput *output = QgsMessageOutput::createMessageOutput();
output->setTitle( tr( "Import to Oracle database" ) );
output->setMessage( tr( "Failed to import some layers!\n\n" ) + importResults.join( "\n" ), QgsMessageOutput::MessageText );
output->showMessage();
}
else
{
QMessageBox::information( 0, tr( "Import to Oracle database" ), tr( "Import was successful." ) );
refresh();
}
if ( state() == Populated )
refresh();
else
populate();
return true;
}

View File

@ -194,16 +194,8 @@ bool QgsPGConnectionItem::handleDrop( const QMimeData *data, const QString &toSc
// TODO: probably should show a GUI with settings etc
QgsDataSourceUri uri = QgsPostgresConn::connUri( mName );
qApp->setOverrideCursor( Qt::WaitCursor );
QProgressDialog *progress = new QProgressDialog( tr( "Copying features..." ), tr( "Abort" ), 0, 0, nullptr );
progress->setWindowTitle( tr( "Import layer" ) );
progress->setWindowModality( Qt::WindowModal );
progress->show();
QStringList importResults;
bool hasError = false;
bool canceled = false;
QgsMimeDataUtils::UriList lst = QgsMimeDataUtils::decodeUriList( data );
Q_FOREACH ( const QgsMimeDataUtils::Uri &u, lst )
@ -228,49 +220,45 @@ bool QgsPGConnectionItem::handleDrop( const QMimeData *data, const QString &toSc
uri.setSchema( toSchema );
}
QgsVectorLayerExporter::ExportError err;
QString importError;
err = QgsVectorLayerExporter::exportLayer( srcLayer, uri.uri( false ), QStringLiteral( "postgres" ), srcLayer->crs(), false, &importError, false, nullptr, progress );
if ( err == QgsVectorLayerExporter::NoError )
importResults.append( tr( "%1: OK!" ).arg( u.name ) );
else if ( err == QgsVectorLayerExporter::ErrUserCanceled )
canceled = true;
else
std::unique_ptr< QgsVectorLayerExporterTask > exportTask( QgsVectorLayerExporterTask::withLayerOwnership( srcLayer, uri.uri( false ), QStringLiteral( "postgres" ), srcLayer->crs() ) );
// when export is successful:
connect( exportTask.get(), &QgsVectorLayerExporterTask::exportComplete, this, [ = ]()
{
importResults.append( QStringLiteral( "%1: %2" ).arg( u.name, importError ) );
hasError = true;
}
// this is gross - TODO - find a way to get access to messageBar from data items
QMessageBox::information( nullptr, tr( "Import to PostGIS database" ), tr( "Import was successful." ) );
refresh();
} );
// when an error occurs:
connect( exportTask.get(), &QgsVectorLayerExporterTask::errorOccurred, this, [ = ]( int error, const QString & errorMessage )
{
if ( error != QgsVectorLayerExporter::ErrUserCanceled )
{
QgsMessageOutput *output = QgsMessageOutput::createMessageOutput();
output->setTitle( tr( "Import to PostGIS database" ) );
output->setMessage( tr( "Failed to import some layers!\n\n" ) + errorMessage, QgsMessageOutput::MessageText );
output->showMessage();
}
refresh();
} );
QgsApplication::taskManager()->addTask( exportTask.release() );
}
else
{
importResults.append( tr( "%1: OK!" ).arg( u.name ) );
importResults.append( tr( "%1: Not a valid layer!" ).arg( u.name ) );
hasError = true;
}
delete srcLayer;
}
delete progress;
qApp->restoreOverrideCursor();
if ( canceled )
{
QMessageBox::information( nullptr, tr( "Import to PostGIS database" ), tr( "Import canceled." ) );
refresh();
}
else if ( hasError )
if ( hasError )
{
QgsMessageOutput *output = QgsMessageOutput::createMessageOutput();
output->setTitle( tr( "Import to PostGIS database" ) );
output->setMessage( tr( "Failed to import some layers!\n\n" ) + importResults.join( QStringLiteral( "\n" ) ), QgsMessageOutput::MessageText );
output->showMessage();
}
else
{
QMessageBox::information( nullptr, tr( "Import to PostGIS database" ), tr( "Import was successful." ) );
refresh();
}
return true;
}

View File

@ -27,7 +27,6 @@
#include <QAction>
#include <QMessageBox>
#include <QProgressDialog>
QGISEXTERN bool deleteLayer( const QString &dbPath, const QString &tableName, QString &errCause );
@ -199,16 +198,8 @@ bool QgsSLConnectionItem::handleDrop( const QMimeData *data, Qt::DropAction )
QgsDataSourceUri destUri;
destUri.setDatabase( mDbPath );
qApp->setOverrideCursor( Qt::WaitCursor );
QProgressDialog *progress = new QProgressDialog( tr( "Copying features..." ), tr( "Abort" ), 0, 0, nullptr );
progress->setWindowTitle( tr( "Import layer" ) );
progress->setWindowModality( Qt::WindowModal );
progress->show();
QStringList importResults;
bool hasError = false;
bool canceled = false;
QgsMimeDataUtils::UriList lst = QgsMimeDataUtils::decodeUriList( data );
Q_FOREACH ( const QgsMimeDataUtils::Uri &u, lst )
@ -227,49 +218,46 @@ bool QgsSLConnectionItem::handleDrop( const QMimeData *data, Qt::DropAction )
{
destUri.setDataSource( QString(), u.name, srcLayer->geometryType() != QgsWkbTypes::NullGeometry ? QStringLiteral( "geom" ) : QString() );
QgsDebugMsg( "URI " + destUri.uri() );
QgsVectorLayerExporter::ExportError err;
QString importError;
err = QgsVectorLayerExporter::exportLayer( srcLayer, destUri.uri(), QStringLiteral( "spatialite" ), srcLayer->crs(), false, &importError, false, nullptr, progress );
if ( err == QgsVectorLayerExporter::NoError )
importResults.append( tr( "%1: OK!" ).arg( u.name ) );
else if ( err == QgsVectorLayerExporter::ErrUserCanceled )
canceled = true;
else
std::unique_ptr< QgsVectorLayerExporterTask > exportTask( QgsVectorLayerExporterTask::withLayerOwnership( srcLayer, destUri.uri(), QStringLiteral( "spatialite" ), srcLayer->crs() ) );
// when export is successful:
connect( exportTask.get(), &QgsVectorLayerExporterTask::exportComplete, this, [ = ]()
{
importResults.append( QStringLiteral( "%1: %2" ).arg( u.name, importError ) );
hasError = true;
}
// this is gross - TODO - find a way to get access to messageBar from data items
QMessageBox::information( nullptr, tr( "Import to SpatiaLite database" ), tr( "Import was successful." ) );
refresh();
} );
// when an error occurs:
connect( exportTask.get(), &QgsVectorLayerExporterTask::errorOccurred, this, [ = ]( int error, const QString & errorMessage )
{
if ( error != QgsVectorLayerExporter::ErrUserCanceled )
{
QgsMessageOutput *output = QgsMessageOutput::createMessageOutput();
output->setTitle( tr( "Import to SpatiaLite database" ) );
output->setMessage( tr( "Failed to import layer!\n\n" ) + errorMessage, QgsMessageOutput::MessageText );
output->showMessage();
}
refresh();
} );
QgsApplication::taskManager()->addTask( exportTask.release() );
}
else
{
importResults.append( tr( "%1: OK!" ).arg( u.name ) );
importResults.append( tr( "%1: Not a valid layer!" ).arg( u.name ) );
hasError = true;
}
delete srcLayer;
}
delete progress;
qApp->restoreOverrideCursor();
if ( canceled )
{
QMessageBox::information( nullptr, tr( "Import to SpatiaLite database" ), tr( "Import canceled." ) );
refresh();
}
else if ( hasError )
if ( hasError )
{
QgsMessageOutput *output = QgsMessageOutput::createMessageOutput();
output->setTitle( tr( "Import to SpatiaLite database" ) );
output->setMessage( tr( "Failed to import some layers!\n\n" ) + importResults.join( QStringLiteral( "\n" ) ), QgsMessageOutput::MessageText );
output->showMessage();
}
else
{
QMessageBox::information( nullptr, tr( "Import to SpatiaLite database" ), tr( "Import was successful." ) );
refresh();
}
return true;
}