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 XMLReader
from Thuban.Model.save import XMLWriter
from Thuban.Model.proj import Projection, ProjFile
from xml.sax import SAXParseException

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.
    
    Raises IOError if the file cannot be opened.
    Raises OSError if the file cannot be read.
    Raises SAXParseException if the file is not valid XML.
    """

    handler = ProjFileReader()
    handler.read(filename)
    return handler.GetProjFile()

def WriteProjFile(pf):
    """Write a single .proj file

    Raises IOError if the file cannot be written.
    """

    saver = ProjFileSaver(pf)
    saver.write(pf.GetFilename())

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

    list = []
    try:
        dirlist = os.listdir(dir)
    except OSError:
        pass # if we can't get a directory listing just return []
    else:
        for file in filter(lambda s: s.endswith(PROJ_EXT), dirlist):
            try:
                filename = os.path.join(dir, file)
                list.append(ReadProjFile(filename))
            except (OSError, IOError, SAXParseException):
                pass # just move onto the next file

    return list

def GetSystemProjFiles():
    """Return a list of ProjFile objects from files that are 
    supplied by Thuban.

    If no files could not be opened return a list with one
    empty projection file set to store data in the default file.
    """
    filename = os.path.join(projdir, "defaults.proj")
    try:
        return [ReadProjFile(filename)]
    except (OSError, IOError, SAXParseException):
        return [ProjFile(filename)]

def GetUserProjFiles():
    """Return a list of ProjFile objects from files that are user-defined.

    If no files could not be opened return a list with one
    empty projection file set to store data in the default file.
    """

    filename = os.path.join(usrdir, "user.proj")
    try:
        return [ReadProjFile(filename)]
    except (OSError, IOError, SAXParseException):
        return [ProjFile(filename)]

class ProjFileReader(XMLReader):

    def __init__(self):
        XMLReader.__init__(self)
        self.__pf = ProjFile("")

        XMLReader.AddDispatchers(self,
            {'projection': ("start_projection", "end_projection"),
             'parameter':  ("start_parameter", None)})
            
    def read(self, file_or_filename):
        XMLReader.read(self, file_or_filename)

        self.__pf.SetFilename(XMLReader.GetFilename(self))
        
    def start_projection(self, name, qname, attrs):
        self.params = []
        self.name = attrs.get((None, 'name'), "Unknown")

    def end_projection(self, name, qname):
        self.__pf.Add(Projection(self.params, self.name))

    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)

    def GetProjFile(self):
        return self.__pf


class ProjFileSaver(XMLWriter):

    def __init__(self, pf):
        XMLWriter.__init__(self)
        self.__pf = pf

    def write(self, file_or_filename):
        XMLWriter.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.GetAllParameters():
                self.write_element("parameter", {"value": param})

            self.close_element("projection")

        self.close_element("projectionlist")


1	# 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 XMLReader
11	from Thuban.Model.save import XMLWriter
12	from Thuban.Model.proj import Projection, ProjFile
13	from xml.sax import SAXParseException
14
15	projdir = \
16	os.path.join(Thuban.__path__[0], os.pardir, "Resources", "Projections")
17
18	usrdir = os.path.expanduser("~/.thuban")
19
20	PROJ_EXT = ".proj"
21
22	def ReadProjFile(filename):
23	"""Read a single .proj file and return a ProjFile object.
24
25	Raises IOError if the file cannot be opened.
26	Raises OSError if the file cannot be read.
27	Raises SAXParseException if the file is not valid XML.
28	"""
29
30	handler = ProjFileReader()
31	handler.read(filename)
32	return handler.GetProjFile()
33
34	def WriteProjFile(pf):
35	"""Write a single .proj file
36
37	Raises IOError if the file cannot be written.
38	"""
39
40	saver = ProjFileSaver(pf)
41	saver.write(pf.GetFilename())
42
43	def GetProjFiles(dir):
44	"""Read all .proj files in the given directory and
45	return a list of ProjFile objects.
46	"""
47
48	list = []
49	try:
50	dirlist = os.listdir(dir)
51	except OSError:
52	pass # if we can't get a directory listing just return []
53	else:
54	for file in filter(lambda s: s.endswith(PROJ_EXT), dirlist):
55	try:
56	filename = os.path.join(dir, file)
57	list.append(ReadProjFile(filename))
58	except (OSError, IOError, SAXParseException):
59	pass # just move onto the next file
60
61	return list
62
63	def GetSystemProjFiles():
64	"""Return a list of ProjFile objects from files that are
65	supplied by Thuban.
66
67	If no files could not be opened return a list with one
68	empty projection file set to store data in the default file.
69	"""
70	filename = os.path.join(projdir, "defaults.proj")
71	try:
72	return [ReadProjFile(filename)]
73	except (OSError, IOError, SAXParseException):
74	return [ProjFile(filename)]
75
76	def GetUserProjFiles():
77	"""Return a list of ProjFile objects from files that are user-defined.
78
79	If no files could not be opened return a list with one
80	empty projection file set to store data in the default file.
81	"""
82
83	filename = os.path.join(usrdir, "user.proj")
84	try:
85	return [ReadProjFile(filename)]
86	except (OSError, IOError, SAXParseException):
87	return [ProjFile(filename)]
88
89	class ProjFileReader(XMLReader):
90
91	def __init__(self):
92	XMLReader.__init__(self)
93	self.__pf = ProjFile("")
94
95	XMLReader.AddDispatchers(self,
96	{'projection': ("start_projection", "end_projection"),
97	'parameter': ("start_parameter", None)})
98
99	def read(self, file_or_filename):
100	XMLReader.read(self, file_or_filename)
101
102	self.__pf.SetFilename(XMLReader.GetFilename(self))
103
104	def start_projection(self, name, qname, attrs):
105	self.params = []
106	self.name = attrs.get((None, 'name'), "Unknown")
107
108	def end_projection(self, name, qname):
109	self.__pf.Add(Projection(self.params, self.name))
110
111	def start_parameter(self, name, qname, attrs):
112	s = attrs.get((None, 'value'))
113	s = str(s) # we can't handle unicode in proj
114	self.params.append(s)
115
116	def GetProjFile(self):
117	return self.__pf
118
119
120	class ProjFileSaver(XMLWriter):
121
122	def __init__(self, pf):
123	XMLWriter.__init__(self)
124	self.__pf = pf
125
126	def write(self, file_or_filename):
127	XMLWriter.write(self, file_or_filename)
128
129	self.write_header("projfile", "thuban.dtd")
130	self.write_projfile(self.__pf)
131	self.close()
132
133	def write_projfile(self, pf):
134
135	self.open_element("projectionlist")
136
137	for p in pf.GetProjections():
138	self.open_element("projection", {"name": p.GetName()})
139
140	for param in p.GetAllParameters():
141	self.write_element("parameter", {"value": param})
142
143	self.close_element("projection")
144
145	self.close_element("projectionlist")
146
147
Name	Value
svn:eol-style	native
svn:keywords	Author Date Id Revision