sharpetronics/QGIS
1
0
Fork 0
mirror of https://github.com/qgis/QGIS.git synced 2025-03-29 00:05:09 -04:00
Code Issues Packages Projects Releases Wiki Activity

base path handler

Browse Source
This commit is contained in:
Alex 2019-03-20 20:01:50 -04:00 committed by GitHub
parent 6f92e7cf4b
commit b03d61d5d6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 573 additions and 540 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

954
src/app/qgshandlebadlayers.cpp
View File

@ -1,461 +1,493 @@
/***************************************************************************
qgshandlebadlayers.cpp - description
-------------------
begin : Sat 5 Mar 2011
copyright : (C) 2011 by Juergen E. Fischer, norBIT GmbH
email : jef at norbit dot de
***************************************************************************/
/***************************************************************************
* *
* 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. *
* *
***************************************************************************/
#include "qgshandlebadlayers.h"
#include "qgisapp.h"
#include "qgsauthconfigselect.h"
#include "qgsdataprovider.h"
#include "qgsguiutils.h"
#include "qgsdatasourceuri.h"
#include "qgslogger.h"
#include "qgsrasterlayer.h"
#include "qgsproviderregistry.h"
#include "qgsmessagebar.h"
#include "qgssettings.h"
#include "qgslayertreeregistrybridge.h"
#include "qgsapplication.h"
#include <QDomDocument>
#include <QDomElement>
#include <QFileDialog>
#include <QPushButton>
#include <QToolButton>
#include <QMessageBox>
#include <QDialogButtonBox>
#include <QUrl>
void QgsHandleBadLayersHandler::handleBadLayers( const QList<QDomNode> &layers )
{
QApplication::setOverrideCursor( Qt::ArrowCursor );
QgsHandleBadLayers *dialog = new QgsHandleBadLayers( layers );
dialog->buttonBox->button( QDialogButtonBox::Ignore )->setToolTip( tr( "Import all unavailable layers unmodified (you can fix them later)." ) );
dialog->buttonBox->button( QDialogButtonBox::Ignore )->setText( tr( "Keep Unavailable Layers" ) );
dialog->buttonBox->button( QDialogButtonBox::Discard )->setToolTip( tr( "Remove all unavailable layers from the project" ) );
dialog->buttonBox->button( QDialogButtonBox::Discard )->setText( tr( "Remove Unavailable Layers" ) );
dialog->buttonBox->button( QDialogButtonBox::Discard )->setIcon( QgsApplication::getThemeIcon( QStringLiteral( "/mActionDeleteSelected.svg" ) ) );
if ( dialog->layerCount() < layers.size() )
QgisApp::instance()->messageBar()->pushMessage(
tr( "Handle unavailable layers" ),
tr( "%1 of %2 unavailable layers were not fixable." )
.arg( layers.size() - dialog->layerCount() )
.arg( layers.size() ),
Qgis::Warning, QgisApp::instance()->messageTimeout() );
if ( dialog->layerCount() > 0 )
{
if ( dialog->exec() == dialog->Accepted )
{
emit layersChanged();
}
}
delete dialog;
QApplication::restoreOverrideCursor();
}
QgsHandleBadLayers::QgsHandleBadLayers( const QList<QDomNode> &layers )
: QDialog( QgisApp::instance() )
, mLayers( layers )
{
setupUi( this );
mVectorFileFilter = QgsProviderRegistry::instance()->fileVectorFilters();
mRasterFileFilter = QgsProviderRegistry::instance()->fileRasterFilters();
mBrowseButton = new QPushButton( tr( "Browse" ) );
buttonBox->addButton( mBrowseButton, QDialogButtonBox::ActionRole );
mBrowseButton->setDisabled( true );
mApplyButton = new QPushButton( tr( "Apply Changes" ) );
mApplyButton->setToolTip( tr( "Apply fixes to unavailable layers (remaining unavailable layers will be removed from the project)." ) );
buttonBox->addButton( mApplyButton, QDialogButtonBox::ActionRole );
connect( mLayerList, &QTableWidget::itemSelectionChanged, this, &QgsHandleBadLayers::selectionChanged );
connect( mBrowseButton, &QAbstractButton::clicked, this, &QgsHandleBadLayers::browseClicked );
connect( mApplyButton, &QAbstractButton::clicked, this, &QgsHandleBadLayers::apply );
connect( buttonBox->button( QDialogButtonBox::Ignore ), &QPushButton::clicked, this, &QgsHandleBadLayers::reject );
connect( buttonBox->button( QDialogButtonBox::Discard ), &QPushButton::clicked, this, &QgsHandleBadLayers::accept );
mLayerList->clear();
mLayerList->setSortingEnabled( true );
mLayerList->setSelectionBehavior( QAbstractItemView::SelectRows );
mLayerList->setColumnCount( 5 );
mLayerList->setColumnWidth( 3, 75 );
mLayerList->setHorizontalHeaderLabels( QStringList()
<< tr( "Layer name" )
<< tr( "Type" )
<< tr( "Provider" )
<< tr( "Auth config" )
<< tr( "Datasource" )
);
int j = 0;
for ( int i = 0; i < mLayers.size(); i++ )
{
const QDomNode &node = mLayers[i];
QString name = node.namedItem( QStringLiteral( "layername" ) ).toElement().text();
QString type = node.toElement().attribute( QStringLiteral( "type" ) );
QString datasource = node.namedItem( QStringLiteral( "datasource" ) ).toElement().text();
QString provider = node.namedItem( QStringLiteral( "provider" ) ).toElement().text();
QString vectorProvider = type == QLatin1String( "vector" ) ? provider : tr( "none" );
bool providerFileBased = ( QgsProviderRegistry::instance()->providerCapabilities( provider ) & QgsDataProvider::File ) != 0;
QgsDebugMsg( QStringLiteral( "name=%1 type=%2 provider=%3 datasource='%4'" )
.arg( name,
type,
vectorProvider,
datasource ) );
mLayerList->setRowCount( j + 1 );
QTableWidgetItem *item = nullptr;
item = new QTableWidgetItem( name );
item->setData( Qt::UserRole + 0, i );
item->setFlags( item->flags() & ~Qt::ItemIsEditable );
mLayerList->setItem( j, 0, item );
item = new QTableWidgetItem( type );
item->setData( Qt::UserRole + 0, providerFileBased );
item->setFlags( item->flags() & ~Qt::ItemIsEditable );
mLayerList->setItem( j, 1, item );
item = new QTableWidgetItem( vectorProvider );
item->setFlags( item->flags() & ~Qt::ItemIsEditable );
mLayerList->setItem( j, 2, item );
if ( QgsAuthConfigUriEdit::hasConfigId( datasource ) )
{
QToolButton *btn = new QToolButton( this );
btn->setMaximumWidth( 75 );
btn->setMinimumHeight( 24 );
btn->setText( tr( "Edit" ) );
btn->setProperty( "row", j );
connect( btn, &QAbstractButton::clicked, this, &QgsHandleBadLayers::editAuthCfg );
mLayerList->setCellWidget( j, 3, btn );
}
else
{
item = new QTableWidgetItem( QString() );
mLayerList->setItem( j, 3, item );
}
item = new QTableWidgetItem( datasource );
mLayerList->setItem( j, 4, item );
j++;
}
// mLayerList->resizeColumnsToContents();
}
void QgsHandleBadLayers::selectionChanged()
{
mRows.clear();
Q_FOREACH ( QTableWidgetItem *item, mLayerList->selectedItems() )
{
if ( item->column() != 0 )
continue;
bool providerFileBased = mLayerList->item( item->row(), 1 )->data( Qt::UserRole + 0 ).toBool();
if ( !providerFileBased )
continue;
mRows << item->row();
}
mBrowseButton->setEnabled( !mRows.isEmpty() );
}
QString QgsHandleBadLayers::filename( int row )
{
QString type = mLayerList->item( row, 1 )->text();
QString provider = mLayerList->item( row, 2 )->text();
QString datasource = mLayerList->item( row, 4 )->text();
if ( type == QLatin1String( "vector" ) )
{
const QVariantMap parts = QgsProviderRegistry::instance()->decodeUri( provider, datasource );
// if parts is empty then provider doesn't handle this method!
return parts.empty() ? datasource : parts.value( QStringLiteral( "path" ) ).toString();
}
else
{
return datasource;
}
}
void QgsHandleBadLayers::setFilename( int row, const QString &filename )
{
if ( !QFileInfo::exists( filename ) )
return;
QString type = mLayerList->item( row, 1 )->text();
QString provider = mLayerList->item( row, 2 )->text();
QTableWidgetItem *item = mLayerList->item( row, 4 );
QString datasource = item->text();
if ( type == QLatin1String( "vector" ) )
{
if ( provider == QLatin1String( "spatialite" ) )
{
QgsDataSourceUri uri( datasource );
uri.setDatabase( filename );
datasource = uri.uri();
}
else if ( provider == QLatin1String( "ogr" ) )
{
QStringList theURIParts = datasource.split( '|' );
theURIParts[0] = filename;
datasource = theURIParts.join( QStringLiteral( "|" ) );
}
else if ( provider == QLatin1String( "delimitedtext" ) )
{
QUrl uriSource = QUrl::fromEncoded( datasource.toLatin1() );
QUrl uriDest = QUrl::fromLocalFile( filename );
uriDest.setQueryItems( uriSource.queryItems() );
datasource = QString::fromLatin1( uriDest.toEncoded() );
}
}
else
{
datasource = filename;
}
item->setText( datasource );
}
void QgsHandleBadLayers::browseClicked()
{
if ( mRows.size() == 1 )
{
int row = mRows.at( 0 );
QString type = mLayerList->item( row, 1 )->text();
QString memoryQualifier, fileFilter;
if ( type == QLatin1String( "vector" ) )
{
memoryQualifier = QStringLiteral( "lastVectorFileFilter" );
fileFilter = mVectorFileFilter;
}
else
{
memoryQualifier = QStringLiteral( "lastRasterFileFilter" );
fileFilter = mRasterFileFilter;
}
QString fn = filename( row );
if ( fn.isNull() )
return;
QStringList selectedFiles;
QString enc;
QString title = tr( "Select File to Replace '%1'" ).arg( fn );
QgsGuiUtils::openFilesRememberingFilter( memoryQualifier, fileFilter, selectedFiles, enc, title );
if ( selectedFiles.size() != 1 )
{
QMessageBox::information( this, title, tr( "Please select exactly one file." ) );
return;
}
setFilename( row, selectedFiles[0] );
}
else if ( mRows.size() > 1 )
{
QString title = tr( "Select New Directory of Selected Files" );
QgsSettings settings;
QString lastDir = settings.value( QStringLiteral( "UI/missingDirectory" ), QDir::homePath() ).toString();
QString selectedFolder = QFileDialog::getExistingDirectory( this, title, lastDir );
if ( selectedFolder.isEmpty() )
{
return;
}
QDir dir( selectedFolder );
if ( !dir.exists() )
{
return;
}
Q_FOREACH ( int row, mRows )
{
bool providerFileBased = mLayerList->item( row, 1 )->data( Qt::UserRole + 0 ).toBool();
if ( !providerFileBased )
continue;
QString fn = filename( row );
if ( fn.isEmpty() )
continue;
QFileInfo fi( fn );
fi.setFile( dir, fi.fileName() );
if ( !fi.exists() )
continue;
setFilename( row, fi.absoluteFilePath() );
}
}
}
void QgsHandleBadLayers::editAuthCfg()
{
QToolButton *btn = qobject_cast<QToolButton *>( sender() );
int row = -1;
for ( int i = 0; i < mLayerList->rowCount(); i++ )
{
if ( mLayerList->cellWidget( i, 3 ) == btn )
{
row = i;
break;
}
}
if ( row == -1 )
return;
QString provider = mLayerList->item( row, 2 )->text();
if ( provider == QLatin1String( "none" ) )
provider.clear();
QString prevuri = mLayerList->item( row, 4 )->text();
QgsAuthConfigUriEdit *dlg = new QgsAuthConfigUriEdit( this, prevuri, provider );
dlg->setWindowModality( Qt::WindowModal );
dlg->resize( 500, 500 );
if ( dlg->exec() )
{
QString newuri( dlg->dataSourceUri() );
if ( newuri != prevuri )
{
mLayerList->item( row, 4 )->setText( newuri );
}
}
dlg->deleteLater();
}
void QgsHandleBadLayers::apply()
{
QgsProject::instance()->layerTreeRegistryBridge()->setEnabled( true );
buttonBox->button( QDialogButtonBox::Ignore )->setEnabled( false );
for ( int i = 0; i < mLayerList->rowCount(); i++ )
{
int idx = mLayerList->item( i, 0 )->data( Qt::UserRole ).toInt();
QDomNode &node = const_cast<QDomNode &>( mLayers[ idx ] );
QTableWidgetItem *item = mLayerList->item( i, 4 );
QString datasource = item->text();
bool dataSourceChanged { false };
const QString layerId { node.namedItem( QStringLiteral( "id" ) ).toElement().text() };
const QString provider { node.namedItem( QStringLiteral( "provider" ) ).toElement().text() };
const QString name { mLayerList->item( i, 0 )->text() };
// Try first to change the datasource of the existing layers, this will
// maintain the current status (checked/unchecked) and group
if ( QgsProject::instance()->mapLayer( layerId ) )
{
QgsDataProvider::ProviderOptions options;
QgsMapLayer *mapLayer = QgsProject::instance()->mapLayer( layerId );
if ( mapLayer )
{
mapLayer->setDataSource( datasource, name, provider, options );
dataSourceChanged = mapLayer->isValid();
}
}
// If the data source was changed successfully, remove the bad layer from the dialog
// otherwise, try to set the new datasource in the XML node and reload the layer,
// finally marks with red all remaining bad layers.
if ( dataSourceChanged )
{
mLayerList->removeRow( i-- );
}
else
{
node.namedItem( QStringLiteral( "datasource" ) ).toElement().firstChild().toText().setData( datasource );
if ( QgsProject::instance()->readLayer( node ) )
{
mLayerList->removeRow( i-- );
}
else
{
item->setForeground( QBrush( Qt::red ) );
}
}
}
// Final cleanup: remove any bad layer (it should not be any btw)
if ( mLayerList->rowCount() == 0 )
{
QList<QgsMapLayer *> toRemove;
const auto mapLayers = QgsProject::instance()->mapLayers();
for ( const auto &l : mapLayers )
{
if ( ! l->isValid() )
toRemove << l;
}
QgsProject::instance()->removeMapLayers( toRemove );
accept();
}
QgsProject::instance()->layerTreeRegistryBridge()->setEnabled( false );
}
void QgsHandleBadLayers::accept()
{
if ( mLayerList->rowCount() > 0 &&
QMessageBox::warning( this,
tr( "Unhandled layer will be lost." ),
tr( "There are still %n unhandled layer(s). If they are not fixed, they will be disabled/deactivated until the project is opened again.",
"unhandled layers",
mLayerList->rowCount() ),
QMessageBox::Ok | QMessageBox::Cancel,
QMessageBox::Cancel ) == QMessageBox::Cancel )
{
return;
}
QList<QgsMapLayer *> toRemove;
for ( const auto &l : QgsProject::instance()->mapLayers( ) )
{
if ( ! l->isValid() )
toRemove << l;
}
QgsProject::instance()->layerTreeRegistryBridge()->setEnabled( true );
QgsProject::instance()->removeMapLayers( toRemove );
QgsProject::instance()->layerTreeRegistryBridge()->setEnabled( false );
mLayerList->clear();
QDialog::accept();
}
int QgsHandleBadLayers::layerCount()
{
return mLayerList->rowCount();
}
/***************************************************************************
qgshandlebadlayers.cpp - description
-------------------
begin : Sat 5 Mar 2011
copyright : (C) 2011 by Juergen E. Fischer, norBIT GmbH
email : jef at norbit dot de
***************************************************************************/
/***************************************************************************
* *
* 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. *
* *
***************************************************************************/
#include "qgshandlebadlayers.h"
#include "qgisapp.h"
#include "qgsauthconfigselect.h"
#include "qgsdataprovider.h"
#include "qgsguiutils.h"
#include "qgsdatasourceuri.h"
#include "qgslogger.h"
#include "qgsrasterlayer.h"
#include "qgsproviderregistry.h"
#include "qgsmessagebar.h"
#include "qgssettings.h"
#include "qgslayertreeregistrybridge.h"
#include "qgsapplication.h"
#include <QDomDocument>
#include <QDomElement>
#include <QFileDialog>
#include <QPushButton>
#include <QToolButton>
#include <QMessageBox>
#include <QDialogButtonBox>
#include <QUrl>
void QgsHandleBadLayersHandler::handleBadLayers( const QList<QDomNode> &layers )
{
QApplication::setOverrideCursor( Qt::ArrowCursor );
QgsHandleBadLayers *dialog = new QgsHandleBadLayers( layers );
dialog->buttonBox->button( QDialogButtonBox::Ignore )->setToolTip( tr( "Import all unavailable layers unmodified (you can fix them later)." ) );
dialog->buttonBox->button( QDialogButtonBox::Ignore )->setText( tr( "Keep Unavailable Layers" ) );
dialog->buttonBox->button( QDialogButtonBox::Discard )->setToolTip( tr( "Remove all unavailable layers from the project" ) );
dialog->buttonBox->button( QDialogButtonBox::Discard )->setText( tr( "Remove Unavailable Layers" ) );
dialog->buttonBox->button( QDialogButtonBox::Discard )->setIcon( QgsApplication::getThemeIcon( QStringLiteral( "/mActionDeleteSelected.svg" ) ) );
if ( dialog->layerCount() < layers.size() )
QgisApp::instance()->messageBar()->pushMessage(
tr( "Handle unavailable layers" ),
tr( "%1 of %2 unavailable layers were not fixable." )
.arg( layers.size() - dialog->layerCount() )
.arg( layers.size() ),
Qgis::Warning, QgisApp::instance()->messageTimeout() );
if ( dialog->layerCount() > 0 )
{
if ( dialog->exec() == dialog->Accepted )
{
emit layersChanged();
}
}
delete dialog;
QApplication::restoreOverrideCursor();
}
QgsHandleBadLayers::QgsHandleBadLayers( const QList<QDomNode> &layers )
: QDialog( QgisApp::instance() )
, mLayers( layers )
{
setupUi( this );
mVectorFileFilter = QgsProviderRegistry::instance()->fileVectorFilters();
mRasterFileFilter = QgsProviderRegistry::instance()->fileRasterFilters();
mBrowseButton = new QPushButton( tr( "Browse" ) );
buttonBox->addButton( mBrowseButton, QDialogButtonBox::ActionRole );
mBrowseButton->setDisabled( true );
mApplyButton = new QPushButton( tr( "Apply Changes" ) );
mApplyButton->setToolTip( tr( "Apply fixes to unavailable layers (remaining unavailable layers will be removed from the project)." ) );
buttonBox->addButton( mApplyButton, QDialogButtonBox::ActionRole );
connect( mLayerList, &QTableWidget::itemSelectionChanged, this, &QgsHandleBadLayers::selectionChanged );
connect( mBrowseButton, &QAbstractButton::clicked, this, &QgsHandleBadLayers::browseClicked );
connect( mApplyButton, &QAbstractButton::clicked, this, &QgsHandleBadLayers::apply );
connect( buttonBox->button( QDialogButtonBox::Ignore ), &QPushButton::clicked, this, &QgsHandleBadLayers::reject );
connect( buttonBox->button( QDialogButtonBox::Discard ), &QPushButton::clicked, this, &QgsHandleBadLayers::accept );
mLayerList->clear();
mLayerList->setSortingEnabled( true );
mLayerList->setSelectionBehavior( QAbstractItemView::SelectRows );
mLayerList->setColumnCount( 5 );
mLayerList->setColumnWidth( 3, 75 );
mLayerList->setHorizontalHeaderLabels( QStringList()
<< tr( "Layer name" )
<< tr( "Type" )
<< tr( "Provider" )
<< tr( "Auth config" )
<< tr( "Datasource" )
);
int j = 0;
for ( int i = 0; i < mLayers.size(); i++ )
{
const QDomNode &node = mLayers[i];
QString name = node.namedItem( QStringLiteral( "layername" ) ).toElement().text();
QString type = node.toElement().attribute( QStringLiteral( "type" ) );
QString datasource = node.namedItem( QStringLiteral( "datasource" ) ).toElement().text();
QString provider = node.namedItem( QStringLiteral( "provider" ) ).toElement().text();
QString vectorProvider = type == QLatin1String( "vector" ) ? provider : tr( "none" );
bool providerFileBased = ( QgsProviderRegistry::instance()->providerCapabilities( provider ) & QgsDataProvider::File ) != 0;
mFileBase[name].append( const datasource.left( datasource.lastIndexOf('/') ) );
QgsDebugMsg( QStringLiteral( "name=%1 type=%2 provider=%3 datasource='%4'" )
.arg( name,
type,
vectorProvider,
datasource ) );
mLayerList->setRowCount( j + 1 );
QTableWidgetItem *item = nullptr;
item = new QTableWidgetItem( name );
item->setData( Qt::UserRole + 0, i );
item->setFlags( item->flags() & ~Qt::ItemIsEditable );
mLayerList->setItem( j, 0, item );
item = new QTableWidgetItem( type );
item->setData( Qt::UserRole + 0, providerFileBased );
item->setFlags( item->flags() & ~Qt::ItemIsEditable );
mLayerList->setItem( j, 1, item );
item = new QTableWidgetItem( vectorProvider );
item->setFlags( item->flags() & ~Qt::ItemIsEditable );
mLayerList->setItem( j, 2, item );
if ( QgsAuthConfigUriEdit::hasConfigId( datasource ) )
{
QToolButton *btn = new QToolButton( this );
btn->setMaximumWidth( 75 );
btn->setMinimumHeight( 24 );
btn->setText( tr( "Edit" ) );
btn->setProperty( "row", j );
connect( btn, &QAbstractButton::clicked, this, &QgsHandleBadLayers::editAuthCfg );
mLayerList->setCellWidget( j, 3, btn );
}
else
{
item = new QTableWidgetItem( QString() );
mLayerList->setItem( j, 3, item );
}
item = new QTableWidgetItem( datasource );
mLayerList->setItem( j, 4, item );
j++;
}
// mLayerList->resizeColumnsToContents();
}
void QgsHandleBadLayers::selectionChanged()
{
mRows.clear();
Q_FOREACH ( QTableWidgetItem *item, mLayerList->selectedItems() )
{
if ( item->column() != 0 )
continue;
bool providerFileBased = mLayerList->item( item->row(), 1 )->data( Qt::UserRole + 0 ).toBool();
if ( !providerFileBased )
continue;
mRows << item->row();
}
mBrowseButton->setEnabled( !mRows.isEmpty() );
}
QString QgsHandleBadLayers::filename( int row )
{
QString type = mLayerList->item( row, 1 )->text();
QString provider = mLayerList->item( row, 2 )->text();
QString datasource = mLayerList->item( row, 4 )->text();
if ( type == QLatin1String( "vector" ) )
{
const QVariantMap parts = QgsProviderRegistry::instance()->decodeUri( provider, datasource );
// if parts is empty then provider doesn't handle this method!
return parts.empty() ? datasource : parts.value( QStringLiteral( "path" ) ).toString();
}
else
{
return datasource;
}
}
void QgsHandleBadLayers::setFilename( int row, const QString &filename )
{
if ( !QFileInfo::exists( filename ) )
return;
QString type = mLayerList->item( row, 1 )->text();
QString provider = mLayerList->item( row, 2 )->text();
QTableWidgetItem *item = mLayerList->item( row, 4 );
QString datasource = item->text();
if ( type == QLatin1String( "vector" ) )
{
if ( provider == QLatin1String( "spatialite" ) )
{
QgsDataSourceUri uri( datasource );
uri.setDatabase( filename );
datasource = uri.uri();
}
else if ( provider == QLatin1String( "ogr" ) )
{
QStringList theURIParts = datasource.split( '|' );
theURIParts[0] = filename;
datasource = theURIParts.join( QStringLiteral( "|" ) );
}
else if ( provider == QLatin1String( "delimitedtext" ) )
{
QUrl uriSource = QUrl::fromEncoded( datasource.toLatin1() );
QUrl uriDest = QUrl::fromLocalFile( filename );
uriDest.setQueryItems( uriSource.queryItems() );
datasource = QString::fromLatin1( uriDest.toEncoded() );
}
}
else
{
datasource = filename;
}
item->setText( datasource );
}
void QgsHandleBadLayers::browseClicked()
{
if ( mRows.size() == 1 )
{
int row = mRows.at( 0 );
QString type = mLayerList->item( row, 1 )->text();
QString memoryQualifier, fileFilter;
if ( type == QLatin1String( "vector" ) )
{
memoryQualifier = QStringLiteral( "lastVectorFileFilter" );
fileFilter = mVectorFileFilter;
}
else
{
memoryQualifier = QStringLiteral( "lastRasterFileFilter" );
fileFilter = mRasterFileFilter;
}
QString fn = filename( row );
if ( fn.isNull() )
return;
QStringList selectedFiles;
QString enc;
QString title = tr( "Select File to Replace '%1'" ).arg( fn );
QgsGuiUtils::openFilesRememberingFilter( memoryQualifier, fileFilter, selectedFiles, enc, title );
if ( selectedFiles.size() != 1 )
{
QMessageBox::information( this, title, tr( "Please select exactly one file." ) );
return;
}
setFilename( row, selectedFiles[0] );
}
else if ( mRows.size() > 1 )
{
QString title = tr( "Select New Directory of Selected Files" );
QgsSettings settings;
QString lastDir = settings.value( QStringLiteral( "UI/missingDirectory" ), QDir::homePath() ).toString();
QString selectedFolder = QFileDialog::getExistingDirectory( this, title, lastDir );
if ( selectedFolder.isEmpty() )
{
return;
}
QDir dir( selectedFolder );
if ( !dir.exists() )
{
return;
}
Q_FOREACH ( int row, mRows )
{
bool providerFileBased = mLayerList->item( row, 1 )->data( Qt::UserRole + 0 ).toBool();
if ( !providerFileBased )
continue;
QString fn = filename( row );
if ( fn.isEmpty() )
continue;
QFileInfo fi( fn );
fi.setFile( dir, fi.fileName() );
if ( !fi.exists() )
continue;
setFilename( row, fi.absoluteFilePath() );
}
}
}
void QgsHandleBadLayers::editAuthCfg()
{
QToolButton *btn = qobject_cast<QToolButton *>( sender() );
int row = -1;
for ( int i = 0; i < mLayerList->rowCount(); i++ )
{
if ( mLayerList->cellWidget( i, 3 ) == btn )
{
row = i;
break;
}
}
if ( row == -1 )
return;
QString provider = mLayerList->item( row, 2 )->text();
if ( provider == QLatin1String( "none" ) )
provider.clear();
QString prevuri = mLayerList->item( row, 4 )->text();
QgsAuthConfigUriEdit *dlg = new QgsAuthConfigUriEdit( this, prevuri, provider );
dlg->setWindowModality( Qt::WindowModal );
dlg->resize( 500, 500 );
if ( dlg->exec() )
{
QString newuri( dlg->dataSourceUri() );
if ( newuri != prevuri )
{
mLayerList->item( row, 4 )->setText( newuri );
}
}
dlg->deleteLater();
}
void QgsHandleBadLayers::apply()
{
QgsProject::instance()->layerTreeRegistryBridge()->setEnabled( true );
buttonBox->button( QDialogButtonBox::Ignore )->setEnabled( false );
QHash<QString, QString> baseChange;
for ( int i = 0; i < mLayerList->rowCount(); i++ )
{
int idx = mLayerList->item( i, 0 )->data( Qt::UserRole ).toInt();
QDomNode &node = const_cast<QDomNode &>( mLayers[ idx ] );
QTableWidgetItem *item = mLayerList->item( i, 4 );
QString datasource = item->text();
const QString basepath = datasource.left( datasource.lastIndexOf('/') )
bool changed = false;
if ( mFileBase[ name ].size() == 1 )
{
if ( mFileBase[ name ][0] != basepath && !baseChange.contains( mFileBase[ name ][0] ) )
{
baseChange[ mFileBase[ name ][0] ] = basepath;
changed = true;
}
}
else if ( mFileBase[ name ].size() > 1 )
{
if ( mFileBase[ name ].indexOf( basepath ) == -1 )
{
const QList fileBases = mFileBase[ name ];
for ( QString fileBase : fileBases )
{
if ( !baseChange.contains( fileBase ) )
{
baseChange[ fileBase ] = basepath;
changed = true;
}
}
}
}
if ( !changed && baseChange.contains( basepath ) )
datasource = datasource.replace( basepath, baseChange( basepath ) );
bool dataSourceChanged { false };
const QString layerId { node.namedItem( QStringLiteral( "id" ) ).toElement().text() };
const QString provider { node.namedItem( QStringLiteral( "provider" ) ).toElement().text() };
const QString name { mLayerList->item( i, 0 )->text() };
// Try first to change the datasource of the existing layers, this will
// maintain the current status (checked/unchecked) and group
if ( QgsProject::instance()->mapLayer( layerId ) )
{
QgsDataProvider::ProviderOptions options;
QgsMapLayer *mapLayer = QgsProject::instance()->mapLayer( layerId );
if ( mapLayer )
{
mapLayer->setDataSource( datasource, name, provider, options );
dataSourceChanged = mapLayer->isValid();
}
}
// If the data source was changed successfully, remove the bad layer from the dialog
// otherwise, try to set the new datasource in the XML node and reload the layer,
// finally marks with red all remaining bad layers.
if ( dataSourceChanged )
{
mLayerList->removeRow( i-- );
}
else
{
node.namedItem( QStringLiteral( "datasource" ) ).toElement().firstChild().toText().setData( datasource );
if ( QgsProject::instance()->readLayer( node ) )
{
mLayerList->removeRow( i-- );
}
else
{
item->setForeground( QBrush( Qt::red ) );
}
}
}
// Final cleanup: remove any bad layer (it should not be any btw)
if ( mLayerList->rowCount() == 0 )
{
QList<QgsMapLayer *> toRemove;
const auto mapLayers = QgsProject::instance()->mapLayers();
for ( const auto &l : mapLayers )
{
if ( ! l->isValid() )
toRemove << l;
}
QgsProject::instance()->removeMapLayers( toRemove );
accept();
}
QgsProject::instance()->layerTreeRegistryBridge()->setEnabled( false );
}
void QgsHandleBadLayers::accept()
{
if ( mLayerList->rowCount() > 0 &&
QMessageBox::warning( this,
tr( "Unhandled layer will be lost." ),
tr( "There are still %n unhandled layer(s). If they are not fixed, they will be disabled/deactivated until the project is opened again.",
"unhandled layers",
mLayerList->rowCount() ),
QMessageBox::Ok | QMessageBox::Cancel,
QMessageBox::Cancel ) == QMessageBox::Cancel )
{
return;
}
QList<QgsMapLayer *> toRemove;
for ( const auto &l : QgsProject::instance()->mapLayers( ) )
{
if ( ! l->isValid() )
toRemove << l;
}
QgsProject::instance()->layerTreeRegistryBridge()->setEnabled( true );
QgsProject::instance()->removeMapLayers( toRemove );
QgsProject::instance()->layerTreeRegistryBridge()->setEnabled( false );
mLayerList->clear();
QDialog::accept();
}
int QgsHandleBadLayers::layerCount()
{
return mLayerList->rowCount();
}

