/[thuban]/branches/WIP-pyshapelib-bramz/Thuban/Model/load.py
ViewVC logotype

Diff of /branches/WIP-pyshapelib-bramz/Thuban/Model/load.py

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 493 by jonathan, Mon Mar 10 10:45:10 2003 UTC revision 1970 by bh, Mon Nov 24 18:36:00 2003 UTC
# Line 1  Line 1 
1  # Copyright (C) 2001, 2002 by Intevation GmbH  # Copyright (C) 2001, 2002, 2003 by Intevation GmbH
2  # Authors:  # Authors:
3  # Jan-Oliver Wagner <[email protected]>  # Jan-Oliver Wagner <[email protected]>
4  # Bernhard Herzog <[email protected]>  # Bernhard Herzog <[email protected]>
# Line 13  Parser for thuban session files. Line 13  Parser for thuban session files.
13    
14  __version__ = "$Revision$"  __version__ = "$Revision$"
15    
16  import sys, string, os  import string, os
17    
18  import xml.sax  import xml.sax
19  import xml.sax.handler  import xml.sax.handler
20  from xml.sax import make_parser, ErrorHandler  from xml.sax import make_parser, ErrorHandler, SAXNotRecognizedException
21    
22  from Thuban import _  from Thuban import _
 from Thuban.common import *  
23    
24  from Thuban.Model.table import FIELDTYPE_INT, FIELDTYPE_DOUBLE, \  from Thuban.Model.table import FIELDTYPE_INT, FIELDTYPE_DOUBLE, \
25       FIELDTYPE_STRING       FIELDTYPE_STRING
26    
27    from Thuban.Model.color import Color, Transparent
28    
29  from Thuban.Model.session import Session  from Thuban.Model.session import Session
30  from Thuban.Model.map import Map  from Thuban.Model.map import Map
31  from Thuban.Model.layer import Layer  from Thuban.Model.layer import Layer, RasterLayer
 from Thuban.Model.color import Color  
32  from Thuban.Model.proj import Projection  from Thuban.Model.proj import Projection
33    from Thuban.Model.range import Range
34  from Thuban.Model.classification import Classification, \  from Thuban.Model.classification import Classification, \
35      ClassGroupDefault, ClassGroupSingleton, ClassGroupRange, ClassGroupMap, \      ClassGroupDefault, ClassGroupSingleton, ClassGroupRange, ClassGroupMap, \
36      ClassGroupProperties      ClassGroupProperties
37    from Thuban.Model.data import DerivedShapeStore, ShapefileStore
38    from Thuban.Model.table import DBFTable
39    from Thuban.Model.transientdb import TransientJoinedTable
40    
41    from Thuban.Model.xmlreader import XMLReader
42    import resource
43    
44    import postgisdb
45    
46    class LoadError(Exception):
47    
48        """Exception raised when the thuban file is corrupted
49    
50        Not all cases of corrupted thuban files will lead to this exception
51        but those that are found by checks in the loading code itself are.
52        """
53    
54    
55    class LoadCancelled(Exception):
56    
57        """Exception raised to indicate that loading was interrupted by the user"""
58    
59    
60  def parse_color(color):  def parse_color(color):
# Line 43  def parse_color(color): Line 65  def parse_color(color):
65      """      """
66      color = string.strip(color)      color = string.strip(color)
67      if color == "None":      if color == "None":
68          result = Color.None          result = Transparent
69      elif color[0] == '#':      elif color[0] == '#':
70          if len(color) == 7:          if len(color) == 7:
71              r = string.atoi(color[1:3], 16) / 255.0              r = string.atoi(color[1:3], 16) / 255.0
# Line 57  def parse_color(color): Line 79  def parse_color(color):
79          raise ValueError(_("Invalid color specification %s") % color)          raise ValueError(_("Invalid color specification %s") % color)
80      return result      return result
81    
82    class AttrDesc:
83    
84  class ProcessSession(xml.sax.handler.ContentHandler):      def __init__(self, name, required = False, default = "",
85                     conversion = None):
86            if not isinstance(name, tuple):
87                fullname = (None, name)
88            else:
89                fullname = name
90                name = name[1]
91            self.name = name
92            self.fullname = fullname
93            self.required = required
94            self.default = default
95            self.conversion = conversion
96    
97            # set by the SessionLoader's check_attrs method
98            self.value = None
99    
100    
101    class SessionLoader(XMLReader):
102    
103        def __init__(self, db_connection_callback = None):
104            """Inititialize the Sax handler."""
105            XMLReader.__init__(self)
106    
107            self.db_connection_callback = db_connection_callback
108    
     # Dictionary mapping element names (or (URI, element name) pairs for  
     # documents using namespaces) to method names. The methods should  
     # accept the same parameters as the startElement (or startElementNS)  
     # methods. The start_dispatcher is used by the default startElement  
     # and startElementNS methods to call a method for the open tag of an  
     # element.  
     start_dispatcher = {}  
   
     # end_dispatcher works just like start_dispatcher but it's used by  
     # endElement and endElementNS. The method whose names it maps to  
     # should accept the same parameters as endElement and endElementNS.  
     end_dispatcher = {}  
   
   
     def __init__(self, directory):  
         """Inititialize the Sax handler.  
   
         The directory parameter should be the directory containing the  
         session file. It's needed to interpret embedded relative  
         filenames.  
         """  
         self.directory = directory  
         self.chars = ''  
