Thuban/Model/save.py

# Copyright (c) 2001, 2002 by Intevation GmbH
# Authors:
# Jan-Oliver Wagner <[email protected]>
# Bernhard Herzog <[email protected]>
#
# This program is free software under the GPL (>=v2)
# Read the file COPYING coming with Thuban for details.

"""
Functions to save a session to a file
"""

__version__ = "$Revision$"

import os
import string

import Thuban.Lib.fileutil

def relative_filename(dir, filename):
    """Return a filename relative to dir for the absolute file name absname.

    This is almost the same as the function in fileutil, except that dir
    can be an empty string in which case filename will be returned
    unchanged.
    """
    if dir:
        return Thuban.Lib.fileutil.relative_filename(dir, filename)
    else:
        return filename

def escape(data):
    """Escape &, \", ', <, and > in a string of data.
    """
    data = string.replace(data, "&", "&amp;")
    data = string.replace(data, "<", "&lt;")
    data = string.replace(data, ">", "&gt;")
    data = string.replace(data, '"', "&quot;")
    data = string.replace(data, "'", "&apos;")
    return data

def save_session(session, filename):
    """Save the session session to the file given by filename"""
    dir = os.path.dirname(filename)
    file = open(filename, 'w')
    write = file.write
    write('<?xml version="1.0" encoding="UTF-8"?>\n')
    write('<!DOCTYPE session SYSTEM "thuban.dtd">\n')
    write('<session title="%s">\n' % escape(session.title))
    for map in session.Maps():
        write('\t<map title="%s">\n' % escape(map.title))
        if map.projection and len(map.projection.params) > 0:
            write('\t\t<projection>\n')
            for param in map.projection.params:
                write('\t\t\t<parameter value="%s"/>\n' % escape(param))
            write('\t\t</projection>\n')

        for layer in map.Layers():
            fill = layer.fill
            if fill is None:
                fill = "None"
            else:
                fill = fill.hex()
            stroke = layer.stroke
            if stroke is None:
                stroke = "None"
            else:
                stroke = stroke.hex()
            write(('\t\t<layer title="%s" filename="%s"'
                   ' fill="%s" stroke="%s" stroke_width="%d"/>\n') %
                  (escape(layer.title),
                   escape(relative_filename(dir, layer.filename)),
                   fill, stroke, layer.stroke_width))
        labels = map.LabelLayer().Labels()
        if labels:
            write('\t\t<labellayer>\n')
            for label in labels:
                write(('\t\t\t<label x="%g" y="%g" text="%s"'
                       ' halign="%s" valign="%s"/>\n')
                      % (label.x, label.y, label.text, label.halign,
                         label.valign))
            write('\t\t</labellayer>\n')
        write('\t</map>\n')
    write('</session>\n')

    # after a successful save consider the session unmodified.
    session.UnsetModified()
1	# Copyright (c) 2001, 2002 by Intevation GmbH
2	# Authors:
3	# Jan-Oliver Wagner <[email protected]>
4	# Bernhard Herzog <[email protected]>
5	#
6	# This program is free software under the GPL (>=v2)
7	# Read the file COPYING coming with Thuban for details.
8
9	"""
10	Functions to save a session to a file
11	"""
12
13	__version__ = "$Revision$"
14
15	import os
16	import string
17
18	import Thuban.Lib.fileutil
19
20	def relative_filename(dir, filename):
21	"""Return a filename relative to dir for the absolute file name absname.
22
23	This is almost the same as the function in fileutil, except that dir
24	can be an empty string in which case filename will be returned
25	unchanged.
26	"""
27	if dir:
28	return Thuban.Lib.fileutil.relative_filename(dir, filename)
29	else:
30	return filename
31
32	def escape(data):
33	"""Escape &, \", ', <, and > in a string of data.
34	"""
35	data = string.replace(data, "&", "&")
36	data = string.replace(data, "<", "<")
37	data = string.replace(data, ">", ">")
38	data = string.replace(data, '"', """)
39	data = string.replace(data, "'", "'")
40	return data
41
42	def save_session(session, filename):
43	"""Save the session session to the file given by filename"""
44	dir = os.path.dirname(filename)
45	file = open(filename, 'w')
46	write = file.write
47	write('<?xml version="1.0" encoding="UTF-8"?>\n')
48	write('<!DOCTYPE session SYSTEM "thuban.dtd">\n')
49	write('<session title="%s">\n' % escape(session.title))
50	for map in session.Maps():
51	write('\t<map title="%s">\n' % escape(map.title))
52	if map.projection and len(map.projection.params) > 0:
53	write('\t\t<projection>\n')
54	for param in map.projection.params:
55	write('\t\t\t<parameter value="%s"/>\n' % escape(param))
56	write('\t\t</projection>\n')
57
58	for layer in map.Layers():
59	fill = layer.fill
60	if fill is None:
61	fill = "None"
62	else:
63	fill = fill.hex()
64	stroke = layer.stroke
65	if stroke is None:
66	stroke = "None"
67	else:
68	stroke = stroke.hex()
69	write(('\t\t<layer title="%s" filename="%s"'
70	' fill="%s" stroke="%s" stroke_width="%d"/>\n') %
71	(escape(layer.title),
72	escape(relative_filename(dir, layer.filename)),
73	fill, stroke, layer.stroke_width))
74	labels = map.LabelLayer().Labels()
75	if labels:
76	write('\t\t<labellayer>\n')
77	for label in labels:
78	write(('\t\t\t<label x="%g" y="%g" text="%s"'
79	' halign="%s" valign="%s"/>\n')
80	% (label.x, label.y, label.text, label.halign,
81	label.valign))
82	write('\t\t</labellayer>\n')
83	write('\t</map>\n')
84	write('</session>\n')
85
86	# after a successful save consider the session unmodified.
87	session.UnsetModified()
Name	Value
svn:eol-style	native
svn:keywords	Author Date Id Revision