Proxy4OWS/wfs/__init__.py

#!/usr/bin/env python
# coding=utf-8

from OWS import OWS
import mapscript
import cgi
from lxml import objectify
import urllib
import urlparse
import logging
from osgeo import ogr
from osgeo import osr
import os

class WFS(OWS):

    service = "WFS"
    wfsns = "http://www.opengis.net/wfs"
    layerDefFile = None

    def __init__(self,url=None,qstring=None,configFiles=None):
        OWS.__init__(self,url,qstring)

    def makeMap(self,mapfilename=None):

        mapobj = self.getMapObj(mapfilename)


        self.layerDefFile = self.createLayerDefinitionFile("wfs",
                os.path.join( os.path.dirname(__file__), "templates",'wfs.xml'))

        ds = ogr.Open(self.layerDefFile)

        self.setMapName(mapobj)

        logging.debug(self.capabilities.contents)

        for name in self.capabilities.contents:
            layer = self.capabilities.contents[name]

            logging.debug("Creating layer %s" % name)

            lyrobj = mapscript.layerObj(mapobj)
            lyrobj.name = name
            lyrobj.title = layer.title
            if layer.title:
                lyrobj.setMetaData("wms_title",layer.title)
            #if layer.abstract:
            #    lyrobj.setMetaData("ows_abstract",  layer.abstract)
            lyrobj.setMetaData("wfs_typename", name)
            lyrobj.setMetaData("wfs_version",self.capabilities.version)
            lyrobj.setConnectionType(mapscript.MS_OGR,'')
            lyrobj.connection = self.layerDefFile
            lyrobj.data = name

            if ds:
                ogrLayer = ds.GetLayerByName(name)
                # TODO : udelat to pomoci owslib
                extent = self.getLayerExtent(layer,layer.crsOptions[0])
                lyrobj.setMetaData("wms_extent","%s %s %s %s" % \
                        (extent[0],extent[1],extent[2],extent[3]))
                lyrobj.type = self._getLayerType(ogrLayer)
            else:
                mapobj.removeLayer(mapobj.numlayers-1)
                logging.debug("No ogrDataSource found")
                continue

            lyrobj.setProjection(layer.crsOptions[0])

            lyrobj.dump = mapscript.MS_TRUE 
            lyrobj.template = "foo"
            cls = mapscript.classObj(lyrobj)
            style = mapscript.styleObj(cls)
            style.outlinecolor=mapscript.colorObj(134,81,0)
            style.color=mapscript.colorObj(238,153,0)
            style.size=5
            style.width=5

        self.saveMapfile(mapobj,mapfilename)
        return mapobj
    
    def getGeomName(self,geomname):

        if geomname.find("LINE") > -1:
            return mapscript.MS_LAYER_LINE
        elif geomname.find("POLYGON") > -1:
            return mapscript.MS_LAYER_POLYGON
        else:
            return mapscript.MS_LAYER_POINT

    def setMapName(self,mapobj):
        mapobj.name = self.config.get("MapServer","name")

        if self.capabilities.identification.title:
            mapobj.setMetaData("wms_title",self.capabilities.identification.title)
        if self.capabilities.identification.abstract:
            mapobj.setMetaData("wms_abstract",self.capabilities.identification.abstract)

    def getLayerExtent(self,layer,crs):
        """Get extent of layer in form of minx, miny, maxx,maxy
        """

        dest = osr.SpatialReference()
        dest.ImportFromEPSG(int(crs.split(":")[1]))
        source = osr.SpatialReference()
        source.ImportFromEPSG(4326)

        bbox = []

        geom = ogr.CreateGeometryFromWkt("""POINT(%s %s)""" % (layer.boundingBoxWGS84[0],layer.boundingBoxWGS84[1]),source)
        geom.TransformTo(dest)
        bbox.append(geom.GetX())
        bbox.append(geom.GetY())

        geom = ogr.CreateGeometryFromWkt("""POINT(%s %s)""" % (layer.boundingBoxWGS84[2],layer.boundingBoxWGS84[3]),source)
        geom.TransformTo(dest)
        bbox.append(geom.GetX())
        bbox.append(geom.GetY())

        logging.debug("Setting extent for layer <%s> to %s" %\
                (layer.id,bbox))
        return bbox

    def _getLayerType(self,layer):
        """Returns MS layer type based on ogr.Layer.GetGeomType

        with ogr:

            wkbGeometryCollection = 7
            wkbGeometryCollection25D = -2147483641
            wkbLineString = 2
            wkbLineString25D = -2147483646
            wkbLinearRing = 101
            wkbMultiLineString = 5
            wkbMultiLineString25D = -2147483643
            wkbMultiPoint = 4
            wkbMultiPoint25D = -2147483644
            wkbMultiPolygon = 6
            wkbMultiPolygon25D = -2147483642
            wkbNDR = 1
            wkbNone = 100
            wkbPoint = 1
            wkbPoint25D = -2147483647
            wkbPolygon = 3
            wkbPolygon25D = -2147483645
            wkbUnknown = 0
        """

        geomType = layer.GetGeomType()
        if geomType in [ogr.wkbPolygon,
                        ogr.wkbMultiPolygon,
                        ogr.wkbLinearRing]:
               return mapscript.MS_LAYER_POLYGON
        elif geomType in [ogr.wkbLineString,
                          ogr.wkbMultiLineString]:
               return mapscript.MS_LAYER_LINE
        else:
               return mapscript.MS_LAYER_POINT