109          self.theSession = None          self.theSession = None
110          self.aMap = None          self.aMap = None
111          self.aLayer = None          self.aLayer = None
112    
113      def startElementNS(self, name, qname, attrs):          # Map ids used in the thuban file to the corresponding objects
114          """Call the method given for name in self.start_dispatcher          # in the session
115          """          self.idmap = {}
116          if name[0] is None:  
117              method_name = self.start_dispatcher.get(name[1])          dispatchers = {
118          else:              'session'       : ("start_session",        "end_session"),
119              # Dispatch with namespace  
120              method_name = self.start_dispatcher.get(name)              'dbconnection': ("start_dbconnection", None),
121          if method_name is not None:  
122              getattr(self, method_name)(name, qname, attrs)              'dbshapesource': ("start_dbshapesource", None),
123                'fileshapesource': ("start_fileshapesource", None),
124      def endElementNS(self, name, qname):              'derivedshapesource': ("start_derivedshapesource", None),
125          """Call the method given for name in self.end_dispatcher              'filetable': ("start_filetable", None),
126                'jointable': ("start_jointable", None),
127    
128                'map'           : ("start_map",            "end_map"),
129                'projection'    : ("start_projection",     "end_projection"),
130                'parameter'     : ("start_parameter",      None),
131                'layer'         : ("start_layer",          "end_layer"),
132                'rasterlayer'   : ("start_rasterlayer",    "end_rasterlayer"),
133                'classification': ("start_classification", "end_classification"),
134                'clnull'        : ("start_clnull",         "end_clnull"),
135                'clpoint'       : ("start_clpoint",        "end_clpoint"),
136                'clrange'       : ("start_clrange",        "end_clrange"),
137                'cldata'        : ("start_cldata",         "end_cldata"),
138                'table'         : ("start_table",          "end_table"),
139                'labellayer'    : ("start_labellayer",     None),
140                'label'         : ("start_label",          None)}
141    
142            # all dispatchers should be used for the 0.8 and 0.9 namespaces too
143            for xmlns in ("http://thuban.intevation.org/dtds/thuban-0.8.dtd",
144                          "http://thuban.intevation.org/dtds/thuban-0.9-dev.dtd",
145                          "http://thuban.intevation.org/dtds/thuban-0.9.dtd",
146                          "http://thuban.intevation.org/dtds/thuban-1.0-dev.dtd"):
147                for key, value in dispatchers.items():
148                    dispatchers[(xmlns, key)] = value
149    
150            XMLReader.AddDispatchers(self, dispatchers)
151    
152        def Destroy(self):
153            """Clear all instance variables to cut cyclic references.
154    
155            The GC would have collected the loader eventually but it can
156            happen that it doesn't run at all until Thuban is closed (2.3
157            but not 2.2 tries a bit harder and forces a collection when the
158            interpreter terminates)
159          """          """
160          if name[0] is None:          self.__dict__.clear()
             method_name = self.end_dispatcher.get(name[1])  
         else:  
             # Dispatch with namespace  
             method_name = self.end_dispatcher.get(name)  
         if method_name is not None:  
             getattr(self, method_name)(name, qname)  
