1 files changed, 68 insertions, 0 deletions
diff --git a/lib/python2.7/site-packages/django/contrib/gis/gdal/driver.py b/lib/python2.7/site-packages/django/contrib/gis/gdal/driver.py
new file mode 100644
index 0000000..55a5d77
--- /dev/null
+++ b/lib/python2.7/site-packages/django/contrib/gis/gdal/driver.py
@@ -0,0 +1,68 @@
+# prerequisites imports
+from ctypes import c_void_p
+from django.contrib.gis.gdal.base import GDALBase
+from django.contrib.gis.gdal.error import OGRException
+from django.contrib.gis.gdal.prototypes import ds as capi
+
+from django.utils import six
+from django.utils.encoding import force_bytes
+
+# For more information, see the OGR C API source code:
+#  http://www.gdal.org/ogr/ogr__api_8h.html
+#
+# The OGR_Dr_* routines are relevant here.
+class Driver(GDALBase):
+    "Wraps an OGR Data Source Driver."
+
+    # Case-insensitive aliases for OGR Drivers.
+    _alias = {'esri' : 'ESRI Shapefile',
+              'shp' : 'ESRI Shapefile',
+              'shape' : 'ESRI Shapefile',
+              'tiger' : 'TIGER',
+              'tiger/line' : 'TIGER',
+              }
+
+    def __init__(self, dr_input):
+        "Initializes an OGR driver on either a string or integer input."
+
+        if isinstance(dr_input, six.string_types):
+            # If a string name of the driver was passed in
+            self._register()
+
+            # Checking the alias dictionary (case-insensitive) to see if an alias
+            #  exists for the given driver.
+            if dr_input.lower() in self._alias:
+                name = self._alias[dr_input.lower()]
+            else:
+                name = dr_input
+
+            # Attempting to get the OGR driver by the string name.
+            dr = capi.get_driver_by_name(force_bytes(name))
+        elif isinstance(dr_input, int):
+            self._register()
+            dr = capi.get_driver(dr_input)
+        elif isinstance(dr_input, c_void_p):
+            dr = dr_input
+        else:
+            raise OGRException('Unrecognized input type for OGR Driver: %s' % str(type(dr_input)))
+
+        # Making sure we get a valid pointer to the OGR Driver
+        if not dr:
+            raise OGRException('Could not initialize OGR Driver on input: %s' % str(dr_input))
+        self.ptr = dr
+
+    def __str__(self):
+        "Returns the string name of the OGR Driver."
+        return capi.get_driver_name(self.ptr)
+
+    def _register(self):
+        "Attempts to register all the data source drivers."
+        # Only register all if the driver count is 0 (or else all drivers
+        # will be registered over and over again)
+        if not self.driver_count: capi.register_all()
+
+    # Driver properties
+    @property
+    def driver_count(self):
+        "Returns the number of OGR data source drivers registered."
+        return capi.get_driver_count()