10 |
from Thuban.Model.load import XMLReader |
from Thuban.Model.load import XMLReader |
11 |
from Thuban.Model.save import XMLWriter |
from Thuban.Model.save import XMLWriter |
12 |
from Thuban.Model.proj import Projection, ProjFile |
from Thuban.Model.proj import Projection, ProjFile |
13 |
|
from xml.sax import SAXParseException |
14 |
|
|
15 |
projdir = \ |
projdir = \ |
16 |
os.path.join(Thuban.__path__[0], os.pardir, "Resources", "Projections") |
os.path.join(Thuban.__path__[0], os.pardir, "Resources", "Projections") |
32 |
return handler.GetProjFile() |
return handler.GetProjFile() |
33 |
|
|
34 |
def WriteProjFile(pf): |
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) |
saver = ProjFileSaver(pf) |
41 |
saver.write(pf.GetFileName()) |
saver.write(pf.GetFilename()) |
42 |
|
|
43 |
def GetProjFiles(dir): |
def GetProjFiles(dir): |
44 |
"""Read all .proj files in the given directory and |
"""Read all .proj files in the given directory and |
63 |
def GetSystemProjFiles(): |
def GetSystemProjFiles(): |
64 |
"""Return a list of ProjFile objects from files that are |
"""Return a list of ProjFile objects from files that are |
65 |
supplied by Thuban. |
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 |
return GetProjFiles(projdir) |
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(): |
def GetUserProjFiles(): |
77 |
"""Return a list of ProjFile objects from files that are user-defined.""" |
"""Return a list of ProjFile objects from files that are user-defined. |
78 |
return GetProjFiles(usrdir) |
|
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): |
class ProjFileReader(XMLReader): |
90 |
|
|
99 |
def read(self, file_or_filename): |
def read(self, file_or_filename): |
100 |
XMLReader.read(self, file_or_filename) |
XMLReader.read(self, file_or_filename) |
101 |
|
|
102 |
self.__pf.SetFileName(XMLReader.GetFileName(self)) |
self.__pf.SetFilename(XMLReader.GetFilename(self)) |
103 |
|
|
104 |
def start_projection(self, name, qname, attrs): |
def start_projection(self, name, qname, attrs): |
105 |
self.params = [] |
self.params = [] |