161    
162      def start_session(self, name, qname, attrs):      def start_session(self, name, qname, attrs):
163          self.theSession = Session(attrs.get((None, 'title'), None))          self.theSession = Session(self.encode(attrs.get((None, 'title'),
164      start_dispatcher['session'] = "start_session"                                                          None)))
165    
166      def end_session(self, name, qname):      def end_session(self, name, qname):
167          pass          pass
168      end_dispatcher['session'] = "end_session"  
169        def check_attrs(self, element, attrs, descr):
170            """Check and convert some of the attributes of an element
171    
172            Parameters:
173               element -- The element name
174               attrs -- The attrs mapping as passed to the start_* methods
175               descr -- Sequence of attribute descriptions (AttrDesc instances)
176    
177            Return a dictionary containig normalized versions of the
178            attributes described in descr. The keys of that dictionary are
179            the name attributes of the attribute descriptions. The attrs
180            dictionary will not be modified.
181    
182            If the attribute is required, i.e. the 'required' attribute of
183            the descrtiption is true, but it is not in attrs, raise a
184            LoadError.
185    
186            If the attribute has a default value and it is not present in
187            attrs, use that default value as the value in the returned dict.
188    
189            The value is converted before putting it into the returned dict.
190            The following conversions are available:
191    
192               'filename' -- The attribute is a filename.
193    
194                             If the filename is a relative name, interpret
195                             it relative to the directory containing the
196                             .thuban file and make it an absolute name
197    
198               'shapestore' -- The attribute is the ID of a shapestore
199                               defined earlier in the .thuban file. Look it
200                               up self.idmap
201    
202               'table' -- The attribute is the ID of a table or shapestore
203                          defined earlier in the .thuban file. Look it up
204                          self.idmap. If it's the ID of a shapestore the
205                          value will be the table of the shapestore.
206    
207               'idref' -- The attribute is the id of an object defined
208                          earlier in the .thuban file. Look it up self.idmap
209    
210               'ascii' -- The attribute is converted to a bytestring with
211                          ascii encoding.
212    
213               a callable -- The attribute value is passed to the callable
214                             and the return value is used as the converted
215                             value
216    
217            If no conversion is specified for an attribute it is converted
218            with self.encode.
219            """
220            normalized = {}
221    
222            for d in descr:
223                if d.required and not attrs.has_key(d.fullname):
224                    raise LoadError("Element %s requires an attribute %r"
225                                    % (element, d.name))
226                value = attrs.get(d.fullname, d.default)
227    
228                if d.conversion in ("idref", "shapesource"):
229                    if value in self.idmap:
230                        value = self.idmap[value]
231                    else:
232                        raise LoadError("Element %s requires an already defined ID"
233                                        " in attribute %r"
234                                        % (element, d.name))
235                elif d.conversion == "table":
236                    if value in self.idmap:
237                        value = self.idmap[value]
238                        if isinstance(value, ShapefileStore):
239                            value = value.Table()
240                    else:
241                        raise LoadError("Element %s requires an already defined ID"
242                                        " in attribute %r"
243                                        % (element, d.name))
244                elif d.conversion == "filename":
245                    value = os.path.abspath(os.path.join(self.GetDirectory(),
246                                                         value))
247                elif d.conversion == "ascii":
248                    value = value.encode("ascii")
249                elif d.conversion:
250                    # Assume it's a callable
251                    value = d.conversion(value)
252                else:
253                   value = self.encode(value)
254    
255                normalized[d.name] = value
256            return normalized
257    
258        def start_dbconnection(self, name, qname, attrs):
259            attrs = self.check_attrs(name, attrs,
260                                     [AttrDesc("id", True),
261                                      AttrDesc("dbtype", True),
262                                      AttrDesc("host", False, ""),
263                                      AttrDesc("port", False, ""),
264                                      AttrDesc("user", False, ""),
265                                      AttrDesc("dbname", True)])
266            ID = attrs["id"]
267            dbtype = attrs["dbtype"]
268            if dbtype != "postgis":
269                raise LoadError("dbtype %r not supported" % filetype)
270    
271            del attrs["id"]
272            del attrs["dbtype"]
273    
274            # Try to open the connection and if it fails ask the user for
275            # the correct parameters repeatedly.
276            # FIXME: it would be better not to insist on getting a
277            # connection here. We should handle this more like the raster
278            # images where the layers etc still are created but are not
279            # drawn in case Thuban can't use the data for various reasons
280            while 1:
281                try:
282                    conn = postgisdb.PostGISConnection(**attrs)
283                    break
284                except postgisdb.ConnectionError, val:
285                    if self.db_connection_callback is not None:
286                        attrs = self.db_connection_callback(attrs, str(val))
287                        if attrs is None:
288                            raise LoadCancelled
289                    else:
290                        raise
291    
292            self.idmap[ID] = conn
293            self.theSession.AddDBConnection(conn)
294    
295        def start_dbshapesource(self, name, qname, attrs):
296            attrs = self.check_attrs(name, attrs,
297                                     [AttrDesc("id", True),
298                                      AttrDesc("dbconn", True,
299                                               conversion = "idref"),
300                                      AttrDesc("tablename", True,
301                                               conversion = "ascii")])
302            ID = attrs["id"]
303            db = attrs["dbconn"]
304            tablename = attrs["tablename"]
305            self.idmap[ID] = self.theSession.OpenDBShapeStore(db, tablename)
306    
307        def start_fileshapesource(self, name, qname, attrs):
308            attrs = self.check_attrs(name, attrs,
309                                      [AttrDesc("id", True),
310                                       AttrDesc("filename", True,
311                                                conversion = "filename"),
312                                       AttrDesc("filetype", True)])
313            ID = attrs["id"]
314            filename = attrs["filename"]
315            filetype = attrs["filetype"]
316            if filetype != "shapefile":
317                raise LoadError("shapesource filetype %r not supported" % filetype)
318            self.idmap[ID] = self.theSession.OpenShapefile(filename)
319    
320        def start_derivedshapesource(self, name, qname, attrs):
321            attrs = self.check_attrs(name, attrs,
322                                     [AttrDesc("id", True),
323                                      AttrDesc("shapesource", True,
324                                               conversion = "shapesource"),
325                                      AttrDesc("table", True, conversion="table")])
326            store = DerivedShapeStore(attrs["shapesource"], attrs["table"])
327            self.theSession.AddShapeStore(store)
328            self.idmap[attrs["id"]] = store
329    
330        def start_filetable(self, name, qname, attrs):
331            attrs = self.check_attrs(name, attrs,
332                                     [AttrDesc("id", True),
333                                      AttrDesc("title", True),
334                                      AttrDesc("filename", True,
335                                               conversion = "filename"),
336                                      AttrDesc("filetype")])
337            filetype = attrs["filetype"]
338            if filetype != "DBF":
339                raise LoadError("shapesource filetype %r not supported" % filetype)
340            table = DBFTable(attrs["filename"])
341            table.SetTitle(attrs["title"])
342            self.idmap[attrs["id"]] = self.theSession.AddTable(table)
343    
344        def start_jointable(self, name, qname, attrs):
345            attrs = self.check_attrs(name, attrs,
346                                     [AttrDesc("id", True),
347                                      AttrDesc("title", True),
348                                      AttrDesc("left", True, conversion="table"),
349                                      AttrDesc("leftcolumn", True),
350                                      AttrDesc("right", True, conversion="table"),
351                                      AttrDesc("rightcolumn", True),
352    
353                                      # jointype is required for file
354                                      # version 0.9 but this attribute
355                                      # wasn't in the 0.8 version because of
356                                      # an oversight so we assume it's
357                                      # optional since we want to handle
358                                      # both file format versions here.
359                                      AttrDesc("jointype", False,
360                                               default="INNER")])
361    
362            jointype = attrs["jointype"]
363            if jointype == "LEFT OUTER":
364                outer_join = True
365            elif jointype == "INNER":
366                outer_join = False
367            else:
368                raise LoadError("jointype %r not supported" % jointype )
369            table = TransientJoinedTable(self.theSession.TransientDB(),
370                                         attrs["left"], attrs["leftcolumn"],
371                                         attrs["right"], attrs["rightcolumn"],
372                                         outer_join = outer_join)
373            table.SetTitle(attrs["title"])
374            self.idmap[attrs["id"]] = self.theSession.AddTable(table)
375    
376      def start_map(self, name, qname, attrs):      def start_map(self, name, qname, attrs):
377          """Start a map."""          """Start a map."""
378          self.aMap = Map(attrs.get((None, 'title'), None))          self.aMap = Map(self.encode(attrs.get((None, 'title'), None)))
     start_dispatcher['map'] = "start_map"  
