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