[processing] add method in OGR algorithms for getting layer name from source

alexbruy · alexbruy · commit 2417a82bd7ab · 2014-11-07T16:11:11.000+02:00
diff --git a/python/plugins/processing/algs/gdal/OgrAlgorithm.py b/python/plugins/processing/algs/gdal/OgrAlgorithm.py
@@ -25,8 +25,8 @@
 
 __revision__ = '$Format:%H$'
 
-import string
 import re
+import os
 
 try:
     from osgeo import ogr
@@ -36,6 +36,7 @@
 
 from PyQt4.QtCore import *
 from PyQt4.QtGui import *
+
 from qgis.core import *
 
 from processing.algs.gdal.GdalAlgorithm import GdalAlgorithm
@@ -66,3 +67,21 @@ def ogrConnectionString(self, uri):
         else:
             ogrstr = unicode(layer.source())
         return ogrstr
+
+    def ogrLayerName(self, uri):
+        layerName = None
+
+        if 'host' in uri:
+            regex = re.compile('(table=")(.+?)(\.")(.+?)"')
+            r = regex.search(uri)
+            return r.groups()[1] + '.' + r.groups()[3]
+        elif 'dbname' in uri:
+            regex = re.compile('(table=")(.+?)"')
+            r = regex.search(uri)
+            return r.groups()[1]
+        elif 'layername' in uri:
+            regex = re.compile('(layername=)(.*)')
+            r = regex.search(uri)
+            return r.groups()[1]
+        else:
+            return os.path.basename(os.path.splitext(uri)[0])