379    
380      def end_map(self, name, qname):      def end_map(self, name, qname):
381          self.theSession.AddMap(self.aMap)          self.theSession.AddMap(self.aMap)
382      end_dispatcher['map'] = "end_map"          self.aMap = None
383    
384      def start_projection(self, name, qname, attrs):      def start_projection(self, name, qname, attrs):
385          self.ProjectionParams = [ ]          attrs = self.check_attrs(name, attrs,
386      start_dispatcher['projection'] = "start_projection"                                   [AttrDesc("name", conversion=self.encode),
387                                      AttrDesc("epsg", default=None,
388                                               conversion=self.encode)])
389            self.projection_name = attrs["name"]
390            self.projection_epsg = attrs["epsg"]
391            self.projection_params = [ ]
392    
393      def end_projection(self, name, qname):      def end_projection(self, name, qname):
394          self.aMap.SetProjection(Projection(self.ProjectionParams))          if self.aLayer is not None:
395      end_dispatcher['projection'] = "end_projection"              obj = self.aLayer
396            elif self.aMap is not None:
397                obj = self.aMap
398            else:
399                assert False, "projection tag out of context"
400                pass
401    
402            obj.SetProjection(Projection(self.projection_params,
403                                         self.projection_name,
404                                         epsg = self.projection_epsg))
405    
406      def start_parameter(self, name, qname, attrs):      def start_parameter(self, name, qname, attrs):
407          s = attrs.get((None, 'value'))          s = attrs.get((None, 'value'))
408          s = str(s) # we can't handle unicode in proj          s = str(s) # we can't handle unicode in proj
409          self.ProjectionParams.append(s)          self.projection_params.append(s)
     start_dispatcher['parameter'] = "start_parameter"  
