annotate frontends/primitivus/files_management.py @ 23:de305d93a503

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