159
src/app/qgshandlebadlayers.h
View File

@ -1,79 +1,80 @@
/***************************************************************************
qgshandlebadlayers.h - description
-------------------
begin : Sat 05 Mar 2011
copyright : (C) 2011 by Juergen E. Fischer, norBIT GmbH
email : jef at norbit dot de
***************************************************************************/
/***************************************************************************
* *
* 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 QGSHANDLEBADLAYERS_H
#define QGSHANDLEBADLAYERS_H
#include "ui_qgshandlebadlayersbase.h"
#include "qgsprojectbadlayerhandler.h"
#include "qgis_app.h"
class APP_EXPORT QgsHandleBadLayersHandler
: public QObject
, public QgsProjectBadLayerHandler
{
Q_OBJECT
public:
QgsHandleBadLayersHandler() = default;
//! Implementation of the handler
void handleBadLayers( const QList<QDomNode> &layers ) override;
signals:
/**
* Emitted when layers have changed
* \since QGIS 3.6
*/
void layersChanged();
};
class QPushButton;
class APP_EXPORT QgsHandleBadLayers
: public QDialog
, public Ui::QgsHandleBadLayersBase
{
Q_OBJECT
public:
QgsHandleBadLayers( const QList<QDomNode> &layers );
int layerCount();
private slots:
void selectionChanged();
void browseClicked();
void editAuthCfg();
void apply();
void accept() override;
private:
QPushButton *mBrowseButton = nullptr;
QPushButton *mApplyButton = nullptr;
const QList<QDomNode> &mLayers;
QList<int> mRows;
QString mVectorFileFilter;
QString mRasterFileFilter;
QString filename( int row );
void setFilename( int row, const QString &filename );
};
#endif
/***************************************************************************
qgshandlebadlayers.h - description
-------------------
begin : Sat 05 Mar 2011
copyright : (C) 2011 by Juergen E. Fischer, norBIT GmbH
email : jef at norbit dot de
***************************************************************************/
/***************************************************************************
* *
* 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 QGSHANDLEBADLAYERS_H
#define QGSHANDLEBADLAYERS_H
#include "ui_qgshandlebadlayersbase.h"
#include "qgsprojectbadlayerhandler.h"
#include "qgis_app.h"
class APP_EXPORT QgsHandleBadLayersHandler
: public QObject
, public QgsProjectBadLayerHandler
{
Q_OBJECT
public:
QgsHandleBadLayersHandler() = default;
//! Implementation of the handler
void handleBadLayers( const QList<QDomNode> &layers ) override;
signals:
/**
* Emitted when layers have changed
* \since QGIS 3.6
*/
void layersChanged();
};
class QPushButton;
class APP_EXPORT QgsHandleBadLayers
: public QDialog
, public Ui::QgsHandleBadLayersBase
{
Q_OBJECT
public:
QgsHandleBadLayers( const QList<QDomNode> &layers );
int layerCount();
private slots:
void selectionChanged();
void browseClicked();
void editAuthCfg();
void apply();
void accept() override;
private:
QPushButton *mBrowseButton = nullptr;
QPushButton *mApplyButton = nullptr;
const QList<QDomNode> &mLayers;
QList<int> mRows;
QString mVectorFileFilter;
QString mRasterFileFilter;
QHash <QString, QList<QString> > mFileBase;
QString filename( int row );
void setFilename( int row, const QString &filename );
};
#endif