Skip to content

Commit

Permalink
[processing] restore Execure SQL algorithm
Browse files Browse the repository at this point in the history
  • Loading branch information
alexbruy committed Oct 11, 2017
1 parent fb958df commit 68863a9
Show file tree
Hide file tree
Showing 4 changed files with 120 additions and 101 deletions.
3 changes: 1 addition & 2 deletions python/plugins/processing/algs/gdal/Dissolve.py
Original file line number Diff line number Diff line change
Expand Up @@ -95,8 +95,7 @@ def initAlgorithm(self, config=None):
self.addParameter(param)

self.addParameter(QgsProcessingParameterVectorDestination(self.OUTPUT,
self.tr('Buffer'),
QgsProcessing.TypeVectorPolygon))
self.tr('Dissolved')))

def name(self):
return 'dissolve'
Expand Down
117 changes: 117 additions & 0 deletions python/plugins/processing/algs/gdal/ExecuteSql.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,117 @@
# -*- coding: utf-8 -*-

"""
***************************************************************************
ExecuteSql.py
---------------------
Date : November 2012
Copyright : (C) 2012 by Victor Olaya
Email : volayaf 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. *
* *
***************************************************************************
"""

__author__ = 'Victor Olaya'
__date__ = 'November 2012'
__copyright__ = '(C) 2012, Victor Olaya'

# This will get replaced with a git SHA1 when you do a git archive

__revision__ = '$Format:%H$'


from qgis.core import (QgsProcessingException,
QgsProcessingParameterDefinition,
QgsProcessingParameterFeatureSource,
QgsProcessingParameterEnum,
QgsProcessingParameterString,
QgsProcessingParameterVectorDestination)
from processing.algs.gdal.GdalAlgorithm import GdalAlgorithm
from processing.algs.gdal.GdalUtils import GdalUtils


class ExecuteSql(GdalAlgorithm):

INPUT = 'INPUT'
SQL = 'SQL'
DIALECT = 'DIALECT'
OPTIONS = 'OPTIONS'
OUTPUT = 'OUTPUT'

def __init__(self):
super().__init__()

def initAlgorithm(self, config=None):
self.dialects = ((self.tr('None'), ''),
(self.tr('OGR SQL'), 'ogrsql'),
(self.tr('SQLite'), 'sqlite'))

self.addParameter(QgsProcessingParameterFeatureSource(self.INPUT,
self.tr('Input layer')))
self.addParameter(QgsProcessingParameterString(self.SQL,
self.tr('SQL expression'),
defaultValue=''))
self.addParameter(QgsProcessingParameterEnum(self.SQL,
self.tr('SQL expression'),
defaultValue=''))
self.addParameter(QgsProcessingParameterEnum(self.DIALECT,
self.tr('SQL dialect'),
options=[i[0] for i in self.dialects],
allowMultiple=False,
defaultValue=0))

options_param = QgsProcessingParameterString(self.OPTIONS,
self.tr('Additional creation options'),
defaultValue='',
optional=True)
options_param.setFlags(options_param.flags() | QgsProcessingParameterDefinition.FlagAdvanced)
self.addParameter(options_param)

self.addParameter(QgsProcessingParameterVectorDestination(self.OUTPUT,
self.tr('SQL result')))

def name(self):
return 'executesql'

def displayName(self):
return self.tr('Execute SQL')

def group(self):
return self.tr('Vector miscellaneous')

def getConsoleCommands(self, parameters, context, feedback):
ogrLayer, layerName = self.getOgrCompatibleSource(self.INPUT, parameters, context, feedback)
sql = self.parameterAsString(parameters, self.SQL, context)
options = self.parameterAsString(parameters, self.OPTIONS, context)
outFile = self.parameterAsOutputLayer(parameters, self.OUTPUT, context)

output, outputFormat = GdalUtils.ogrConnectionStringAndFormat(outFile, context)

if not sql:
raise QgsProcessingException(
self.tr('Empty SQL. Please enter valid SQL expression and try again.'))

arguments = []
arguments.append(output)
arguments.append(ogrLayer)
arguments.append('-sql')
arguments.append(sql)

dialect = self.dialects[self.parameterAsEnum(parameters, self.DIALECT, context)][1]
if dialect:
arguments.append('-dialect')
arguments.append(dialect)

if options:
arguments.append(options)

if outputFormat:
arguments.append('-f {}'.format(outputFormat))

return ['ogr2ogr', GdalUtils.escapeAndJoin(arguments)]
4 changes: 2 additions & 2 deletions python/plugins/processing/algs/gdal/GdalAlgorithmProvider.py
Original file line number Diff line number Diff line change
Expand Up @@ -77,6 +77,7 @@
from .ClipVectorByExtent import ClipVectorByExtent
from .ClipVectorByMask import ClipVectorByMask
from .Dissolve import Dissolve
from .ExecuteSql import ExecuteSql
from .OffsetCurve import OffsetCurve
from .ogr2ogr import ogr2ogr
from .ogrinfo import ogrinfo
Expand All @@ -86,7 +87,6 @@

# from .ogr2ogrtopostgislist import Ogr2OgrToPostGisList
# from .ogr2ogrtabletopostgislist import Ogr2OgrTableToPostGisList
# from .ogrsql import OgrSql

pluginPath = os.path.normpath(os.path.join(
os.path.split(os.path.dirname(__file__))[0], os.pardir))
Expand Down Expand Up @@ -179,6 +179,7 @@ def loadAlgorithms(self):
ClipVectorByExtent(),
ClipVectorByMask(),
Dissolve(),
ExecuteSql(),
OffsetCurve(),
ogr2ogr(),
ogrinfo(),
Expand All @@ -187,7 +188,6 @@ def loadAlgorithms(self):
PointsAlongLines(),
# Ogr2OgrToPostGisList(),
# Ogr2OgrTableToPostGisList(),
# OgrSql(),
]
for a in self.algs:
self.addAlgorithm(a)
Expand Down
97 changes: 0 additions & 97 deletions python/plugins/processing/algs/gdal/ogrsql.py

This file was deleted.

0 comments on commit 68863a9

Please sign in to comment.