410    
411      def start_layer(self, name, qname, attrs, layer_class = Layer):      def start_layer(self, name, qname, attrs, layer_class = Layer):
412          """Start a layer          """Start a layer
# Line 147  class ProcessSession(xml.sax.handler.Con Line 415  class ProcessSession(xml.sax.handler.Con
415          attrs which may be a dictionary as well as the normal SAX attrs          attrs which may be a dictionary as well as the normal SAX attrs
416          object and bind it to self.aLayer.          object and bind it to self.aLayer.
417          """          """
418          title = attrs.get((None, 'title'), "")          title = self.encode(attrs.get((None, 'title'), ""))
419          filename = attrs.get((None, 'filename'), "")          filename = attrs.get((None, 'filename'), "")
420          filename = os.path.join(self.directory, filename)          filename = os.path.join(self.GetDirectory(), filename)
421            filename = self.encode(filename)
422            visible  = self.encode(attrs.get((None, 'visible'), "true")) != "false"
423          fill = parse_color(attrs.get((None, 'fill'), "None"))          fill = parse_color(attrs.get((None, 'fill'), "None"))
424          stroke = parse_color(attrs.get((None, 'stroke'), "#000000"))          stroke = parse_color(attrs.get((None, 'stroke'), "#000000"))
425          stroke_width = int(attrs.get((None, 'stroke_width'), "1"))          stroke_width = int(attrs.get((None, 'stroke_width'), "1"))
426          self.aLayer = layer_class(title, filename, fill = fill,          if attrs.has_key((None, "shapestore")):
427                                    stroke = stroke, lineWidth = stroke_width)              store = self.idmap[attrs[(None, "shapestore")]]
428      start_dispatcher['layer'] = "start_layer"          else:
429                store = self.theSession.OpenShapefile(filename)
430            self.aLayer = layer_class(title, store,
431                                      fill = fill, stroke = stroke,
432                                      lineWidth = stroke_width,
433                                      visible = visible)
434    
435      def end_layer(self, name, qname):      def end_layer(self, name, qname):
436          self.aMap.AddLayer(self.aLayer)          self.aMap.AddLayer(self.aLayer)
437      end_dispatcher['layer'] = "end_layer"          self.aLayer = None
438    
439        def start_rasterlayer(self, name, qname, attrs, layer_class = RasterLayer):
440            title = self.encode(attrs.get((None, 'title'), ""))
441            filename = attrs.get((None, 'filename'), "")
442            filename = os.path.join(self.GetDirectory(), filename)
443            filename = self.encode(filename)
444            visible  = self.encode(attrs.get((None, 'visible'), "true")) != "false"
445    
446            self.aLayer = layer_class(title, filename, visible = visible)
447    
448        def end_rasterlayer(self, name, qname):
449            self.aMap.AddLayer(self.aLayer)
450            self.aLayer = None
451    
452      def start_classification(self, name, qname, attrs):      def start_classification(self, name, qname, attrs):
453          field = attrs.get((None, 'field'), None)          attrs = self.check_attrs(name, attrs,
454                                     [AttrDesc("field", True),
455                                      AttrDesc("field_type", True)])
456            field = attrs["field"]
457            fieldType = attrs["field_type"]
458    
         fieldType = attrs.get((None, 'field_type'), None)  
