source: cowsclient/trunk/cowsclient/controllers/wmsviz.py @ 5626

Subversion URL: http://proj.badc.rl.ac.uk/svn/ndg/cowsclient/trunk/cowsclient/controllers/wmsviz.py@5626
Revision 5626, 9.0 KB checked in by domlowe, 11 years ago (diff)

MERGING the Qesdi branch back into the trunk. Version 0.2.1

Line 
1# Copyright (C) 2007 STFC & NERC (Science and Technology Facilities Council).
2# This software may be distributed under the terms of the
3# Q Public License, version 1.0 or later.
4# http://ndg.nerc.ac.uk/public_docs/QPublic_license.txt
5#
6"""
7 Controller for the 'View' tab - allowing the display of WMC map layers
8
9 @author C Byrom Feb 08, Modified D Lowe, May 09
10"""
11
12
13from cowsclient.lib.base import * #BaseController, g, response, config, request, c, session, render etc
14from paste.request import parse_querystring
15#from ows_server.models import Utilities
16from cowsclient.lib.wmc_util import GetWebMapContext, GetWebMapCapabilities, GetLegend, openURL, GetResponse
17from cowsclient.model import selectedItem
18from cowsclient.lib.build_figure import build_figure
19
20import simplejson as json
21import urlparse
22from cStringIO import StringIO
23try:
24    from PIL import Image, ImageChops
25except:
26    import Image, ImageChops
27import logging
28log = logging.getLogger(__name__)
29import urllib, urllib2
30
31class WmsvizController(BaseController):
32   
33    _pilImageFormats = {
34        'image/png': 'PNG',
35        'image/jpg': 'JPEG',
36        'image/gif': 'GIF',
37        'image/tiff': 'TIFF'
38    }
39   
40    def index(self):
41        """
42        Default controller method to handle the initial requests to the page
43        """
44        log.debug('wmsviz controller')
45       
46       
47        g.helpIcon='layout/icons/help.png'  #needs to go in config
48       
49        self.inputs=dict(parse_querystring(request.environ))
50        log.info(self.inputs)
51        c.wmcURL = ""
52       
53        # check if all we're doing is removing a view item
54        if 'removeItem' in self.inputs:
55            return self.removeViewItem(self.inputs['removeItem'])
56       
57       
58        log.debug("self.inputs = %s" % (self.inputs,))
59        # check if we're doing an AJAX callback to get some WMC data
60        if 'REQUEST' in self.inputs:
61                       
62            if self.inputs['REQUEST'] == 'GetWebMapContext':
63                wmc= GetWebMapContext(self)
64                return wmc
65           
66            if self.inputs['REQUEST'] == 'GetWebMapCapabilities':
67                wmcDoc = GetWebMapCapabilities(self.inputs['ENDPOINT'])
68               
69                response.headers['Content-Type'] = 'text/xml'
70                response.write(wmcDoc)
71                return           
72           
73            elif self.inputs['REQUEST'] == 'GetLegend':
74                return GetLegend(self)
75           
76            if self.inputs['REQUEST'] == 'GetDisplayOptions':
77               
78                jsonTxt = GetResponse(self.inputs['URL'])
79               
80                response.headers['Content-Type'] = 'application/json'
81                response.write(jsonTxt)
82                return
83               
84
85       
86        #get server information from config file
87        g.server=config['app_conf']['serverurl']
88   
89   
90        initialSetup = self._buildInitialSetup(self.inputs.get('ENDPOINT'))
91       
92        #TODO: WORK OUT HOW TO COUPLE THIS TO BROWSE
93        # otherwise, we can get here by two routes:
94        # i) either by clicking on the WMC icon in the details view - if so, get passed endpoint and add to
95        #    selected items;
96        # ii) or from the selections tab - which will already have set up the correct selected items
97           
98        # if ENDPOINT specified, we've reached the page via the WMC icon
99       
100        session['test'] = "test"
101       
102        #This will ensure that NDG security is triggered if a secured dataset is in Endpoint
103        if 'ENDPOINT' in self.inputs:
104            #clear out old endpoints NOTE. this means only one endpoint at a time can be viewed. May want to
105        #rethink this to enable 'shopping cart' type selection.
106#        self.removeAllViewItems()
107            for endpoint in initialSetup:
108                urlstring=endpoint['url']+'?request=GetCapabilities&Service=WMS'
109                #urlstring=str(self.inputs['ENDPOINT'])
110                req = urllib2.Request(urlstring)
111                req.add_header('Cookie', request.headers.get('Cookie', ''))
112                try:
113                   filehandle = openURL(req)
114                   filehandle.close()
115                except urllib2.HTTPError, e:           
116                    log.exception("exception occurred")
117                    if e.code == 401:
118                        log.info ('401 unauthorized error in cowsclient')
119                        return abort(401) #triggers ndg security framework
120                    elif e.code == 403:  #TODO: 403 response is UNTESTED.
121                        # User is authenticated but doesn't have the required permissions
122                        # or an error occurred in the authorization process
123                        # Read response                   
124                        xresponse = e.read()
125                        # Send response to user
126                        self.start_response("%d %s" % (e.code, e.msg), e.headers.dict.items())
127                        return xresponse
128               
129       
130        # avoid page crashing if we come here without view items defined
131#        if 'viewItems' not in session:
132#            h.redirect_to(h.url_for(controller='discovery'))
133#                                       
134        session.save() 
135        log.info('SAVED SESSION')
136       
137        # check if page has been visited before; if not display tab
138#        if (Utilities.isTabRequired(c.pageTabs, 'View')):
139#            c.pageTabs.append(('View', h.url_for(controller='viewItems',action='index')))
140
141
142           
143        c.initialSetupJSON = json.dumps(initialSetup).replace('"', '\\"') 
144
145       
146        params = {}
147        for paramString in config['baselayer.params'].split(','):
148            key, value = paramString.split(':')
149            params[key] = value
150       
151        baseLayerObj = {"url":config['baselayer.url'], "params": params }
152       
153        c.baseLayerJSON = json.dumps(baseLayerObj).replace('"', '\\"') 
154       
155
156        return render('wmsviz')
157
158    def _buildInitialSetup(self, endpointParam):
159        initialSetup = []
160       
161        if endpointParam != None:
162           
163            for ep in self.inputs['ENDPOINT'].split(','):
164                endpoint = {}
165                o = urlparse.urlparse(ep)
166               
167                if o.path.find(':') > 0:
168                    path = o.path[:o.path.find(':')]
169
170                    url = "%(scheme)s://%(hostname)s%(port)s%(path)s" % {
171                        'scheme' : o.scheme if o.scheme != None else '',
172                        'hostname' : o.hostname if o.hostname != None else '',
173                        'port' : ':' + str(o.port) if o.port != None else '',
174                        'path': path,
175                    }
176                    layers = o.path[o.path.find(':')+1:].split('|')
177                   
178                    endpoint['layers'] = layers
179                else:
180                    url = ep
181                    layers = ""
182               
183                endpoint['url'] = url
184
185                initialSetup.append(endpoint)
186               
187        return initialSetup       
188
189    def addViewItem(self,endpoint):
190        """
191        Add a selected item to the session
192         - if this is the first item, then display the selections tab
193         @param endpoint: WMC endpoint
194        """
195       
196        item = selectedItem.SelectedItem(None, None, None, endpoint)
197       
198        selections = [item,]
199        # avoid duplicates
200        if 'viewItems' in session:
201            for selection in session['viewItems']:
202                if selection.wmcURL != endpoint:
203                    selections.append(selection)
204                   
205        session['viewItems'] = selections
206        session.save() 
207       
208   
209    def removeViewItem(self,endpoint):
210        """
211        Remove view item from session data
212        - NB, do this by rebuilding the session data without the input data included
213        @param endpoint: The WMC endpoint of the view item to remove
214        """
215        selections = []
216        for selection in session['viewItems']:
217            if selection.wmcURL != endpoint:
218                selections.append(selection)
219               
220        # if the new list is empty, remove the session variable
221        if len(selections) == 0:
222            del session['viewItems']
223            c.UpdatePageTabs=1
224        else:
225            session['viewItems'] = selections
226        session.save()
227       
228    def removeAllViewItems(self):
229        """
230        Remove all old view items - clears out old endpoints
231        """
232        session['viewItems']=[]
233        session.save()
234       
235
236
237   
238    def get_figure(self):
239        log.debug("running wmsvis.get_figure")
240       
241        params = request.params
242       
243        log.debug("params = %s" % (params,))
244       
245        format = params.pop('figFormat', 'image/png')
246       
247        finalImage = build_figure(params)
248       
249        response.headers['content-type'] = format
250       
251        buffer = StringIO()
252        finalImage.save(buffer, self._pilImageFormats[format])
253       
254        response.headers['Content-Type'] = format
255        response.write(buffer.getvalue()) 
256
257
258
Note: See TracBrowser for help on using the repository browser.