def test_filebrowser_opendir(self):
# --- multiple selection
path = os.path.expanduser('~')
fb = FileBrowser(self.window, initialdir=path, initialfile="test", mode="opendir",
multiple_selection=True, defaultext=".png",
title="Test", filetypes=[],
okbuttontext=None, cancelbuttontext="Cancel",
foldercreation=False)
self.window.update()
fb.right_tree.focus_force()
self.window.update()
fb.event_generate('<Control-a>')
self.window.update()
self.window.update_idletasks()
fb.validate()
walk = os.walk(path)
root, dirs, _ = walk.send(None)
res = list(fb.get_result())
res.sort()
dirs = [os.path.realpath(os.path.join(root, d)) for d in dirs]
dirs.sort()
self.assertEqual(res, dirs)
# --- single selection
fb = FileBrowser(self.window, initialdir=".", initialfile="test", mode="opendir",
multiple_selection=False, defaultext=".png",
title="Test", filetypes=[],
okbuttontext=None, cancelbuttontext="Cancel",
foldercreation=False)
self.window.update()
fb.validate()
self.assertTrue(os.path.isdir(fb.get_result()))
def test_filebrowser_openfile(self):
# --- multiple selection
path = os.path.expanduser('~')
fb = FileBrowser(self.window, initialdir=path, initialfile="test", mode="openfile",
multiple_selection=True, defaultext=".png",
title="Test", filetypes=[],
okbuttontext=None, cancelbuttontext="Cancel",
foldercreation=False)
self.window.update()
fb.right_tree.focus_force()
self.window.update()
fb.event_generate('<Control-a>')
self.window.update()
self.window.update_idletasks()
fb.validate()
walk = os.walk(path)
root, _, files = walk.send(None)
res = list(fb.get_result())
res.sort()
files = [os.path.realpath(os.path.join(root, f)) for f in files]
files.sort()
self.assertEqual(res, files)
# --- single selection
fb = FileBrowser(self.window, initialdir=".", initialfile="test", mode="openfile",
multiple_selection=False, defaultext="",
title="Test", filetypes=[("PNG", '*.png'), ('ALL', '*')],
okbuttontext=None, cancelbuttontext="Cancel",
foldercreation=False)
self.window.update()
fb.validate()
self.assertEqual(fb.get_result(), '')
fb = FileBrowser(self.window, initialdir=".", initialfile="test", mode="openfile",
multiple_selection=False, defaultext="",
title="Test", filetypes=[("PNG", '*.png'), ('ALL', '*')],
okbuttontext=None, cancelbuttontext="Cancel",
foldercreation=False)
self.window.update()
fb.validate()
self.assertEqual(fb.get_result(), '')
fb = FileBrowser(self.window, initialdir=".", initialfile="test", mode="openfile",
multiple_selection=False, defaultext="",
title="Test", filetypes=[],
okbuttontext=None, cancelbuttontext="Cancel",
foldercreation=False)
self.window.update()
# walk = os.walk(os.path.abspath("."))
# root, _, files = walk.send(None)
files = fb.right_tree.tag_has('file')
if files:
fb.right_tree.selection_set(files[0])
fb.validate()
self.assertTrue(os.path.isfile(fb.get_result()))
else:
fb.validate()
self.assertEqual(fb.get_result(), '')
def test_filebrowser_save(self):
fb = FileBrowser(self.window, initialdir="/", initialfile="test", mode="save",
multiple_selection=True, defaultext=".png",
title="Test", filetypes=[("PNG", '*.png'), ('ALL', '*')],
okbuttontext=None, cancelbuttontext="Cancel",
foldercreation=True)
self.window.update()
fb.validate()
self.assertEqual(os.path.abspath(fb.get_result()),
os.path.abspath('/test.png'))
fb = FileBrowser(self.window, initialdir="/", initialfile="test.png", mode="save",
filetypes=[("PNG", '*.png|*.PNG'), ("JPG", '*.jpg|*.JPG'),
('ALL', '*')])
self.window.update()
self.assertEqual(fb.entry.get(), "test.png")
fb.filetype.set('JPG')
self.window.update()
self.assertEqual(fb.entry.get(), "test.jpg")
fb.filetype.set('ALL')
self.window.update()
self.assertEqual(fb.entry.get(), "test.jpg")
fb.filetype.set('PNG')
self.window.update()
self.assertEqual(fb.entry.get(), "test.png")
fb.entry.delete(0, 'end')
fb.entry.insert(0, "test.JPG")
fb.filetype.set('JPG')
self.window.update()
self.assertEqual(fb.entry.get(), "test.JPG")
def askopendirnames(parent=None, title=_("Open"), **kwargs):
"""
Return () or the tuple of the absolute paths of the chosen directories
Options:
* initialdir: initial folder whose content is displayed
* initialfile: initial selected item (just the name, not the full path)
* filetypes: [('name', '*.ext1|*.ext2|..'), ...]
show only files of given filetype ("*" for all files)
* okbuttontext: text displayed on the validate button, if None, the
default text corresponding to the mode is used (either Open or Save)
* cancelbuttontext: text displayed on the button that cancels the
selection.
* foldercreation: enable the user to create new folders if True (default)
"""
dialog = FileBrowser(parent,
mode="opendir",
multiple_selection=True,
title=title,
**kwargs)
dialog.wait_window(dialog)
res = dialog.get_result()
if not res: # type consistency: always return a tuple
res = ()
return res
def askopendirnames(parent=None, title=_("Open"), **kwargs):
"""
Return :obj:`()` or the tuple of the absolute paths of the chosen directories
Arguments:
parent : Tk or Toplevel instance
parent window
title : str
the title of the filebrowser window
initialdir : str
directory whose content is initially displayed
initialfile : str
initially selected item (just the name, not the full path)
filetypes : list :obj:`[("name", "*.ext1|*.ext2|.."), ...]`
only the files of given filetype will be displayed,
e.g. to allow the user to switch between displaying only PNG or JPG
pictures or dispalying all files:
:obj:`filtypes=[("Pictures", "\*.png|\*.PNG|\*.jpg|\*.JPG'), ("All files", "\*")]`
okbuttontext : str
text displayed on the validate button, default is "Open".
cancelbuttontext : str
text displayed on the button that cancels the selection, default is "Cancel".
foldercreation : bool
enable the user to create new folders if True (default)
"""
dialog = FileBrowser(parent,
mode="opendir",
multiple_selection=True,
title=title,
**kwargs)
dialog.wait_window(dialog)
res = dialog.get_result()
if not res: # type consistency: always return a tuple
res = ()
return res
def asksaveasfilename(parent=None, title=_("Save As"), **kwargs):
"""
Return '' or the chosen absolute path (the file might not exist)
Options:
* initialdir: initial folder whose content is displayed
* initialfile: initial selected item (just the name, not the full path)
* defaultext (save mode only): extension added to filename if none is given
* filetypes: [('name', '*.ext1|*.ext2|..'), ...]
show only files of given filetype ("*" for all files)
* okbuttontext: text displayed on the validate button, if None, the
default text corresponding to the mode is used (either Open or Save)
* cancelbuttontext: text displayed on the button that cancels the
selection.
* foldercreation: enable the user to create new folders if True (default)
"""
dialog = FileBrowser(parent, mode="save", title=title, **kwargs)
dialog.wait_window(dialog)
return dialog.get_result()
def asksaveasfilename(parent=None, title=_("Save As"), **kwargs):
"""
Return :obj:`''` or the chosen path (the file might not exist)
Arguments:
parent : Tk or Toplevel instance
parent window
title : str
the title of the filebrowser window
initialdir : str
directory whose content is initially displayed
initialfile : str
initially selected item (just the name, not the full path)
defaultext : str (e.g. '.png')
extension added to filename if none is given (default is none)
filetypes : list :obj:`[("name", "*.ext1|*.ext2|.."), ...]`
only the files of given filetype will be displayed,
e.g. to allow the user to switch between displaying only PNG or JPG
pictures or dispalying all files:
:obj:`filtypes=[("Pictures", "\*.png|\*.PNG|\*.jpg|\*.JPG'), ("All files", "\*")]`
okbuttontext : str
text displayed on the validate button, default is "Open".
cancelbuttontext : str
text displayed on the button that cancels the selection, default is "Cancel".
foldercreation : bool
enable the user to create new folders if True (default)
"""
dialog = FileBrowser(parent, mode="save", title=title, **kwargs)
dialog.wait_window(dialog)
return dialog.get_result()
def test_filebrowser_keybrowse(self):
# --- openfile
fb = FileBrowser(self.window,
initialdir="/",
mode="openfile",
multiple_selection=True)
if not fb.hide:
fb.toggle_hidden()
self.window.update()
fb.right_tree.focus_force()
self.window.update()
ch = fb.right_tree.tag_has('file')
letters = [fb.right_tree.item(c, 'text')[0] for c in ch]
i = 65
while chr(i) in letters:
i += 1
letter = chr(i)
if letter.isalnum():
fb.right_tree.event_generate('<%s>' % letter)
self.window.update()
self.assertTrue(fb.key_browse_entry.winfo_ismapped())
self.assertEqual(fb.key_browse_entry.get(), letter)
fb.right_tree.event_generate('<Return>')
self.window.update()
self.assertFalse(fb.key_browse_entry.winfo_ismapped())
if ch:
letter = fb.right_tree.item(ch[0], 'text')[0]
fb.right_tree.event_generate('<%s>' % letter)
self.window.update()
self.assertTrue(fb.key_browse_entry.winfo_ismapped())
self.assertEqual(fb.key_browse_entry.get(), letter)
self.assertEqual(fb.right_tree.selection(), (ch[0], ))
l = [c for c in ch if fb.right_tree.item(c, 'text')[0] == letter]
fb.key_browse_entry.focus_force()
fb.key_browse_entry.event_generate('<Down>')
self.window.update()
if len(l) > 1:
self.assertEqual(tuple(fb.right_tree.selection()), (l[1], ))
else:
self.assertEqual(tuple(fb.right_tree.selection()), (l[0], ))
fb.key_browse_entry.focus_force()
fb.key_browse_entry.event_generate('<Up>')
self.window.update()
self.assertEqual(tuple(fb.right_tree.selection()), (l[0], ))
fb.key_browse_entry.focus_force()
fb.key_browse_entry.event_generate('<Escape>')
self.window.update()
self.assertFalse(fb.key_browse_entry.winfo_ismapped())
fb.right_tree.focus_force()
fb.right_tree.event_generate('<%s>' % letter)
self.window.update()
fb.right_tree.event_generate('<Return>')
self.window.update()
self.assertEqual(fb.get_result(), (ch[0], ))
# --- opendir
fb = FileBrowser(self.window,
initialdir="/",
mode="opendir",
multiple_selection=True)
self.window.update()
fb.right_tree.focus_force()
self.window.update()
ch = fb.right_tree.tag_has('folder')
letters = [fb.right_tree.item(c, 'text')[0] for c in ch]
i = 65
while chr(i) in letters:
i += 1
letter = chr(i)
if letter.isalnum():
fb.right_tree.event_generate('<%s>' % letter)
self.window.update()
self.assertTrue(fb.key_browse_entry.winfo_ismapped())
self.assertEqual(fb.key_browse_entry.get(), letter)
fb.right_tree.event_generate('<Return>')
self.window.update()
self.assertEqual(fb.get_result(), ('/', ))
fb = FileBrowser(self.window,
initialdir="/",
mode="opendir",
multiple_selection=True)
self.window.update()
fb.right_tree.focus_force()
if ch:
letter = fb.right_tree.item(ch[-1], 'text')[0]
l = [c for c in ch if fb.right_tree.item(c, 'text')[0] == letter]
fb.right_tree.event_generate('<%s>' % letter)
self.window.update()
self.assertTrue(fb.key_browse_entry.winfo_ismapped())
self.assertEqual(fb.key_browse_entry.get(), letter)
self.assertEqual(fb.right_tree.selection(), (l[0], ))
fb.key_browse_entry.focus_force()
fb.key_browse_entry.event_generate('<Down>')
self.window.update()
if len(l) > 1:
self.assertEqual(tuple(fb.right_tree.selection()), (l[1], ))
else:
self.assertEqual(tuple(fb.right_tree.selection()), (l[0], ))
fb.key_browse_entry.focus_force()
fb.key_browse_entry.event_generate('<Up>')
self.window.update()
self.assertEqual(tuple(fb.right_tree.selection()), (l[0], ))
fb.key_browse_entry.focus_force()
fb.key_browse_entry.event_generate('<Escape>')
self.window.update()
self.assertFalse(fb.key_browse_entry.winfo_ismapped())
fb.right_tree.focus_force()
fb.right_tree.event_generate('<%s>' % letter)
self.window.update()
fb.right_tree.event_generate('<Return>')
self.window.update()
self.assertEqual(fb.get_result(), (l[0], ))
# --- arrow nav
fb = FileBrowser(self.window,
initialdir="/",
mode="opendir",
multiple_selection=True)
self.window.update()
fb.right_tree.focus_force()
self.window.update()
fb.event_generate('<Left>')
self.window.update()
fb.left_tree.focus_force()
fb.event_generate('<Up>')
self.window.update()
fb.right_tree.focus_force()
fb.event_generate('<Left>')
self.window.update()
fb.left_tree.focus_force()
fb.event_generate('<Down>')
self.window.update()
fb.right_tree.focus_force()
fb.event_generate('<Down>')
self.window.update()
fb.right_tree.focus_force()
fb.event_generate('<Alt-Left>')
self.window.update()
fb.right_tree.focus_force()
fb.event_generate('<Alt-Right>')
self.window.update()
fb.right_tree.focus_force()
fb.event_generate('<Alt-Up>')
self.window.update()
fb.right_tree.focus_force()
fb.event_generate('<Alt-Down>')
self.window.update()
def test_filebrowser_keybrowse(self):
# --- openfile
fb = FileBrowser(self.window, initialdir="/", mode="openfile",
multiple_selection=True)
if not fb.hide:
fb.toggle_hidden()
self.window.update()
fb.right_tree.focus_force()
self.window.update()
ch = fb.right_tree.get_children('')
letters = [fb.right_tree.item(c, 'text')[0].lower() for c in ch]
i = 65
while chr(i).lower() in letters:
i += 1
letter = chr(i).lower()
keyboard = Controller()
if letter.isalnum():
fb.right_tree.focus_force()
keyboard.press(letter)
keyboard.release(letter)
self.window.update()
self.assertTrue(fb.key_browse_entry.winfo_ismapped())
self.assertEqual(fb.key_browse_entry.get().lower(), letter)
fb.right_tree.event_generate('<Escape>')
self.window.update()
self.assertFalse(fb.key_browse_entry.winfo_ismapped())
if ch:
fb.right_tree.focus_force()
letter = fb.right_tree.item(ch[0], 'text')[0]
keyboard.press(letter)
keyboard.release(letter)
self.window.update()
self.assertTrue(fb.key_browse_entry.winfo_ismapped())
self.assertEqual(fb.key_browse_entry.get().lower(), letter)
self.assertEqual(fb.right_tree.selection(), (ch[0],))
l = [c for c in ch if fb.right_tree.item(c, 'text')[0] == letter]
fb.key_browse_entry.focus_force()
fb.key_browse_entry.event_generate('<Down>')
self.window.update()
if len(l) > 1:
self.assertEqual(tuple(fb.right_tree.selection()), (l[1],))
else:
self.assertEqual(tuple(fb.right_tree.selection()), (l[0],))
fb.key_browse_entry.focus_force()
fb.key_browse_entry.event_generate('<Up>')
self.window.update()
self.assertEqual(tuple(fb.right_tree.selection()), (l[0],))
fb.key_browse_entry.focus_force()
fb.key_browse_entry.event_generate('<Escape>')
self.window.update()
self.assertFalse(fb.key_browse_entry.winfo_ismapped())
fb.right_tree.focus_force()
keyboard.press(letter)
keyboard.release(letter)
self.window.update()
fb.right_tree.event_generate('<Return>')
self.window.update()
item = os.path.realpath(ch[0])
if os.path.isdir(item):
self.assertEqual(fb.history[-1], ch[0])
else:
self.assertEqual(fb.get_result(), (item,))
# --- opendir
fb = FileBrowser(self.window, initialdir="/", mode="opendir",
multiple_selection=True)
self.window.update()
fb.right_tree.focus_force()
self.window.update()
ch = fb.right_tree.tag_has('folder')
letters = [fb.right_tree.item(c, 'text')[0].lower() for c in ch]
i = 65
while chr(i).lower() in letters:
i += 1
letter = chr(i).lower()
if letter.isalnum():
fb.right_tree.focus_force()
keyboard.press(letter)
keyboard.release(letter)
self.window.update()
self.assertTrue(fb.key_browse_entry.winfo_ismapped())
self.assertEqual(fb.key_browse_entry.get(), letter)
fb.right_tree.event_generate('<Return>')
self.window.update()
self.assertEqual(fb.get_result(), (os.path.abspath('/'),))
fb = FileBrowser(self.window, initialdir="/", mode="opendir",
multiple_selection=True)
self.window.update()
fb.right_tree.focus_force()
if ch:
letter = fb.right_tree.item(ch[-1], 'text')[0].lower()
l = [c for c in ch if fb.right_tree.item(c, 'text')[0].lower() == letter]
fb.right_tree.focus_force()
keyboard.press(letter)
keyboard.release(letter)
self.window.update()
self.assertTrue(fb.key_browse_entry.winfo_ismapped())
self.assertEqual(fb.key_browse_entry.get(), letter)
self.assertEqual(fb.right_tree.selection(), (l[0],))
fb.key_browse_entry.focus_force()
fb.key_browse_entry.event_generate('<Down>')
self.window.update()
if len(l) > 1:
self.assertEqual(tuple(fb.right_tree.selection()), (l[1],))
else:
self.assertEqual(tuple(fb.right_tree.selection()), (l[0],))
fb.key_browse_entry.focus_force()
fb.key_browse_entry.event_generate('<Up>')
self.window.update()
self.assertEqual(tuple(fb.right_tree.selection()), (l[0],))
fb.key_browse_entry.focus_force()
fb.key_browse_entry.event_generate('<Escape>')
self.window.update()
self.assertFalse(fb.key_browse_entry.winfo_ismapped())
fb.right_tree.focus_force()
keyboard.press(letter)
keyboard.release(letter)
self.window.update()
fb.right_tree.event_generate('<Return>')
self.window.update()
self.assertEqual(fb.get_result(), (l[0],))
# --- arrow nav
fb = FileBrowser(self.window, initialdir="/", mode="opendir",
multiple_selection=True)
self.window.update()
fb.right_tree.focus_force()
self.window.update()
fb.event_generate('<Left>')
self.window.update()
fb.left_tree.focus_force()
fb.event_generate('<Up>')
self.window.update()
fb.right_tree.focus_force()
fb.event_generate('<Left>')
self.window.update()
fb.left_tree.focus_force()
fb.event_generate('<Down>')
self.window.update()
fb.right_tree.focus_force()
fb.event_generate('<Down>')
self.window.update()
fb.right_tree.focus_force()
fb.event_generate('<Alt-Left>')
self.window.update()
fb.right_tree.focus_force()
fb.event_generate('<Alt-Right>')
self.window.update()
fb.right_tree.focus_force()
fb.event_generate('<Alt-Up>')
self.window.update()
fb.right_tree.focus_force()
fb.event_generate('<Alt-Down>')
self.window.update()