459          dbFieldType = self.aLayer.GetFieldType(field)          dbFieldType = self.aLayer.GetFieldType(field)
460    
461          if fieldType != dbFieldType:          if fieldType != dbFieldType:
# Line 179  class ProcessSession(xml.sax.handler.Con Line 470  class ProcessSession(xml.sax.handler.Con
470          elif fieldType == FIELDTYPE_DOUBLE:          elif fieldType == FIELDTYPE_DOUBLE:
471              self.conv = float              self.conv = float
472    
473          self.aLayer.GetClassification().SetField(field)          self.aLayer.SetClassificationColumn(field)
   
     start_dispatcher['classification'] = "start_classification"  
474    
475      def end_classification(self, name, qname):      def end_classification(self, name, qname):
476          pass          pass
     end_dispatcher['classification'] = "end_classification"  
477    
478      def start_clnull(self, name, qname, attrs):      def start_clnull(self, name, qname, attrs):
479          self.cl_group = ClassGroupDefault()          self.cl_group = ClassGroupDefault()
480          self.cl_group.SetLabel(attrs.get((None, 'label'), ""))          self.cl_group.SetLabel(self.encode(attrs.get((None, 'label'), "")))
481          self.cl_prop = ClassGroupProperties()          self.cl_prop = ClassGroupProperties()
     start_dispatcher['clnull'] = "start_clnull"  
