Mercurial > urwid-satext
annotate urwid_satext/files_management.py @ 92:fdd0543677d4
use of new keys module in files_management
author | Goffi <goffi@goffi.org> |
---|---|
date | Mon, 08 Sep 2014 15:42:25 +0200 |
parents | eddb8369ba06 |
children | 900014ae36b8 |
rev | line source |
---|---|
21 | 1 #!/usr/bin/python |
2 # -*- coding: utf-8 -*- | |
3 | |
64 | 4 # Urwid SàT extensions |
71 | 5 # Copyright (C) 2009, 2010, 2011, 2012, 2013, 2014 Jérôme Poisson (goffi@goffi.org) |
64 | 6 # |
7 # This program is free software: you can redistribute it and/or modify | |
8 # it under the terms of the GNU Lesser General Public License as published by | |
9 # the Free Software Foundation, either version 3 of the License, or | |
10 # (at your option) any later version. | |
11 # | |
12 # This program is distributed in the hope that it will be useful, | |
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
15 # GNU Lesser General Public License for more details. | |
16 # | |
17 # You should have received a copy of the GNU Lesser General Public License | |
18 # along with this program. If not, see <http://www.gnu.org/licenses/>. | |
21 | 19 |
20 import urwid | |
30
1aeb3540aa49
files reorganisation after project separation. new README, and COPYING files
Goffi <goffi@goffi.org>
parents:
26
diff
changeset
|
21 import sat_widgets |
21 | 22 import os, os.path |
23 from xml.dom import minidom | |
24 from logging import debug, info, error | |
23 | 25 from time import time |
92
fdd0543677d4
use of new keys module in files_management
Goffi <goffi@goffi.org>
parents:
71
diff
changeset
|
26 from .keys import action_key_map as a_key |
34
875ff127b2dd
- i18n management: gettext integration + french translation
Goffi <goffi@goffi.org>
parents:
32
diff
changeset
|
27 |
875ff127b2dd
- i18n management: gettext integration + french translation
Goffi <goffi@goffi.org>
parents:
32
diff
changeset
|
28 import gettext |
875ff127b2dd
- i18n management: gettext integration + french translation
Goffi <goffi@goffi.org>
parents:
32
diff
changeset
|
29 gettext.install('urwid_satext', unicode=True) |
21 | 30 |
30
1aeb3540aa49
files reorganisation after project separation. new README, and COPYING files
Goffi <goffi@goffi.org>
parents:
26
diff
changeset
|
31 class PathEdit(sat_widgets.AdvancedEdit): |
23 | 32 """AdvancedEdit with manage file paths""" |
66 | 33 |
21 | 34 def keypress(self, size, key): |
23 | 35 if key == '~' and self.edit_pos==0: |
36 expanded = os.path.expanduser('~') | |
37 self.set_edit_text(os.path.normpath(expanded+'/'+self.edit_text)) | |
38 self.set_edit_pos(len(expanded)+1) | |
92
fdd0543677d4
use of new keys module in files_management
Goffi <goffi@goffi.org>
parents:
71
diff
changeset
|
39 elif key == a_key['EDIT_DELETE_LAST_WORD']: |
21 | 40 if self.edit_pos<2: |
41 return | |
42 before = self.edit_text[:self.edit_pos] | |
43 pos = (before[:-1] if before.endswith('/') else before).rfind("/")+1 | |
44 self.set_edit_text(before[:pos] + self.edit_text[self.edit_pos:]) | |
45 self.set_edit_pos(pos) | |
46 return | |
47 else: | |
66 | 48 return super(PathEdit, self).keypress(size, key) |
21 | 49 |
23 | 50 class FilesViewer(urwid.WidgetWrap): |
51 """List specialised for files""" | |
52 | |
58
7155b81ffdd5
new 'dir' style in FileDialog (get a dir path instead of a file
Goffi <goffi@goffi.org>
parents:
34
diff
changeset
|
53 def __init__(self, onPreviousDir, onDirClick, onFileClick = None): |
23 | 54 self.path='' |
55 self.key_cache = '' | |
56 self.key_time = time() | |
57 self.onPreviousDir = onPreviousDir | |
58 self.onDirClick = onDirClick | |
59 self.onFileClick = onFileClick | |
60 self.files_list = urwid.SimpleListWalker([]) | |
66 | 61 self.show_hidden = True |
23 | 62 listbox = urwid.ListBox(self.files_list) |
63 urwid.WidgetWrap.__init__(self, listbox) | |
64 | |
65 def keypress(self, size, key): | |
92
fdd0543677d4
use of new keys module in files_management
Goffi <goffi@goffi.org>
parents:
71
diff
changeset
|
66 if key==a_key['FILES_HIDDEN_HIDE']: |
23 | 67 #(un)hide hidden files |
68 self.show_hidden = not self.show_hidden | |
69 self.showDirectory(self.path) | |
92
fdd0543677d4
use of new keys module in files_management
Goffi <goffi@goffi.org>
parents:
71
diff
changeset
|
70 elif key==a_key['FILES_JUMP_DIRECTORIES']: |
23 | 71 #jump to directories |
72 if self.files_list: | |
73 self._w.set_focus(0) | |
92
fdd0543677d4
use of new keys module in files_management
Goffi <goffi@goffi.org>
parents:
71
diff
changeset
|
74 elif key==a_key['FILES_JUMP_FILES']: |
23 | 75 for idx in range(len(self.files_list)): |
76 if isinstance(self.files_list[idx].base_widget,urwid.Divider): | |
77 if idx<len(self.files_list)-1: | |
78 self._w.set_focus(idx+1) | |
79 break | |
80 elif len(key) == 1: | |
81 if time() - self.key_time > 2: | |
82 self.key_cache=key | |
83 else: | |
84 self.key_cache+=key | |
85 self.key_time = time() | |
86 for idx in range(len(self.files_list)): | |
30
1aeb3540aa49
files reorganisation after project separation. new README, and COPYING files
Goffi <goffi@goffi.org>
parents:
26
diff
changeset
|
87 if isinstance(self.files_list[idx],sat_widgets.ClickableText) and self.files_list[idx].get_text().lower().startswith(self.key_cache.lower()): |
23 | 88 self._w.set_focus(idx) |
89 break | |
90 else: | |
91 return self._w.keypress(size, key) | |
92 | |
93 def showDirectory(self, path): | |
94 self.path = path | |
95 del self.files_list[:] | |
96 directories = [] | |
97 files = [] | |
98 try: | |
99 for filename in os.listdir(path): | |
100 fullpath = os.path.join(path,filename) | |
101 if os.path.isdir(fullpath): | |
102 directories.append(filename) | |
103 else: | |
104 files.append(filename) | |
105 except OSError: | |
66 | 106 self.files_list.append(urwid.Text(("warning",_("Impossible to list directory")),'center')) |
23 | 107 directories.sort() |
108 files.sort() | |
109 if os.path.abspath(path)!='/' and os.path.abspath(path) != '//': | |
30
1aeb3540aa49
files reorganisation after project separation. new README, and COPYING files
Goffi <goffi@goffi.org>
parents:
26
diff
changeset
|
110 previous_wid = sat_widgets.ClickableText(('directory','..')) |
23 | 111 urwid.connect_signal(previous_wid,'click',self.onPreviousDir) |
112 self.files_list.append(previous_wid) | |
113 for directory in directories: | |
114 if directory.startswith('.') and not self.show_hidden: | |
115 continue | |
30
1aeb3540aa49
files reorganisation after project separation. new README, and COPYING files
Goffi <goffi@goffi.org>
parents:
26
diff
changeset
|
116 dir_wid = sat_widgets.ClickableText(('directory',directory)) |
23 | 117 urwid.connect_signal(dir_wid,'click',self.onDirClick) |
118 self.files_list.append(dir_wid) | |
119 self.files_list.append(urwid.AttrMap(urwid.Divider('-'),'separator')) | |
120 for filename in files: | |
121 if filename.startswith('.') and not self.show_hidden: | |
122 continue | |
30
1aeb3540aa49
files reorganisation after project separation. new README, and COPYING files
Goffi <goffi@goffi.org>
parents:
26
diff
changeset
|
123 file_wid = sat_widgets.ClickableText(filename) |
58
7155b81ffdd5
new 'dir' style in FileDialog (get a dir path instead of a file
Goffi <goffi@goffi.org>
parents:
34
diff
changeset
|
124 if self.onFileClick: |
7155b81ffdd5
new 'dir' style in FileDialog (get a dir path instead of a file
Goffi <goffi@goffi.org>
parents:
34
diff
changeset
|
125 urwid.connect_signal(file_wid,'click',self.onFileClick) |
23 | 126 self.files_list.append(file_wid) |
127 | |
128 | |
129 | |
21 | 130 class FileDialog(urwid.WidgetWrap): |
131 | |
23 | 132 def __init__(self, ok_cb, cancel_cb, title=_("Please select a file"), style=[]): |
133 """Create file dialog | |
134 @param title: title of the window/popup | |
58
7155b81ffdd5
new 'dir' style in FileDialog (get a dir path instead of a file
Goffi <goffi@goffi.org>
parents:
34
diff
changeset
|
135 @param style: list of string: |
7155b81ffdd5
new 'dir' style in FileDialog (get a dir path instead of a file
Goffi <goffi@goffi.org>
parents:
34
diff
changeset
|
136 - 'dir' if a dir path must be selected |
23 | 137 """ |
138 self.ok_cb = ok_cb | |
58
7155b81ffdd5
new 'dir' style in FileDialog (get a dir path instead of a file
Goffi <goffi@goffi.org>
parents:
34
diff
changeset
|
139 self._type = 'dir' if 'dir' in style else 'normal' |
21 | 140 self.__home_path = os.path.expanduser('~') |
23 | 141 self.path_wid = PathEdit(_('Path: ')) |
142 self.path_wid.setCompletionMethod(self._directory_completion) | |
21 | 143 urwid.connect_signal(self.path_wid, 'change', self.onPathChange) |
144 header = urwid.Pile([self.path_wid, urwid.Divider(u'─')]) | |
145 bookm_list = urwid.SimpleListWalker([]) | |
146 self.bookmarks = list(self.getBookmarks()) | |
147 self.bookmarks.sort() | |
148 for bookmark in self.bookmarks: | |
149 if bookmark.startswith(self.__home_path): | |
150 bookmark="~"+bookmark[len(self.__home_path):] | |
30
1aeb3540aa49
files reorganisation after project separation. new README, and COPYING files
Goffi <goffi@goffi.org>
parents:
26
diff
changeset
|
151 book_wid = sat_widgets.ClickableText(bookmark) |
21 | 152 urwid.connect_signal(book_wid, 'click', self.onBookmarkSelected) |
153 bookm_list.append(book_wid) | |
154 bookm_wid = urwid.Frame(urwid.ListBox(bookm_list), urwid.AttrMap(urwid.Text(_('Bookmarks'),'center'),'title')) | |
58
7155b81ffdd5
new 'dir' style in FileDialog (get a dir path instead of a file
Goffi <goffi@goffi.org>
parents:
34
diff
changeset
|
155 self.files_wid = FilesViewer(self.onPreviousDir, self.onDirClick, self.onFileClick if self._type == 'normal' else None) |
21 | 156 center_row = urwid.Columns([('weight',2,bookm_wid), |
30
1aeb3540aa49
files reorganisation after project separation. new README, and COPYING files
Goffi <goffi@goffi.org>
parents:
26
diff
changeset
|
157 ('weight',8,sat_widgets.VerticalSeparator(self.files_wid))]) |
23 | 158 |
159 buttons = [] | |
58
7155b81ffdd5
new 'dir' style in FileDialog (get a dir path instead of a file
Goffi <goffi@goffi.org>
parents:
34
diff
changeset
|
160 if self._type == 'dir': |
7155b81ffdd5
new 'dir' style in FileDialog (get a dir path instead of a file
Goffi <goffi@goffi.org>
parents:
34
diff
changeset
|
161 buttons.append(sat_widgets.CustomButton(_('Ok'), self._validateDir)) |
30
1aeb3540aa49
files reorganisation after project separation. new README, and COPYING files
Goffi <goffi@goffi.org>
parents:
26
diff
changeset
|
162 buttons.append(sat_widgets.CustomButton(_('Cancel'),cancel_cb)) |
23 | 163 max_len = max([button.getSize() for button in buttons]) |
164 buttons_wid = urwid.GridFlow(buttons,max_len,1,0,'center') | |
30
1aeb3540aa49
files reorganisation after project separation. new README, and COPYING files
Goffi <goffi@goffi.org>
parents:
26
diff
changeset
|
165 main_frame = sat_widgets.FocusFrame(center_row, header, buttons_wid) |
1aeb3540aa49
files reorganisation after project separation. new README, and COPYING files
Goffi <goffi@goffi.org>
parents:
26
diff
changeset
|
166 decorated = sat_widgets.LabelLine(main_frame, sat_widgets.SurroundedText(title)) |
21 | 167 urwid.WidgetWrap.__init__(self, decorated) |
23 | 168 self.path_wid.set_edit_text(os.getcwdu()) |
169 | |
58
7155b81ffdd5
new 'dir' style in FileDialog (get a dir path instead of a file
Goffi <goffi@goffi.org>
parents:
34
diff
changeset
|
170 def _validateDir(self, wid): |
7155b81ffdd5
new 'dir' style in FileDialog (get a dir path instead of a file
Goffi <goffi@goffi.org>
parents:
34
diff
changeset
|
171 """ call ok callback if current path is a dir """ |
7155b81ffdd5
new 'dir' style in FileDialog (get a dir path instead of a file
Goffi <goffi@goffi.org>
parents:
34
diff
changeset
|
172 path = os.path.abspath(self.path_wid.get_edit_text()) |
7155b81ffdd5
new 'dir' style in FileDialog (get a dir path instead of a file
Goffi <goffi@goffi.org>
parents:
34
diff
changeset
|
173 if os.path.isdir(path): |
7155b81ffdd5
new 'dir' style in FileDialog (get a dir path instead of a file
Goffi <goffi@goffi.org>
parents:
34
diff
changeset
|
174 self.ok_cb(path) |
7155b81ffdd5
new 'dir' style in FileDialog (get a dir path instead of a file
Goffi <goffi@goffi.org>
parents:
34
diff
changeset
|
175 |
23 | 176 def _directory_completion(self, path, completion_data): |
177 path=os.path.abspath(path) | |
178 if not os.path.isdir(path): | |
179 head,dir_start = os.path.split(path) | |
180 else: | |
181 head=path | |
182 dir_start='' | |
183 try: | |
184 filenames = os.listdir(head) | |
185 filenames.sort() | |
186 try: | |
187 start_idx=filenames.index(completion_data['last_dir'])+1 | |
188 if start_idx == len(filenames): | |
189 start_idx = 0 | |
190 except (KeyError,ValueError): | |
191 start_idx = 0 | |
192 for idx in range(start_idx,len(filenames)) + range(0,start_idx): | |
193 full_path = os.path.join(head,filenames[idx]) | |
194 if filenames[idx].lower().startswith(dir_start.lower()) and os.path.isdir(full_path): | |
195 completion_data['last_dir'] = filenames[idx] | |
66 | 196 return full_path |
23 | 197 except OSError: |
198 pass | |
199 return path | |
21 | 200 |
201 def getBookmarks(self): | |
23 | 202 gtk_bookm = os.path.expanduser("~/.gtk-bookmarks") |
203 kde_bookm = os.path.expanduser("~/.kde/share/apps/kfileplaces/bookmarks.xml") | |
21 | 204 bookmarks = set() |
205 try: | |
23 | 206 with open(gtk_bookm) as gtk_fd: |
207 for bm in gtk_fd.readlines(): | |
21 | 208 if bm.startswith("file:///"): |
209 bookmarks.add(bm[7:].replace('\n','')) | |
210 except IOError: | |
23 | 211 info(_('No GTK bookmarks file found')) |
21 | 212 pass |
66 | 213 |
21 | 214 try: |
215 dom = minidom.parse(kde_bookm) | |
23 | 216 for elem in dom.getElementsByTagName('bookmark'): |
21 | 217 bm = elem.getAttribute("href") |
218 if bm.startswith("file:///"): | |
219 bookmarks.add(bm[7:]) | |
220 except IOError: | |
221 info(_('No KDE bookmarks file found')) | |
222 pass | |
66 | 223 |
21 | 224 return bookmarks |
225 | |
226 def onBookmarkSelected(self, button): | |
227 self.path_wid.set_edit_text(os.path.expanduser(button.get_text())) | |
228 | |
23 | 229 def onPathChange(self, edit, path): |
230 if os.path.isdir(path): | |
231 self.files_wid.showDirectory(path) | |
232 | |
233 def onPreviousDir(self, wid): | |
234 path = os.path.abspath(self.path_wid.get_edit_text()) | |
235 if not os.path.isdir(path): | |
236 path = dirname(path) | |
237 self.path_wid.set_edit_text(os.path.split(path)[0]) | |
238 | |
239 def onDirClick(self, wid): | |
240 path = os.path.abspath(self.path_wid.get_edit_text()) | |
241 if not os.path.isdir(path): | |
242 path = dirname(path) | |
243 self.path_wid.set_edit_text(os.path.join(path,wid.get_text())) | |
66 | 244 |
23 | 245 def onFileClick(self, wid): |
246 self.ok_cb(os.path.abspath(os.path.join(self.files_wid.path,wid.get_text()))) |