Quick search

Table Of Contents

Source code for kivy.uix.filechooser

'''
FileChooser
===========

.. versionadded:: 1.0.5


.. versionchanged:: 1.2.0
    In the chooser template, the `controller` is not a direct reference anymore
    but a weak-reference.
    You must update all the notation `root.controller.xxx` to
    `root.controller().xxx`.

Simple example
--------------

main.py

.. include:: ../../examples/RST_Editor/main.py
    :literal:

editor.kv

.. highlight:: kv

.. include:: ../../examples/RST_Editor/editor.kv
    :literal:

'''

__all__ = ('FileChooserListView', 'FileChooserIconView',
           'FileChooserController', 'FileChooserProgressBase',
           'FileSystemAbstract', 'FileSystemLocal')

from weakref import ref
from time import time
from kivy.compat import string_types
from kivy.factory import Factory
from kivy.clock import Clock
from kivy.lang import Builder
from kivy.logger import Logger
from kivy.utils import platform as core_platform
from kivy.uix.floatlayout import FloatLayout
from kivy.properties import (
    StringProperty, ListProperty, BooleanProperty, ObjectProperty,
    NumericProperty)
from os import listdir
from os.path import (
    basename, join, sep, normpath, expanduser, altsep,
    splitdrive, realpath, getsize, isdir)
from fnmatch import fnmatch
import collections

platform = core_platform
filesize_units = ('B', 'KB', 'MB', 'GB', 'TB')

_have_win32file = False
if platform == 'win':
    # Import that module here as it's not available on non-windows machines.
    # See http://bit.ly/i9klJE except that the attributes are defined in
    # win32file not win32com (bug on page).
    # Note: For some reason this doesn't work after a os.chdir(), no matter to
    #       what directory you change from where. Windows weirdness.
    try:
        from win32file import FILE_ATTRIBUTE_HIDDEN, GetFileAttributesExW, error
        _have_win32file = True
    except ImportError:
        Logger.error('filechooser: win32file module is missing')
        Logger.error('filechooser: we cant check if a file is hidden or not')


def alphanumeric_folders_first(files, filesystem):
    return (sorted(f for f in files if filesystem.is_dir(f)) +
            sorted(f for f in files if not filesystem.is_dir(f)))