482    
483      def end_clnull(self, name, qname):      def end_clnull(self, name, qname):
484          self.cl_group.SetProperties(self.cl_prop)          self.cl_group.SetProperties(self.cl_prop)
485          self.aLayer.GetClassification().SetDefaultGroup(self.cl_group)          self.aLayer.GetClassification().SetDefaultGroup(self.cl_group)
486          del self.cl_group, self.cl_prop          del self.cl_group, self.cl_prop
     end_dispatcher['clnull'] = "end_clnull"  
487    
488      def start_clpoint(self, name, qname, attrs):      def start_clpoint(self, name, qname, attrs):
489          attrib_value = attrs.get((None, 'value'), "0")          attrib_value = attrs.get((None, 'value'), "0")
490    
491          #try:          field = self.aLayer.GetClassificationColumn()
492              #value  = Str2Num(attrib_value)          if self.aLayer.GetFieldType(field) == FIELDTYPE_STRING:
493          #except:              value = self.encode(attrib_value)
494              #value  = attrib_value          else:
495                value = self.conv(attrib_value)
         value = self.conv(attrib_value)  
   
496          self.cl_group = ClassGroupSingleton(value)          self.cl_group = ClassGroupSingleton(value)
497          self.cl_group.SetLabel(attrs.get((None, 'label'), ""))          self.cl_group.SetLabel(self.encode(attrs.get((None, 'label'), "")))
498          self.cl_prop = ClassGroupProperties()          self.cl_prop = ClassGroupProperties()
499    
     start_dispatcher['clpoint'] = "start_clpoint"  
500    
501      def end_clpoint(self, name, qname):      def end_clpoint(self, name, qname):
502          self.cl_group.SetProperties(self.cl_prop)          self.cl_group.SetProperties(self.cl_prop)
503          self.aLayer.GetClassification().AddGroup(self.cl_group)          self.aLayer.GetClassification().AppendGroup(self.cl_group)
504          del self.cl_group, self.cl_prop          del self.cl_group, self.cl_prop
     end_dispatcher['clpoint'] = "end_clpoint"  
505    
506      def start_clrange(self, name, qname, attrs):      def start_clrange(self, name, qname, attrs):
507    
508            range = attrs.get((None, 'range'), None)
509            # for backward compatibility (min/max are not saved)
510            min   = attrs.get((None, 'min'), None)
511            max   = attrs.get((None, 'max'), None)
512    
513          try:          try:
514              min = self.conv(attrs.get((None, 'min'), "0"))              if range is not None:
515              max = self.conv(attrs.get((None, 'max'), "0"))                  self.cl_group = ClassGroupRange(Range(range))
516              #min = Str2Num(attrs.get((None, 'min'), "0"))              elif min is not None and max is not None:
517              #max = Str2Num(attrs.get((None, 'max'), "0"))                  self.cl_group = ClassGroupRange((self.conv(min),
518                                                     self.conv(max)))
519                else:
520                    self.cl_group = ClassGroupRange(Range(None))
521    
522          except ValueError:          except ValueError:
523              raise ValueError(_("Classification range is not a number!"))              raise ValueError(_("Classification range is not a number!"))
524    
         self.cl_group = ClassGroupRange(min, max)  
525          self.cl_group.SetLabel(attrs.get((None, 'label'), ""))          self.cl_group.SetLabel(attrs.get((None, 'label'), ""))
526          self.cl_prop = ClassGroupProperties()          self.cl_prop = ClassGroupProperties()
527    
     start_dispatcher['clrange'] = "start_clrange"  
