Thuban/Model/resource.py

# Copyright (c) 2003 by Intevation GmbH
# Authors:
# Jonathan Coles <[email protected]>
#
# This program is free software under the GPL (>=v2)
# Read the file COPYING coming with Thuban for details.

import os
import Thuban
from Thuban.Model.load import XMLProcessor, load_xmlfile
from Thuban.Model.save import XMLSaver
from Thuban.Model.proj import Projection, ProjFile

projdir = \
        os.path.join(Thuban.__path__[0], os.pardir, "Resources", "Projections")

usrdir  = os.path.expanduser("~/.thuban")

PROJ_EXT = ".proj"

def ReadProjFile(filename):
    """Read a single .proj file and return a ProjFile object."""

    projs = []
    try:
        handler = ProjFileReader(filename)

        load_xmlfile(filename, handler)
        pf = handler.GetProjFile()
    except OSError:
        pass

    return pf 

def WriteProjFile(pf):
    saver = ProjFileSaver(pf)
    saver.write(pf.GetFileName())

def GetProjections(dir):
    """Read all .proj files in the given directory and 
    return a list of ProjFile objects.
    """

    list = []
    try:
        for file in filter(lambda s: s.endswith(PROJ_EXT), os.listdir(dir)):
            filename = os.path.join(dir, file)
            list.extend(ReadProjFile(filename))
    except OSError:
        pass

    return list

def GetSuppliedProjections():
    """Return a list of ProjFile objects from files that are 
    supplied by Thuban.
    """
    return GetProjections(projdir)

def GetUserProjections():
    """Return a list of ProjFile objects from files that are user-defined."""
    return GetProjections(usrdir)

class ProjFileReader(XMLProcessor):

    def __init__(self, filename):
        XMLProcessor.__init__(self, os.path.dirname(filename))
        
        self.pf = ProjFile(filename)
        
    def start_projection(self, name, qname, attrs):
        self.params = []
        self.name = attrs.get((None, 'name'), "Unknown")
    XMLProcessor.start_dispatcher['projection'] = "start_projection"

    def end_projection(self, name, qname):
        self.pf.Add(Projection(self.params, self.name))
    XMLProcessor.end_dispatcher['projection'] = "end_projection"

    def start_parameter(self, name, qname, attrs):
        s = attrs.get((None, 'value'))
        s = str(s) # we can't handle unicode in proj
        self.params.append(s)
    XMLProcessor.start_dispatcher['parameter'] = "start_parameter"

    def GetProjFile(self):
        return self.pf


class ProjFileSaver(XMLSaver):

    def __init__(self, pf):
        XMLSaver.__init__(self)
        self.pf = pf

    def write(self, file_or_filename):
        XMLSaver.write(self, file_or_filename)

        self.write_header("projfile", "thuban.dtd")
        self.write_projfile(self.pf)
        self.close()

    def write_projfile(self, pf):

        self.open_element("projectionlist")

        for p in pf.GetProjections():
            self.open_element("projection", {"name": p.GetName()})

            for param in p.GetParameters():
                self.write_element("parameter", {"value": param})

            self.close_element("projection")

        self.close_element("projectionlist")


1	jonathan	696	# Copyright (c) 2003 by Intevation GmbH
2			# Authors:
3			# Jonathan Coles <[email protected]>
4			#
5			# This program is free software under the GPL (>=v2)
6			# Read the file COPYING coming with Thuban for details.
7
8			import os
9			import Thuban
10			from Thuban.Model.load import XMLProcessor, load_xmlfile
11			from Thuban.Model.save import XMLSaver
12			from Thuban.Model.proj import Projection, ProjFile
13
14			projdir = \
15			os.path.join(Thuban.__path__[0], os.pardir, "Resources", "Projections")
16
17			usrdir = os.path.expanduser("~/.thuban")
18
19			PROJ_EXT = ".proj"
20
21			def ReadProjFile(filename):
22			"""Read a single .proj file and return a ProjFile object."""
23
24			projs = []
25			try:
26			handler = ProjFileReader(filename)
27
28			load_xmlfile(filename, handler)
29			pf = handler.GetProjFile()
30			except OSError:
31			pass
32
33			return pf
34
35			def WriteProjFile(pf):
36			saver = ProjFileSaver(pf)
37			saver.write(pf.GetFileName())
38
39			def GetProjections(dir):
40			"""Read all .proj files in the given directory and
41			return a list of ProjFile objects.
42			"""
43
44			list = []
45			try:
46			for file in filter(lambda s: s.endswith(PROJ_EXT), os.listdir(dir)):
47			filename = os.path.join(dir, file)
48			list.extend(ReadProjFile(filename))
49			except OSError:
50			pass
51
52			return list
53
54			def GetSuppliedProjections():
55			"""Return a list of ProjFile objects from files that are
56			supplied by Thuban.
57			"""
58			return GetProjections(projdir)
59
60			def GetUserProjections():
61			"""Return a list of ProjFile objects from files that are user-defined."""
62			return GetProjections(usrdir)
63
64			class ProjFileReader(XMLProcessor):
65
66			def __init__(self, filename):
67			XMLProcessor.__init__(self, os.path.dirname(filename))
68
69			self.pf = ProjFile(filename)
70
71			def start_projection(self, name, qname, attrs):
72			self.params = []
73			self.name = attrs.get((None, 'name'), "Unknown")
74			XMLProcessor.start_dispatcher['projection'] = "start_projection"
75
76			def end_projection(self, name, qname):
77			self.pf.Add(Projection(self.params, self.name))
78			XMLProcessor.end_dispatcher['projection'] = "end_projection"
79
80			def start_parameter(self, name, qname, attrs):
81			s = attrs.get((None, 'value'))
82			s = str(s) # we can't handle unicode in proj
83			self.params.append(s)
84			XMLProcessor.start_dispatcher['parameter'] = "start_parameter"
85
86			def GetProjFile(self):
87			return self.pf
88
89
90			class ProjFileSaver(XMLSaver):
91
92			def __init__(self, pf):
93			XMLSaver.__init__(self)
94			self.pf = pf
95
96			def write(self, file_or_filename):
97			XMLSaver.write(self, file_or_filename)
98
99			self.write_header("projfile", "thuban.dtd")
100			self.write_projfile(self.pf)
101			self.close()
102
103			def write_projfile(self, pf):
104
105			self.open_element("projectionlist")
106
107			for p in pf.GetProjections():
108			self.open_element("projection", {"name": p.GetName()})
109
110			for param in p.GetParameters():
111			self.write_element("parameter", {"value": param})
112
113			self.close_element("projection")
114
115			self.close_element("projectionlist")
116
117
Name	Value
svn:eol-style	native
svn:keywords	Author Date Id Revision