1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
|
#-----------------------------------------------------------------------------
#
# Copyright (c) 2005-2007 by Enthought, Inc.
# All rights reserved.
#
#-----------------------------------------------------------------------------
"""
The Envisage service providing the model state for the single
project plugin.
"""
# Standard imports
import logging
import os
import shutil
# Enthought library imports
from enthought.envisage.api import IApplication
from enthought.preferences.api import IPreferences
from enthought.traits.api import Any, HasTraits, Instance, List
# Setup a logger for this module.
logger = logging.getLogger(__name__)
class ModelService(HasTraits):
"""
The Envisage service providing the model state for the single
project plugin.
"""
##########################################################################
# Attributes (Traits)
##########################################################################
### public 'ModelService' interface ######################################
# The Envisage application that this service is part of.
application = Instance(IApplication)
# The factory to use for creating new projects
factory = Instance('enthought.envisage.ui.single_project.project_factory.'
'ProjectFactory')
# The preferences to be exposed through this service.
preferences = Instance(IPreferences)
# The currently open project
project = Instance('enthought.envisage.ui.single_project.project.Project')
# The current selection within the current project.
selection = List(Any)
##########################################################################
# 'object' interface.
##########################################################################
### operator methods #####################################################
def __init__(self, application, factory, **traits):
"""
Constructor.
We require a reference to an Envisage application and a project
factory to create an instance.
"""
super(ModelService, self).__init__(application = application,
factory = factory, **traits)
return
##########################################################################
# 'ModelService' interface.
##########################################################################
### public interface #####################################################
def are_projects_files(self):
"""
Returns True if project instances are saved as files and False if
they are saved as directories.
"""
return self.factory.PROJECT_CLASS.PROJECTS_ARE_FILES
def clean_location(self, location):
"""
Ensures that there are no existing files or directories at the
specified location by removing them. Exceptions are raised if
there are any errors cleaning out existing files or directories.
"""
logger.debug('Trying to clean location [%s]', location)
if os.path.isfile(location):
os.path.remove(location)
else:
shutil.rmtree(location)
return
def get_default_path(self):
"""
Return the default location for projects.
"""
return self.factory.PROJECT_CLASS.get_default_path(self.application)
### trait handlers #######################################################
def _project_changed(self, old, new):
"""
Called whenever the current project is changed.
We hook this to make sure the new project knows it's current and
the old project knows it's not.
"""
logger.debug('Detected project change from [%s] to [%s] in '
'ModelService [%s]', old, new, self)
if old is not None:
old.stop()
self.selection = []
if new is not None:
new.start()
return
def _selection_changed(self, old, new):
"""
Called whenever the selection within the project is changed.
Implemented simply to log the change.
"""
logger.debug('ModelService [%s] selection changed from [%s] to [%s] ',
self, old, new)
return
### EOF ######################################################################
|
