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
162
163
164
165
166
|
# Copyright (C) 2009, 2010, 2011 Rickard Lindberg, Roger Lindberg
#
# This file is part of Timeline.
#
# Timeline is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Timeline is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Timeline. If not, see <http://www.gnu.org/licenses/>.
import wx
from timelinelib.db.exceptions import TimelineIOError
from timelinelib.db.objects.category import sort_categories
# Border, in pixels, between controls in a window (should always be used when
# border is needed)
BORDER = 5
# Used by dialogs as a return code when a TimelineIOError has been raised
ID_ERROR = wx.NewId()
class WildcardHelper(object):
def __init__(self, name, extensions):
self.name = name
self.ext_data = {}
self.ext_names = []
self._extract_ext_info(extensions)
def wildcard_string(self):
return "%s (%s)|%s" % (
self.name,
", ".join(["*." + e for e in self.ext_names]),
";".join(["*." + e for e in self.ext_names]))
def get_path(self, dialog):
path = dialog.GetPath()
for ext_name in self.ext_names:
if path.endswith("." + ext_name):
return path
return "%s.%s" % (path, self.ext_names[0])
def get_extension_data(self, path):
split_path = path.split(".")
if len(split_path) > 1:
ext_name = split_path[-1]
return self.ext_data.get(ext_name, None)
return None
def _extract_ext_info(self, extensions):
for ext in extensions:
if isinstance(ext, tuple):
name, data = ext
self.ext_data[name] = data
self.ext_names.append(name)
else:
self.ext_names.append(ext)
def category_tree(category_list, parent=None, remove=None):
"""
Transform flat list of categories to a tree based on parent attribute.
The top-level categories have the given parent and each level in the tree
is sorted.
If remove is given then the subtree with remove as root will not be
included.
The tree is represented as a list of tuples, (cat, sub-tree), where cat is
the parent category and subtree is the same tree representation of the
children.
"""
children = [child for child in category_list
if (child.parent is parent and child is not remove)]
sorted_children = sort_categories(children)
tree = [(x, category_tree(category_list, x, remove))
for x in sorted_children]
return tree
def show_modal(fn_create_dialog, fn_handle_db_error, fn_success=None):
"""Show a modal dialog using error handling pattern."""
try:
dialog = fn_create_dialog()
except TimelineIOError, e:
fn_handle_db_error(e)
else:
dialog_result = dialog.ShowModal()
if dialog_result == ID_ERROR:
fn_handle_db_error(dialog.error)
elif fn_success:
fn_success(dialog)
dialog.Destroy()
def create_dialog_db_error_handler(dialog):
def handler(error):
handle_db_error_in_dialog(dialog, error)
return handler
def handle_db_error_in_dialog(dialog, error):
if dialog.IsShown():
# Close the dialog and let the code that created it handle the error.
# Eventually this error will end up in the main frame (which is the
# only object which can handle the error properly).
dialog.error = error
dialog.EndModal(ID_ERROR)
else:
# Re-raise the TimelineIOError exception and let the code that created
# the dialog handle the error.
raise error
def _set_focus_and_select(ctrl):
ctrl.SetFocus()
if hasattr(ctrl, "SelectAll"):
ctrl.SelectAll()
def _display_error_message(message, parent=None):
"""Display an error message in a modal dialog box"""
dial = wx.MessageDialog(parent, message, _("Error"), wx.OK | wx.ICON_ERROR)
dial.ShowModal()
def _ask_question(question, parent=None):
"""Ask a yes/no question and return the reply."""
return wx.MessageBox(question, _("Question"),
wx.YES_NO|wx.CENTRE|wx.NO_DEFAULT, parent)
def set_wait_cursor(parent):
parent.SetCursor(wx.StockCursor(wx.CURSOR_WAIT))
def set_default_cursor(parent):
parent.SetCursor(wx.StockCursor(wx.CURSOR_DEFAULT))
def time_picker_for(time_type):
from timelinelib.wxgui.components.numtimepicker import NumTimePicker
from timelinelib.wxgui.components.pydatetimepicker import PyDateTimePicker
from timelinelib.wxgui.components.wxdatetimepicker import WxDateTimePicker
from timelinelib.time import NumTimeType
from timelinelib.time import PyTimeType
from timelinelib.time import WxTimeType
if isinstance(time_type, PyTimeType):
return PyDateTimePicker
elif isinstance(time_type, WxTimeType):
return WxDateTimePicker
elif isinstance(time_type, NumTimeType):
return NumTimePicker
else:
raise ValueError("Unsupported time type: %s" % time_type)
|