[docs]class FileSystemAbstract(object): '''Class for implementing a File System view that can be used with the :class:`FileChooser`.:attr:`~FileChooser.file_system`. .. versionadded:: 1.8.0 '''
[docs] def listdir(self, fn): '''Return the list of files in the directory `fn` ''' pass
[docs] def getsize(self, fn): '''Return the size in bytes of a file ''' pass
[docs] def is_hidden(self, fn): '''Return True if the file is hidden ''' pass
[docs] def is_dir(self, fn): '''Return True if the argument passed to this method is a directory ''' pass
[docs]class FileSystemLocal(FileSystemAbstract): '''Implementation of :class:`FileSystemAbstract` for local files .. versionadded:: 1.8.0 ''' def listdir(self, fn): return listdir(fn) def getsize(self, fn): return getsize(fn) def is_hidden(self, fn): if platform == 'win': if not _have_win32file: return False try: return GetFileAttributesExW(fn)[0] & FILE_ATTRIBUTE_HIDDEN except error: # This error can occured when a file is already accessed by # someone else. So don't return to True, because we have lot # of chances to not being able to do anything with it. Logger.exception('unable to access to <%s>' % fn) return True return basename(fn).startswith('.') def is_dir(self, fn): return isdir(fn)
[docs]class FileChooserProgressBase(FloatLayout): '''Base for implementing a progress view. This view is used when too many entries need to be created and are delayed over multiple frames. .. versionadded:: 1.2.0 ''' path = StringProperty('') '''Current path of the FileChooser, read-only. ''' index = NumericProperty(0) '''Current index of :attr:`total` entries to be loaded. ''' total = NumericProperty(1) '''Total number of entries to load. '''
[docs] def cancel(self, *largs): '''Cancel any action from the FileChooserController. ''' if self.parent: self.parent.cancel()
def on_touch_down(self, touch): if self.collide_point(*touch.pos): super(FileChooserProgressBase, self).on_touch_down(touch) return True def on_touch_move(self, touch): if self.collide_point(*touch.pos): super(FileChooserProgressBase, self).on_touch_move(touch) return True def on_touch_up(self, touch): if self.collide_point(*touch.pos): super(FileChooserProgressBase, self).on_touch_up(touch) return True
class FileChooserProgress(FileChooserProgressBase): pass
[docs]class FileChooserController(FloatLayout): '''Base for implementing a FileChooser. Don't use this class directly, but prefer using an implementation such as the :class:`FileChooserListView` or :class:`FileChooserIconView`. :Events: `on_entry_added`: entry, parent Fired when a root-level entry is added to the file list. `on_entries_cleared` Fired when the the entries list is cleared, usually when the root is refreshed. `on_subentry_to_entry`: entry, parent Fired when a sub-entry is added to an existing entry. `on_remove_subentry`: entry, parent Fired when entries are removed from an entry, usually when a node is closed. `on_submit`: selection, touch Fired when a file has been selected with a double-tap. ''' _ENTRY_TEMPLATE = None path = StringProperty(u'/') ''' :class:`~kivy.properties.StringProperty`, defaults to the current working directory as a unicode string. It specifies the path on the filesystem that this controller should refer to. .. warning:: If a unicode path is specified, all the files returned will be in unicode allowing the display of unicode files and paths. If a bytes path is specified, only files and paths with ascii names will be displayed properly: non-ascii filenames will be displayed and listed with questions marks (?) instead of their unicode characters. ''' filters = ListProperty([]) ''':class:`~kivy.properties.ListProperty`, defaults to [], equal to '\*'. Specifies the filters to be applied to the files in the directory. The filters are not reset when the path changes. You need to do that yourself if desired. There are two kinds of filters: patterns and callbacks. #. Patterns e.g. ['\*.png']. You can use the following patterns: ========== ================================= Pattern Meaning ========== ================================= \* matches everything ? matches any single character [seq] matches any character in seq [!seq] matches any character not in seq ========== ================================= #. Callbacks You can specify a function that will be called for each file. The callback will be passed the folder and file name as the first and second parameters respectively. It should return True to indicate a match and False otherwise. .. versionchanged:: 1.4.0 If the filter is a callable (function or method), it will be called with the path and the file name as arguments for each file in the directory. The callable should returns True to indicate a match and False overwise. ''' filter_dirs = BooleanProperty(False) ''' :class:`~kivy.properties.BooleanProperty`, defaults to False. Indicates whether filters should also apply to directories. ''' sort_func = ObjectProperty(alphanumeric_folders_first) ''' :class:`~kivy.properties.ObjectProperty`. Provides a function to be called with a list of filenames, and the filesystem implementation as the second argument. Returns a list of filenames sorted for display in the view. .. versionchanged:: 1.8.0 The signature needs now 2 arguments: first the list of files, second the filesystem class to use. ''' files = ListProperty([]) ''' Read-only :class:`~kivy.properties.ListProperty`. The list of files in the directory specified by path after applying the filters. ''' show_hidden = BooleanProperty(False) ''' :class:`~kivy.properties.BooleanProperty`, defaults to False. Determines whether hidden files and folders should be shown. ''' selection = ListProperty([]) ''' Read-only :class:`~kivy.properties.ListProperty`. Contains the list of files that are currently selected. ''' multiselect = BooleanProperty(False) ''' :class:`~kivy.properties.BooleanProperty`, defaults to False. Determines whether the user is able to select multiple files or not. ''' dirselect = BooleanProperty(False) ''' :class:`~kivy.properties.BooleanProperty`, defaults to False. Determines whether directories are valid selections or not. .. versionadded:: 1.1.0 ''' rootpath = StringProperty(None, allownone=True) ''' Root path to use instead of the system root path. If set, it will not show a ".." directory to go up to the root path. For example, if you set rootpath to /users/foo, the user will be unable to go to /users or to any other directory not starting with /users/foo. .. versionadded:: 1.2.0 :class:`~kivy.properties.StringProperty`, defaults to None. .. note:: Similar to :attr:`path`, if `rootpath` is specified, whether it's a bytes or unicode string determines the type of the filenames and paths read. ''' progress_cls = ObjectProperty(FileChooserProgress) '''Class to use for displaying a progress indicator for filechooser loading. .. versionadded:: 1.2.0 :class:`~kivy.properties.ObjectProperty`, defaults to :class:`FileChooserProgress`. .. versionchanged:: 1.8.0 If you set a string, the :class:`~kivy.factory.Factory` will be used to resolve the class. ''' file_encodings = ListProperty(['utf-8', 'latin1', 'cp1252']) '''Possible encodings for decoding a filename to unicode. In the case that the user has a weird filename, undecodable without knowing it's initial encoding, we have no other choice than to guess it. Please note that if you encounter an issue because of a missing encoding here, we'll be glad to add it to this list. .. versionadded:: 1.3.0 .. deprecated:: 1.8.0 This property is no longer used as the filechooser no longer decodes the file names. file_encodings is a :class:`~kivy.properties.ListProperty` and defaults to ['utf-8', 'latin1', 'cp1252'], ''' file_system = ObjectProperty(FileSystemLocal(), baseclass=FileSystemAbstract) '''Implementation to access the file system. Must be an instance of FileSystemAbstract. .. versionadded:: 1.8.0 :class:`~kivy.properties.ObjectProperty`, defaults to :class:`FileSystemLocal()` ''' __events__ = ('on_entry_added', 'on_entries_cleared', 'on_subentry_to_entry', 'on_remove_subentry', 'on_submit') def __init__(self, **kwargs): self._progress = None super(FileChooserController, self).__init__(**kwargs) self._items = [] self.bind(selection=self._update_item_selection) self._previous_path = [self.path] self.bind(path=self._save_previous_path) self.bind(path=self._trigger_update, filters=self._trigger_update, rootpath=self._trigger_update) self._trigger_update() def on_touch_down(self, touch): # don't respond to touchs outside self if not self.collide_point(*touch.pos): return if self.disabled: return True return super(FileChooserController, self).on_touch_down(touch) def on_touch_up(self, touch): # don't respond to touchs outside self if not self.collide_point(*touch.pos): return True if self.disabled: return True return super(FileChooserController, self).on_touch_up(touch) def _update_item_selection(self, *args): for item in self._items: item.selected = item.path in self.selection def _save_previous_path(self, instance, value): self._previous_path.append(value) self._previous_path = self._previous_path[-2:] def _trigger_update(self, *args): Clock.unschedule(self._update_files) Clock.schedule_once(self._update_files) def on_entry_added(self, node, parent=None): pass def on_entries_cleared(self): pass def on_subentry_to_entry(self, subentry, entry): pass def on_remove_subentry(self, subentry, entry): pass def on_submit(self, selected, touch=None): pass
[docs] def entry_touched(self, entry, touch): '''(internal) This method must be called by the template when an entry is touched by the user. ''' if ( 'button' in touch.profile and touch.button in ( 'scrollup', 'scrolldown', 'scrollleft', 'scrollright')): return False if self.multiselect: if self.file_system.is_dir(entry.path) and touch.is_double_tap: self.open_entry(entry) else: if entry.path in self.selection: self.selection.remove(entry.path) else: self.selection.append(entry.path) else: if self.file_system.is_dir(entry.path): if self.dirselect: self.selection = [entry.path, ] else: self.selection = [entry.path, ]
[docs] def entry_released(self, entry, touch): '''(internal) This method must be called by the template when an entry is touched by the user. .. versionadded:: 1.1.0 ''' if ( 'button' in touch.profile and touch.button in ( 'scrollup', 'scrolldown', 'scrollleft', 'scrollright')): return False if not self.multiselect: if self.file_system.is_dir(entry.path) and not self.dirselect: self.open_entry(entry) elif touch.is_double_tap: if self.dirselect and self.file_system.is_dir(entry.path): self.open_entry(entry) else: self.dispatch('on_submit', self.selection, touch)
def open_entry(self, entry): try: # Just check if we can list the directory. This is also what # _add_file does, so if it fails here, it would also fail later # on. Do the check here to prevent setting path to an invalid # directory that we cannot list. self.file_system.listdir(entry.path) except OSError: entry.locked = True else: self.path = join(self.path, entry.path) self.selection = [] def _apply_filters(self, files): if not self.filters: return files filtered = [] for filter in self.filters: if isinstance(filter, collections.Callable): filtered.extend([fn for fn in files if list(filter(self.path, fn))]) else: filtered.extend([fn for fn in files if fnmatch(fn, filter)]) if not self.filter_dirs: dirs = [fn for fn in files if self.file_system.is_dir(fn)] filtered.extend(dirs) return list(set(filtered))
[docs] def get_nice_size(self, fn): '''Pass the filepath. Returns the size in the best human readable format or '' if it is a directory (Don't recursively calculate size.). ''' if self.file_system.is_dir(fn): return '' try: size = self.file_system.getsize(fn) except OSError: return '--' for unit in filesize_units: if size < 1024.0: return '%1.0f %s' % (size, unit) size /= 1024.0
def _update_files(self, *args, **kwargs): # trigger to start gathering the files in the new directory # we'll start a timer that will do the job, 10 times per frames # (default) self._gitems = [] self._gitems_parent = kwargs.get('parent', None) self._gitems_gen = self._generate_file_entries( path=kwargs.get('path', self.path), parent=self._gitems_parent) # cancel any previous clock if exist Clock.unschedule(self._create_files_entries) # show the progression screen self._hide_progress() if self._create_files_entries(): # not enough for creating all the entries, all a clock to continue # start a timer for the next 100 ms Clock.schedule_interval(self._create_files_entries, .1) def _create_files_entries(self, *args): # create maximum entries during 50ms max, or 10 minimum (slow system) # (on a "fast system" (core i7 2700K), we can create up to 40 entries # in 50 ms. So 10 is fine for low system. start = time() finished = False index = total = count = 1 while time() - start < 0.05 or count < 10: try: index, total, item = next(self._gitems_gen) self._gitems.append(item) count += 1 except StopIteration: finished = True break # if this wasn't enough for creating all the entries, show a progress # bar, and report the activity to the user. if not finished: self._show_progress() self._progress.total = total self._progress.index = index return True # we created all the files, now push them on the view self._items = items = self._gitems parent = self._gitems_parent if parent is None: self.dispatch('on_entries_cleared') for entry in items: self.dispatch('on_entry_added', entry, parent) else: parent.entries[:] = items for entry in items: self.dispatch('on_subentry_to_entry', entry, parent) self.files[:] = [file.path for file in items] # stop the progression / creation self._hide_progress() self._gitems = None self._gitems_gen = None Clock.unschedule(self._create_files_entries) return False
[docs] def cancel(self, *largs): '''Cancel any background action started by filechooser, such as loading a new directory. .. versionadded:: 1.2.0 ''' Clock.unschedule(self._create_files_entries) self._hide_progress() if len(self._previous_path) > 1: # if we cancel any action, the path will be set same as the # previous one, so we can safely cancel the update of the previous # path. self.path = self._previous_path[-2] Clock.unschedule(self._update_files)
def _show_progress(self): if self._progress: return cls = self.progress_cls if isinstance(cls, string_types): cls = Factory.get(cls) self._progress = cls(path=self.path) self._progress.value = 0 self.add_widget(self._progress) def _hide_progress(self): if self._progress: self.remove_widget(self._progress) self._progress = None def _generate_file_entries(self, *args, **kwargs): # Generator that will create all the files entries. # the generator is used via _update_files() and _create_files_entries() # don't use it directly. is_root = False path = kwargs.get('path', self.path) have_parent = kwargs.get('parent', None) is not None # Add the components that are always needed if self.rootpath: rootpath = realpath(self.rootpath) path = realpath(path) if not path.startswith(rootpath): self.path = rootpath return elif path == rootpath: is_root = True else: if platform == 'win': is_root = splitdrive(path)[1] in (sep, altsep) elif platform in ('macosx', 'linux', 'android', 'ios'): is_root = normpath(expanduser(path)) == sep else: # Unknown fs, just always add the .. entry but also log Logger.warning('Filechooser: Unsupported OS: %r' % platform) # generate an entries to go back to previous if not is_root and not have_parent: back = '..' + sep pardir = Builder.template(self._ENTRY_TEMPLATE, **dict( name=back, size='', path=back, controller=ref(self), isdir=True, parent=None, sep=sep, get_nice_size=lambda: '')) yield 0, 1, pardir # generate all the entries for files try: for index, total, item in self._add_files(path): yield index, total, item except OSError: Logger.exception('Unable to open directory <%s>' % self.path) self.files[:] = [] def _add_files(self, path, parent=None): path = expanduser(path) files = [] fappend = files.append for f in self.file_system.listdir(path): try: # In the following, use fully qualified filenames fappend(normpath(join(path, f))) except UnicodeDecodeError: Logger.exception('unable to decode <{}>'.format(f)) except UnicodeEncodeError: Logger.exception('unable to encode <{}>'.format(f)) # Apply filename filters files = self._apply_filters(files) # Sort the list of files files = self.sort_func(files, self.file_system) is_hidden = self.file_system.is_hidden if not self.show_hidden: files = [x for x in files if not is_hidden(x)] self.files[:] = files total = len(files) wself = ref(self) for index, fn in enumerate(files): def get_nice_size(): # Use a closure for lazy-loading here return self.get_nice_size(fn) ctx = {'name': basename(fn), 'get_nice_size': get_nice_size, 'path': fn, 'controller': wself, 'isdir': self.file_system.is_dir(fn), 'parent': parent, 'sep': sep} entry = Builder.template(self._ENTRY_TEMPLATE, **ctx) yield index, total, entry def entry_subselect(self, entry): if not self.file_system.is_dir(entry.path): return self._update_files(path=entry.path, parent=entry) def close_subselection(self, entry): for subentry in entry.entries: self.dispatch('on_remove_subentry', subentry, entry)
[docs]class FileChooserListView(FileChooserController): '''Implementation of :class:`FileChooserController` using a list view. ''' _ENTRY_TEMPLATE = 'FileListEntry'
[docs]class FileChooserIconView(FileChooserController): '''Implementation of :class:`FileChooserController` using an icon view. ''' _ENTRY_TEMPLATE = 'FileIconEntry'
if __name__ == '__main__': from kivy.app import App from pprint import pprint import sys class FileChooserApp(App): def build(self): view = FileChooserListView if len(sys.argv) > 1: v = view(path=sys.argv[1]) else: v = view() v.bind(selection=lambda *x: pprint("selection: %s" % x[1:])) v.bind(path=lambda *x: pprint("path: %s" % x[1:])) return v FileChooserApp().run()