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]> |
5 |
|
# Jonathan Coles <[email protected]> |
6 |
# |
# |
7 |
# This program is free software under the GPL (>=v2) |
# This program is free software under the GPL (>=v2) |
8 |
# Read the file COPYING coming with GRASS for details. |
# Read the file COPYING coming with GRASS for details. |
13 |
|
|
14 |
__version__ = "$Revision$" |
__version__ = "$Revision$" |
15 |
|
|
16 |
import sys, string, os |
import string, os |
17 |
|
|
18 |
|
import xml.sax |
19 |
|
import xml.sax.handler |
20 |
|
from xml.sax import make_parser, ErrorHandler, SAXNotRecognizedException |
21 |
|
|
22 |
|
from Thuban import _ |
23 |
|
|
24 |
|
from Thuban.Model.table import FIELDTYPE_INT, FIELDTYPE_DOUBLE, \ |
25 |
|
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, \ |
35 |
|
ClassGroupDefault, ClassGroupSingleton, ClassGroupRange, ClassGroupMap, \ |
36 |
|
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 |
oldPython=0 |
from Thuban.Model.xmlreader import XMLReader |
42 |
|
import resource |
43 |
|
|
44 |
if not sys.__dict__.has_key("version_info"): |
import postgisdb |
|
# We can assume to have python 1.5.2 or lower here now |
|
|
oldPython=1 |
|
|
|
|
|
if oldPython: |
|
|
try: |
|
|
from xml.sax.saxexts import make_parser |
|
|
from xml.sax.saxlib import HandlerBase |
|
|
from xml.sax import saxutils |
|
|
except ImportError: |
|
|
sys.stdout.write(("You need to have Python-XML installed or" |
|
|
" a modern Python!\n" |
|
|
"Check www.python.org/sigs/xml-sig/\n\n")) |
|
|
raise |
|
|
else: |
|
|
# Do the python 2.0 standard xml thing and map it on the old names |
|
|
import xml.sax |
|
|
import xml.sax.handler |
|
|
HandlerBase=xml.sax.handler.ContentHandler |
|
|
from xml.sax import make_parser |
|
|
|
|
|
class testSAXContentHandler(HandlerBase): |
|
|
# SAX compliant |
|
|
def characters(self, ch, start, length): |
|
|
pass |
|
|
|
|
|
def test_for_broken_SAX(): |
|
|
ch=testSAXContentHandler() |
|
|
try: |
|
|
xml.sax.parseString("""<?xml version="1.0"?> |
|
|
<child1 name="paul">Text goes here</child1> |
|
|
""",ch) |
|
|
except TypeError: |
|
|
return 1 |
|
|
return 0 |
|
45 |
|
|
46 |
|
class LoadError(Exception): |
47 |
|
|
48 |
def parse_color(color): |
"""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 |
Return the color object for the string color. Color may be either |
|
54 |
'None' or of the form '#RRGGBB' in the usual HTML color notation |
|
55 |
|
class LoadCancelled(Exception): |
56 |
|
|
57 |
|
"""Exception raised to indicate that loading was interrupted by the user""" |
58 |
|
|
59 |
|
|
60 |
|
def parse_color(color): |
61 |
|
"""Return the color object for the string color. |
62 |
|
|
63 |
|
Color may be either 'None' or of the form '#RRGGBB' in the usual |
64 |
|
HTML color notation |
65 |
""" |
""" |
66 |
color = string.strip(color) |
color = string.strip(color) |
67 |
if color == "None": |
if color == "None": |
68 |
result = 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 |
73 |
b = string.atoi(color[5:7], 16) / 255.0 |
b = string.atoi(color[5:7], 16) / 255.0 |
74 |
result = Color(r, g, b) |
result = Color(r, g, b) |
75 |
else: |
else: |
76 |
raise ValueError("Invalid hexadecimal color specification %s" |
raise ValueError(_("Invalid hexadecimal color specification %s") |
77 |
% color) |
% color) |
78 |
else: |
else: |
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(HandlerBase): |
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 |
def __init__(self, directory): |
# set by the SessionLoader's check_attrs method |
98 |
"""Inititialize the Sax handler. |
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 |
|
|
|
directory is 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 startElement(self, name, attrs): |
# Map ids used in the thuban file to the corresponding objects |
114 |
if name == 'session': |
# in the session |
115 |
self.theSession = Session(attrs.get('title', None)) |
self.idmap = {} |
116 |
elif name == 'map': |
|
117 |
self.aMap = Map(attrs.get('title', None)) |
dispatchers = { |
118 |
elif name == 'projection': |
'session' : ("start_session", "end_session"), |
119 |
self.ProjectionParams = [ ] |
|
120 |
elif name == 'parameter': |
'dbconnection': ("start_dbconnection", None), |
121 |
self.ProjectionParams.append(attrs.get('value', None)) |
|
122 |
elif name == 'layer': |
'dbshapesource': ("start_dbshapesource", None), |
123 |
title = attrs.get('title', "") |
'fileshapesource': ("start_fileshapesource", None), |
124 |
filename = attrs.get('filename', "") |
'derivedshapesource': ("start_derivedshapesource", None), |
125 |
filename = os.path.join(self.directory, filename) |
'filetable': ("start_filetable", None), |
126 |
fill = parse_color(attrs.get('fill', "None")) |
'jointable': ("start_jointable", None), |
127 |
stroke = parse_color(attrs.get('stroke', "#000000")) |
|
128 |
stroke_width = int(attrs.get("stroke_width", "1")) |
'map' : ("start_map", "end_map"), |
129 |
self.aLayer = Layer(title, filename, fill = fill, stroke = stroke, |
'projection' : ("start_projection", "end_projection"), |
130 |
stroke_width = stroke_width) |
'parameter' : ("start_parameter", None), |
131 |
elif name == 'table': |
'layer' : ("start_layer", "end_layer"), |
132 |
print "table title: %s" % attrs.get('title', None) |
'rasterlayer' : ("start_rasterlayer", "end_rasterlayer"), |
133 |
elif name == 'labellayer': |
'classification': ("start_classification", "end_classification"), |
134 |
self.aLayer = self.aMap.LabelLayer() |
'clnull' : ("start_clnull", "end_clnull"), |
135 |
elif name == 'label': |
'clpoint' : ("start_clpoint", "end_clpoint"), |
136 |
x = float(attrs['x']) |
'clrange' : ("start_clrange", "end_clrange"), |
137 |
y = float(attrs['y']) |
'cldata' : ("start_cldata", "end_cldata"), |
138 |
text = attrs['text'] |
'table' : ("start_table", "end_table"), |
139 |
halign = attrs['halign'] |
'labellayer' : ("start_labellayer", None), |
140 |
valign = attrs['valign'] |
'label' : ("start_label", None)} |
141 |
self.aLayer.AddLabel(x, y, text, halign = halign, valign = valign) |
|
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 |
if not oldPython and test_for_broken_SAX(): |
"http://thuban.intevation.org/dtds/thuban-0.9-dev.dtd", |
145 |
# works with python 2.0, but is not SAX compliant |
"http://thuban.intevation.org/dtds/thuban-0.9.dtd", |
146 |
def characters(self, ch): |
"http://thuban.intevation.org/dtds/thuban-1.0-dev.dtd", |
147 |
self.my_characters(ch) |
"http://thuban.intevation.org/dtds/thuban-1.0rc1.dtd"): |
148 |
else: |
for key, value in dispatchers.items(): |
149 |
# SAX compliant |
dispatchers[(xmlns, key)] = value |
150 |
def characters(self, ch, start, length): |
|
151 |
self.my_characters(ch[start:start+length]) |
XMLReader.AddDispatchers(self, dispatchers) |
152 |
|
|
153 |
def my_characters(self, ch): |
def Destroy(self): |
154 |
self.chars = self.chars + ch |
"""Clear all instance variables to cut cyclic references. |
155 |
|
|
156 |
def endElement(self, name): |
The GC would have collected the loader eventually but it can |
157 |
# If it's not a parameter element, ignore it |
happen that it doesn't run at all until Thuban is closed (2.3 |
158 |
if name == 'session': |
but not 2.2 tries a bit harder and forces a collection when the |
159 |
#print "end of session" |
interpreter terminates) |
160 |
pass |
""" |
161 |
if name == 'map': |
self.__dict__.clear() |
162 |
self.theSession.AddMap(self.aMap) |
|
163 |
if name == 'projection': |
def start_session(self, name, qname, attrs): |
164 |
self.aMap.SetProjection(Projection(self.ProjectionParams)) |
self.theSession = Session(self.encode(attrs.get((None, 'title'), |
165 |
if name == 'layer': |
None))) |
166 |
self.aMap.AddLayer(self.aLayer) |
|
167 |
if name == 'table': |
def end_session(self, name, qname): |
168 |
#print "end of table" |
pass |
169 |
|
|
170 |
|
def check_attrs(self, element, attrs, descr): |
171 |
|
"""Check and convert some of the attributes of an element |
172 |
|
|
173 |
|
Parameters: |
174 |
|
element -- The element name |
175 |
|
attrs -- The attrs mapping as passed to the start_* methods |
176 |
|
descr -- Sequence of attribute descriptions (AttrDesc instances) |
177 |
|
|
178 |
|
Return a dictionary containig normalized versions of the |
179 |
|
attributes described in descr. The keys of that dictionary are |
180 |
|
the name attributes of the attribute descriptions. The attrs |
181 |
|
dictionary will not be modified. |
182 |
|
|
183 |
|
If the attribute is required, i.e. the 'required' attribute of |
184 |
|
the descrtiption is true, but it is not in attrs, raise a |
185 |
|
LoadError. |
186 |
|
|
187 |
|
If the attribute has a default value and it is not present in |
188 |
|
attrs, use that default value as the value in the returned dict. |
189 |
|
|
190 |
|
The value is converted before putting it into the returned dict. |
191 |
|
The following conversions are available: |
192 |
|
|
193 |
|
'filename' -- The attribute is a filename. |
194 |
|
|
195 |
|
If the filename is a relative name, interpret |
196 |
|
it relative to the directory containing the |
197 |
|
.thuban file and make it an absolute name |
198 |
|
|
199 |
|
'shapestore' -- The attribute is the ID of a shapestore |
200 |
|
defined earlier in the .thuban file. Look it |
201 |
|
up self.idmap |
202 |
|
|
203 |
|
'table' -- The attribute is the ID of a table or shapestore |
204 |
|
defined earlier in the .thuban file. Look it up |
205 |
|
self.idmap. If it's the ID of a shapestore the |
206 |
|
value will be the table of the shapestore. |
207 |
|
|
208 |
|
'idref' -- The attribute is the id of an object defined |
209 |
|
earlier in the .thuban file. Look it up self.idmap |
210 |
|
|
211 |
|
'ascii' -- The attribute is converted to a bytestring with |
212 |
|
ascii encoding. |
213 |
|
|
214 |
|
a callable -- The attribute value is passed to the callable |
215 |
|
and the return value is used as the converted |
216 |
|
value |
217 |
|
|
218 |
|
If no conversion is specified for an attribute it is converted |
219 |
|
with self.encode. |
220 |
|
""" |
221 |
|
normalized = {} |
222 |
|
|
223 |
|
for d in descr: |
224 |
|
if d.required and not attrs.has_key(d.fullname): |
225 |
|
raise LoadError("Element %s requires an attribute %r" |
226 |
|
% (element, d.name)) |
227 |
|
value = attrs.get(d.fullname, d.default) |
228 |
|
|
229 |
|
if d.conversion in ("idref", "shapesource"): |
230 |
|
if value in self.idmap: |
231 |
|
value = self.idmap[value] |
232 |
|
else: |
233 |
|
raise LoadError("Element %s requires an already defined ID" |
234 |
|
" in attribute %r" |
235 |
|
% (element, d.name)) |
236 |
|
elif d.conversion == "table": |
237 |
|
if value in self.idmap: |
238 |
|
value = self.idmap[value] |
239 |
|
if isinstance(value, ShapefileStore): |
240 |
|
value = value.Table() |
241 |
|
else: |
242 |
|
raise LoadError("Element %s requires an already defined ID" |
243 |
|
" in attribute %r" |
244 |
|
% (element, d.name)) |
245 |
|
elif d.conversion == "filename": |
246 |
|
value = os.path.abspath(os.path.join(self.GetDirectory(), |
247 |
|
value)) |
248 |
|
elif d.conversion == "ascii": |
249 |
|
value = value.encode("ascii") |
250 |
|
elif d.conversion: |
251 |
|
# Assume it's a callable |
252 |
|
value = d.conversion(value) |
253 |
|
else: |
254 |
|
value = self.encode(value) |
255 |
|
|
256 |
|
normalized[d.name] = value |
257 |
|
return normalized |
258 |
|
|
259 |
|
def start_dbconnection(self, name, qname, attrs): |
260 |
|
attrs = self.check_attrs(name, attrs, |
261 |
|
[AttrDesc("id", True), |
262 |
|
AttrDesc("dbtype", True), |
263 |
|
AttrDesc("host", False, ""), |
264 |
|
AttrDesc("port", False, ""), |
265 |
|
AttrDesc("user", False, ""), |
266 |
|
AttrDesc("dbname", True)]) |
267 |
|
ID = attrs["id"] |
268 |
|
dbtype = attrs["dbtype"] |
269 |
|
if dbtype != "postgis": |
270 |
|
raise LoadError("dbtype %r not supported" % filetype) |
271 |
|
|
272 |
|
del attrs["id"] |
273 |
|
del attrs["dbtype"] |
274 |
|
|
275 |
|
# Try to open the connection and if it fails ask the user for |
276 |
|
# the correct parameters repeatedly. |
277 |
|
# FIXME: it would be better not to insist on getting a |
278 |
|
# connection here. We should handle this more like the raster |
279 |
|
# images where the layers etc still are created but are not |
280 |
|
# drawn in case Thuban can't use the data for various reasons |
281 |
|
while 1: |
282 |
|
try: |
283 |
|
conn = postgisdb.PostGISConnection(**attrs) |
284 |
|
break |
285 |
|
except postgisdb.ConnectionError, val: |
286 |
|
if self.db_connection_callback is not None: |
287 |
|
attrs = self.db_connection_callback(attrs, str(val)) |
288 |
|
if attrs is None: |
289 |
|
raise LoadCancelled |
290 |
|
else: |
291 |
|
raise |
292 |
|
|
293 |
|
self.idmap[ID] = conn |
294 |
|
self.theSession.AddDBConnection(conn) |
295 |
|
|
296 |
|
def start_dbshapesource(self, name, qname, attrs): |
297 |
|
attrs = self.check_attrs(name, attrs, |
298 |
|
[AttrDesc("id", True), |
299 |
|
AttrDesc("dbconn", True, |
300 |
|
conversion = "idref"), |
301 |
|
AttrDesc("tablename", True, |
302 |
|
conversion = "ascii")]) |
303 |
|
ID = attrs["id"] |
304 |
|
db = attrs["dbconn"] |
305 |
|
tablename = attrs["tablename"] |
306 |
|
self.idmap[ID] = self.theSession.OpenDBShapeStore(db, tablename) |
307 |
|
|
308 |
|
def start_fileshapesource(self, name, qname, attrs): |
309 |
|
attrs = self.check_attrs(name, attrs, |
310 |
|
[AttrDesc("id", True), |
311 |
|
AttrDesc("filename", True, |
312 |
|
conversion = "filename"), |
313 |
|
AttrDesc("filetype", True)]) |
314 |
|
ID = attrs["id"] |
315 |
|
filename = attrs["filename"] |
316 |
|
filetype = attrs["filetype"] |
317 |
|
if filetype != "shapefile": |
318 |
|
raise LoadError("shapesource filetype %r not supported" % filetype) |
319 |
|
self.idmap[ID] = self.theSession.OpenShapefile(filename) |
320 |
|
|
321 |
|
def start_derivedshapesource(self, name, qname, attrs): |
322 |
|
attrs = self.check_attrs(name, attrs, |
323 |
|
[AttrDesc("id", True), |
324 |
|
AttrDesc("shapesource", True, |
325 |
|
conversion = "shapesource"), |
326 |
|
AttrDesc("table", True, conversion="table")]) |
327 |
|
store = DerivedShapeStore(attrs["shapesource"], attrs["table"]) |
328 |
|
self.theSession.AddShapeStore(store) |
329 |
|
self.idmap[attrs["id"]] = store |
330 |
|
|
331 |
|
def start_filetable(self, name, qname, attrs): |
332 |
|
attrs = self.check_attrs(name, attrs, |
333 |
|
[AttrDesc("id", True), |
334 |
|
AttrDesc("title", True), |
335 |
|
AttrDesc("filename", True, |
336 |
|
conversion = "filename"), |
337 |
|
AttrDesc("filetype")]) |
338 |
|
filetype = attrs["filetype"] |
339 |
|
if filetype != "DBF": |
340 |
|
raise LoadError("shapesource filetype %r not supported" % filetype) |
341 |
|
table = DBFTable(attrs["filename"]) |
342 |
|
table.SetTitle(attrs["title"]) |
343 |
|
self.idmap[attrs["id"]] = self.theSession.AddTable(table) |
344 |
|
|
345 |
|
def start_jointable(self, name, qname, attrs): |
346 |
|
attrs = self.check_attrs(name, attrs, |
347 |
|
[AttrDesc("id", True), |
348 |
|
AttrDesc("title", True), |
349 |
|
AttrDesc("left", True, conversion="table"), |
350 |
|
AttrDesc("leftcolumn", True), |
351 |
|
AttrDesc("right", True, conversion="table"), |
352 |
|
AttrDesc("rightcolumn", True), |
353 |
|
|
354 |
|
# jointype is required for file |
355 |
|
# version 0.9 but this attribute |
356 |
|
# wasn't in the 0.8 version because of |
357 |
|
# an oversight so we assume it's |
358 |
|
# optional since we want to handle |
359 |
|
# both file format versions here. |
360 |
|
AttrDesc("jointype", False, |
361 |
|
default="INNER")]) |
362 |
|
|
363 |
|
jointype = attrs["jointype"] |
364 |
|
if jointype == "LEFT OUTER": |
365 |
|
outer_join = True |
366 |
|
elif jointype == "INNER": |
367 |
|
outer_join = False |
368 |
|
else: |
369 |
|
raise LoadError("jointype %r not supported" % jointype ) |
370 |
|
table = TransientJoinedTable(self.theSession.TransientDB(), |
371 |
|
attrs["left"], attrs["leftcolumn"], |
372 |
|
attrs["right"], attrs["rightcolumn"], |
373 |
|
outer_join = outer_join) |
374 |
|
table.SetTitle(attrs["title"]) |
375 |
|
self.idmap[attrs["id"]] = self.theSession.AddTable(table) |
376 |
|
|
377 |
|
def start_map(self, name, qname, attrs): |
378 |
|
"""Start a map.""" |
379 |
|
self.aMap = Map(self.encode(attrs.get((None, 'title'), None))) |
380 |
|
|
381 |
|
def end_map(self, name, qname): |
382 |
|
self.theSession.AddMap(self.aMap) |
383 |
|
self.aMap = None |
384 |
|
|
385 |
|
def start_projection(self, name, qname, attrs): |
386 |
|
attrs = self.check_attrs(name, attrs, |
387 |
|
[AttrDesc("name", conversion=self.encode), |
388 |
|
AttrDesc("epsg", default=None, |
389 |
|
conversion=self.encode)]) |
390 |
|
self.projection_name = attrs["name"] |
391 |
|
self.projection_epsg = attrs["epsg"] |
392 |
|
self.projection_params = [ ] |
393 |
|
|
394 |
|
def end_projection(self, name, qname): |
395 |
|
if self.aLayer is not None: |
396 |
|
obj = self.aLayer |
397 |
|
elif self.aMap is not None: |
398 |
|
obj = self.aMap |
399 |
|
else: |
400 |
|
assert False, "projection tag out of context" |
401 |
pass |
pass |
402 |
|
|
403 |
def load_session(filename): |
obj.SetProjection(Projection(self.projection_params, |
404 |
"""Load a Thuban session from the file object file""" |
self.projection_name, |
405 |
dir = os.path.dirname(filename) |
epsg = self.projection_epsg)) |
406 |
file = open(filename) |
|
407 |
handler = ProcessSession(dir) |
def start_parameter(self, name, qname, attrs): |
408 |
|
s = attrs.get((None, 'value')) |
409 |
if oldPython: |
s = str(s) # we can't handle unicode in proj |
410 |
parser = make_parser() |
self.projection_params.append(s) |
411 |
parser.setDocumentHandler(handler) |
|
412 |
parser.setErrorHandler(saxutils.ErrorPrinter()) |
def start_layer(self, name, qname, attrs, layer_class = Layer): |
413 |
parser.parseFile(file) |
"""Start a layer |
414 |
parser.close() |
|
415 |
else: |
Instantiate a layer of class layer_class from the attributes in |
416 |
xml.sax.parse(file,handler) |
attrs which may be a dictionary as well as the normal SAX attrs |
417 |
|
object and bind it to self.aLayer. |
418 |
|
""" |
419 |
|
title = self.encode(attrs.get((None, 'title'), "")) |
420 |
|
filename = attrs.get((None, 'filename'), "") |
421 |
|
filename = os.path.join(self.GetDirectory(), filename) |
422 |
|
filename = self.encode(filename) |
423 |
|
visible = self.encode(attrs.get((None, 'visible'), "true")) != "false" |
424 |
|
fill = parse_color(attrs.get((None, 'fill'), "None")) |
425 |
|
stroke = parse_color(attrs.get((None, 'stroke'), "#000000")) |
426 |
|
stroke_width = int(attrs.get((None, 'stroke_width'), "1")) |
427 |
|
if attrs.has_key((None, "shapestore")): |
428 |
|
store = self.idmap[attrs[(None, "shapestore")]] |
429 |
|
else: |
430 |
|
store = self.theSession.OpenShapefile(filename) |
431 |
|
self.aLayer = layer_class(title, store, |
432 |
|
fill = fill, stroke = stroke, |
433 |
|
lineWidth = stroke_width, |
434 |
|
visible = visible) |
435 |
|
|
436 |
|
def end_layer(self, name, qname): |
437 |
|
self.aMap.AddLayer(self.aLayer) |
438 |
|
self.aLayer = None |
439 |
|
|
440 |
|
def start_rasterlayer(self, name, qname, attrs, layer_class = RasterLayer): |
441 |
|
title = self.encode(attrs.get((None, 'title'), "")) |
442 |
|
filename = attrs.get((None, 'filename'), "") |
443 |
|
filename = os.path.join(self.GetDirectory(), filename) |
444 |
|
filename = self.encode(filename) |
445 |
|
visible = self.encode(attrs.get((None, 'visible'), "true")) != "false" |
446 |
|
|
447 |
|
self.aLayer = layer_class(title, filename, visible = visible) |
448 |
|
|
449 |
|
def end_rasterlayer(self, name, qname): |
450 |
|
self.aMap.AddLayer(self.aLayer) |
451 |
|
self.aLayer = None |
452 |
|
|
453 |
|
def start_classification(self, name, qname, attrs): |
454 |
|
attrs = self.check_attrs(name, attrs, |
455 |
|
[AttrDesc("field", True), |
456 |
|
AttrDesc("field_type", True)]) |
457 |
|
field = attrs["field"] |
458 |
|
fieldType = attrs["field_type"] |
459 |
|
|
460 |
|
dbFieldType = self.aLayer.GetFieldType(field) |
461 |
|
|
462 |
|
if fieldType != dbFieldType: |
463 |
|
raise ValueError(_("xml field type differs from database!")) |
464 |
|
|
465 |
|
# setup conversion routines depending on the kind of data |
466 |
|
# we will be seeing later on |
467 |
|
if fieldType == FIELDTYPE_STRING: |
468 |
|
self.conv = str |
469 |
|
elif fieldType == FIELDTYPE_INT: |
470 |
|
self.conv = lambda p: int(float(p)) |
471 |
|
elif fieldType == FIELDTYPE_DOUBLE: |
472 |
|
self.conv = float |
473 |
|
|
474 |
|
self.aLayer.SetClassificationColumn(field) |
475 |
|
|
476 |
|
def end_classification(self, name, qname): |
477 |
|
pass |
478 |
|
|
479 |
|
def start_clnull(self, name, qname, attrs): |
480 |
|
self.cl_group = ClassGroupDefault() |
481 |
|
self.cl_group.SetLabel(self.encode(attrs.get((None, 'label'), ""))) |
482 |
|
self.cl_prop = ClassGroupProperties() |
483 |
|
|
484 |
|
def end_clnull(self, name, qname): |
485 |
|
self.cl_group.SetProperties(self.cl_prop) |
486 |
|
self.aLayer.GetClassification().SetDefaultGroup(self.cl_group) |
487 |
|
del self.cl_group, self.cl_prop |
488 |
|
|
489 |
|
def start_clpoint(self, name, qname, attrs): |
490 |
|
attrib_value = attrs.get((None, 'value'), "0") |
491 |
|
|
492 |
|
field = self.aLayer.GetClassificationColumn() |
493 |
|
if self.aLayer.GetFieldType(field) == FIELDTYPE_STRING: |
494 |
|
value = self.encode(attrib_value) |
495 |
|
else: |
496 |
|
value = self.conv(attrib_value) |
497 |
|
self.cl_group = ClassGroupSingleton(value) |
498 |
|
self.cl_group.SetLabel(self.encode(attrs.get((None, 'label'), ""))) |
499 |
|
self.cl_prop = ClassGroupProperties() |
500 |
|
|
501 |
|
|
502 |
|
def end_clpoint(self, name, qname): |
503 |
|
self.cl_group.SetProperties(self.cl_prop) |
504 |
|
self.aLayer.GetClassification().AppendGroup(self.cl_group) |
505 |
|
del self.cl_group, self.cl_prop |
506 |
|
|
507 |
|
def start_clrange(self, name, qname, attrs): |
508 |
|
|
509 |
|
range = attrs.get((None, 'range'), None) |
510 |
|
# for backward compatibility (min/max are not saved) |
511 |
|
min = attrs.get((None, 'min'), None) |
512 |
|
max = attrs.get((None, 'max'), None) |
513 |
|
|
514 |
|
try: |
515 |
|
if range is not None: |
516 |
|
self.cl_group = ClassGroupRange(Range(range)) |
517 |
|
elif min is not None and max is not None: |
518 |
|
self.cl_group = ClassGroupRange((self.conv(min), |
519 |
|
self.conv(max))) |
520 |
|
else: |
521 |
|
self.cl_group = ClassGroupRange(Range(None)) |
522 |
|
|
523 |
|
except ValueError: |
524 |
|
raise ValueError(_("Classification range is not a number!")) |
525 |
|
|
526 |
|
self.cl_group.SetLabel(attrs.get((None, 'label'), "")) |
527 |
|
self.cl_prop = ClassGroupProperties() |
528 |
|
|
529 |
|
|
530 |
|
def end_clrange(self, name, qname): |
531 |
|
self.cl_group.SetProperties(self.cl_prop) |
532 |
|
self.aLayer.GetClassification().AppendGroup(self.cl_group) |
533 |
|
del self.cl_group, self.cl_prop |
534 |
|
|
535 |
|
def start_cldata(self, name, qname, attrs): |
536 |
|
self.cl_prop.SetLineColor( |
537 |
|
parse_color(attrs.get((None, 'stroke'), "None"))) |
538 |
|
self.cl_prop.SetLineWidth( |
539 |
|
int(attrs.get((None, 'stroke_width'), "0"))) |
540 |
|
self.cl_prop.SetFill(parse_color(attrs.get((None, 'fill'), "None"))) |
541 |
|
|
542 |
|
def end_cldata(self, name, qname): |
543 |
|
pass |
544 |
|
|
545 |
|
def start_labellayer(self, name, qname, attrs): |
546 |
|
self.aLayer = self.aMap.LabelLayer() |
547 |
|
|
548 |
|
def start_label(self, name, qname, attrs): |
549 |
|
attrs = self.check_attrs(name, attrs, |
550 |
|
[AttrDesc("x", True, conversion = float), |
551 |
|
AttrDesc("y", True, conversion = float), |
552 |
|
AttrDesc("text", True), |
553 |
|
AttrDesc("halign", True, |
554 |
|
conversion = "ascii"), |
555 |
|
AttrDesc("valign", True, |
556 |
|
conversion = "ascii")]) |
557 |
|
x = attrs['x'] |
558 |
|
y = attrs['y'] |
559 |
|
text = attrs['text'] |
560 |
|
halign = attrs['halign'] |
561 |
|
valign = attrs['valign'] |
562 |
|
if halign not in ("left", "center", "right"): |
563 |
|
raise LoadError("Unsupported halign value %r" % halign) |
564 |
|
if valign not in ("top", "center", "bottom"): |
565 |
|
raise LoadError("Unsupported valign value %r" % valign) |
566 |
|
self.aLayer.AddLabel(x, y, text, halign = halign, valign = valign) |
567 |
|
|
568 |
|
def characters(self, chars): |
569 |
|
pass |
570 |
|
|
571 |
|
|
572 |
|
def load_session(filename, db_connection_callback = None): |
573 |
|
"""Load a Thuban session from the file object file |
574 |
|
|
575 |
|
The db_connection_callback, if given should be a callable object |
576 |
|
that can be called like this: |
577 |
|
db_connection_callback(params, message) |
578 |
|
|
579 |
|
where params is a dictionary containing the known connection |
580 |
|
parameters and message is a string with a message why the connection |
581 |
|
failed. db_connection_callback should return a new dictionary with |
582 |
|
corrected and perhaps additional parameters like a password or None |
583 |
|
to indicate that the user cancelled. |
584 |
|
""" |
585 |
|
handler = SessionLoader(db_connection_callback) |
586 |
|
handler.read(filename) |
587 |
|
|
588 |
session = handler.theSession |
session = handler.theSession |
589 |
# Newly loaded session aren't modified |
# Newly loaded session aren't modified |
590 |
session.UnsetModified() |
session.UnsetModified() |
591 |
|
|
592 |
|
handler.Destroy() |
593 |
|
|
594 |
return session |
return session |
595 |
|
|
|
if __name__ == "__main__": |
|
|
# find out the command to run |
|
|
if len(sys.argv) > 1: |
|
|
print "usage: cat <file> | " + sys.argv[0] |
|
|
else: |
|
|
parseSession(sys.stdin) |
|