528    
529      def end_clrange(self, name, qname):      def end_clrange(self, name, qname):
530          self.cl_group.SetProperties(self.cl_prop)          self.cl_group.SetProperties(self.cl_prop)
531          self.aLayer.GetClassification().AddGroup(self.cl_group)          self.aLayer.GetClassification().AppendGroup(self.cl_group)
532          del self.cl_group, self.cl_prop          del self.cl_group, self.cl_prop
     end_dispatcher['clrange'] = "end_clrange"  
533    
534      def start_cldata(self, name, qname, attrs):      def start_cldata(self, name, qname, attrs):
535          self.cl_prop.SetLineColor(          self.cl_prop.SetLineColor(
# Line 249  class ProcessSession(xml.sax.handler.Con Line 537  class ProcessSession(xml.sax.handler.Con
537          self.cl_prop.SetLineWidth(          self.cl_prop.SetLineWidth(
538              int(attrs.get((None, 'stroke_width'), "0")))              int(attrs.get((None, 'stroke_width'), "0")))
539          self.cl_prop.SetFill(parse_color(attrs.get((None, 'fill'), "None")))          self.cl_prop.SetFill(parse_color(attrs.get((None, 'fill'), "None")))
     start_dispatcher['cldata'] = "start_cldata"  
540    
541      def end_cldata(self, name, qname):      def end_cldata(self, name, qname):
542          pass          pass
     end_dispatcher['cldata'] = "end_cldata"  
   
     def start_table(self, name, qname, attrs):  
         #print "table title: %s" % attrs.get('title', None)  
         pass  
     start_dispatcher['table'] = "start_table"  
   
     def end_table(self, name, qname):  
         pass  
     end_dispatcher['table'] = "end_table"  
543    
544      def start_labellayer(self, name, qname, attrs):      def start_labellayer(self, name, qname, attrs):
545          self.aLayer = self.aMap.LabelLayer()          self.aLayer = self.aMap.LabelLayer()
     start_dispatcher['labellayer'] = "start_labellayer"  
546    
547      def start_label(self, name, qname, attrs):      def start_label(self, name, qname, attrs):
548          x = float(attrs[(None, 'x')])          x = float(attrs[(None, 'x')])
549          y = float(attrs[(None, 'y')])          y = float(attrs[(None, 'y')])
550          text = attrs[(None, 'text')]          text = self.encode(attrs[(None, 'text')])
551          halign = attrs[(None, 'halign')]          halign = attrs[(None, 'halign')]
552          valign = attrs[(None, 'valign')]          valign = attrs[(None, 'valign')]
553          self.aLayer.AddLabel(x, y, text, halign = halign, valign = valign)          self.aLayer.AddLabel(x, y, text, halign = halign, valign = valign)
     start_dispatcher['label'] = "start_label"  
554    
555      def characters(self, chars):      def characters(self, chars):
556          pass          pass
557    
558    
559  def load_session(filename):  def load_session(filename, db_connection_callback = None):
560      """Load a Thuban session from the file object file"""      """Load a Thuban session from the file object file
561      dir = os.path.dirname(filename)  
562      file = open(filename)      The db_connection_callback, if given should be a callable object
563      handler = ProcessSession(dir)      that can be called like this:
564           db_connection_callback(params, message)
565      parser = make_parser()  
566      parser.setContentHandler(handler)      where params is a dictionary containing the known connection
567      parser.setErrorHandler(ErrorHandler())      parameters and message is a string with a message why the connection
568      parser.setFeature(xml.sax.handler.feature_namespaces, 1)      failed. db_connection_callback should return a new dictionary with
569      parser.parse(file)      corrected and perhaps additional parameters like a password or None
570        to indicate that the user cancelled.
571        """
572        handler = SessionLoader(db_connection_callback)
573        handler.read(filename)
574    
575      session = handler.theSession      session = handler.theSession
576      # Newly loaded session aren't modified      # Newly loaded session aren't modified
577      session.UnsetModified()      session.UnsetModified()
578    
579        handler.Destroy()
580    
581      return session      return session
582    
Legend:
Removed from v.493  
changed lines
  Added in v.1970
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26