Source code for pyworkflow.gui.browser

# **************************************************************************
# *
# * Authors:     J.M. De la Rosa Trevin (jmdelarosa@cnb.csic.es) [1]
# *              Jose Gutierrez (jose.gutierrez@cnb.csic.es) [2]
# *
# * [1] SciLifeLab, Stockholm University
# * [2] Unidad de  Bioinformatica of Centro Nacional de Biotecnologia , CSIC
# *
# * This program is free software: you can redistribute it and/or modify
# * it under the terms of the GNU General Public License as published by
# * the Free Software Foundation, either version 3 of the License, or
# * (at your option) any later version.
# *
# * This program is distributed in the hope that it will be useful,
# * but WITHOUT ANY WARRANTY; without even the implied warranty of
# * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# * GNU General Public License for more details.
# *
# * You should have received a copy of the GNU General Public License
# * along with this program.  If not, see <https://www.gnu.org/licenses/>.
# *
# *  All comments concerning this program package may be sent to the
# *  e-mail address 'scipion@cnb.csic.es'
# *
# **************************************************************************
"""
In this module a simple ObjectBrowser is implemented.
This class can be subclasses to extend its functionality.
A concrete use of ObjectBrowser is FileBrowser, where the
elements to inspect and preview are files.
"""
import os.path
import stat
import tkinter as tk
import time

import pyworkflow.utils as pwutils
from . import gui
from .tree import BoundTree, TreeProvider
from .text import TaggedText, openTextFileEditor
from .widgets import Button, HotButton
from .. import Config

PARENT_FOLDER = ".."


[docs]class ObjectBrowser(tk.Frame): """ This class will implement a simple object browser. Basically, it will display a list of elements at the left panel and can display a preview and description on the right panel for the selected element. An ObjectView will be used to grab information for each element such as: icon, preview and description. A TreeProvider will be used to populate the list (Tree). """ def __init__(self, parent, treeProvider, showPreview=True, showPreviewTop=True, **args): tk.Frame.__init__(self, parent, **args) self.treeProvider = treeProvider self._lastSelected = None gui.configureWeigths(self) self.showPreviewTop = showPreviewTop # The main layout will be two panes, # At the left containing the elements list # and the right containing the preview and description p = tk.PanedWindow(self, orient=tk.HORIZONTAL) p.grid(row=0, column=0, sticky='news') leftPanel = tk.Frame(p) self._fillLeftPanel(leftPanel) p.add(leftPanel, padx=5, pady=5) p.paneconfig(leftPanel, minsize=300) if showPreview: rightPanel = tk.Frame(p) self._fillRightPanel(rightPanel) p.add(rightPanel, padx=5, pady=5) p.paneconfig(rightPanel, minsize=200) # Register a callback when the item is clicked self.tree.itemClick = self._itemClicked def _fillLeftPanel(self, frame): gui.configureWeigths(frame) self.tree = BoundTree(frame, self.treeProvider) self.tree.grid(row=0, column=0, sticky='news') self.itemConfig = self.tree.itemConfig self.getImage = self.tree.getImage def _fillRightPanel(self, frame): frame.columnconfigure(0, weight=1) if self.showPreviewTop: top = tk.Frame(frame) top.grid(row=0, column=0, sticky='news') frame.rowconfigure(0, weight=3) gui.configureWeigths(top) top.rowconfigure(0, minsize=200) self._fillRightTop(top) bottom = tk.Frame(frame) bottom.grid(row=1, column=0, sticky='news') frame.rowconfigure(1, weight=1) gui.configureWeigths(bottom) bottom.rowconfigure(1, weight=1) self._fillRightBottom(bottom) def _fillRightTop(self, top): self.noImage = self.getImage('no-image128.gif') self.label = tk.Label(top, image=self.noImage) self.label.grid(row=0, column=0, sticky='news') def _fillRightBottom(self, bottom): self.text = TaggedText(bottom, width=40, height=15, bg='white', takefocus=0) self.text.grid(row=0, column=0, sticky='news') def _itemClicked(self, obj): self._lastSelected = obj img, desc = self.treeProvider.getObjectPreview(obj) # Update image preview if self.showPreviewTop: if isinstance(img, str): img = self.getImage(img) if img is None: img = self.noImage self.label.config(image=img) # Update text preview self.text.setReadOnly(False) self.text.clear() if desc is not None: self.text.addText(desc) self.text.setReadOnly(True) if hasattr(self, 'entryLabel') and not self._lastSelected.isDir(): self.entryVar.set(self._lastSelected.getFileName())
[docs] def getSelected(self): """ Return the selected object. """ return self._lastSelected
# ------------ Classes and Functions related to File browsing --------------
[docs]class FileInfo(object): """ This class will store some information about a file. It will serve to display files items in the Tree. """ def __init__(self, path, filename): self._fullpath = os.path.join(path, filename) self._filename = filename if os.path.exists(self._fullpath): self._stat = os.stat(self._fullpath) else: self._stat = None
[docs] def isDir(self): return stat.S_ISDIR(self._stat.st_mode) if self._stat else False
[docs] def getFileName(self): return self._filename
[docs] def getPath(self): return self._fullpath
[docs] def getSize(self): return self._stat.st_size if self._stat else 0
[docs] def getSizeStr(self): """ Return a human readable string of the file size.""" return pwutils.prettySize(self.getSize()) if self._stat else '0'
[docs] def getDateStr(self): return pwutils.dateStr(self.getDate()) if self._stat else '0'
[docs] def getDate(self): return self._stat.st_mtime if self._stat else 0
[docs]class FileHandler(object): """ This class will be used to get the icon, preview and info from the different types of objects. It should be used with FileTreeProvider, where different types of handlers can be registered. """
[docs] def getFileIcon(self, objFile): """ Return the icon name for a given file. """ if objFile.isDir(): icon = 'file_folder.gif' else: icon = 'file_generic.gif' return icon
[docs] def getFilePreview(self, objFile): """ Return the preview image and description for the specific object.""" if objFile.isDir(): return 'fa-folder-open.gif', None return None, None
[docs] def getFileActions(self, objFile): """ Return actions that can be done with this object. Actions will be displayed in the context menu and the first one will be the default when double-click. """ return []
[docs]class TextFileHandler(FileHandler): def __init__(self, textIcon): FileHandler.__init__(self) self._icon = textIcon
[docs] def getFileIcon(self, objFile): return self._icon
[docs]class SqlFileHandler(FileHandler):
[docs] def getFileIcon(self, objFile): return 'file_sqlite.gif'
[docs]class FileTreeProvider(TreeProvider): """ Populate a tree with files and folders of a given path """ _FILE_HANDLERS = {} _DEFAULT_HANDLER = FileHandler() FILE_COLUMN = 'File' SIZE_COLUMN = 'Size'
[docs] @classmethod def registerFileHandler(cls, fileHandler, *extensions): """ Register a FileHandler for a given file extension. Params: fileHandler: the FileHandler that will take care of extensions. *extensions: the extensions list that will be associated to this FileHandler. """ for fileExt in extensions: cls._FILE_HANDLERS[fileExt] = fileHandler
def __init__(self, currentDir=None, showHidden=False, onlyFolders=False): TreeProvider.__init__(self, sortingColumnName=self.FILE_COLUMN) self._currentDir = os.path.abspath(currentDir) self._showHidden = showHidden self._onlyFolders = onlyFolders self.getColumns = lambda: [(self.FILE_COLUMN, 300), (self.SIZE_COLUMN, 70), ('Time', 150)]
[docs] def getFileHandler(self, obj): filename = obj.getFileName() fileExt = pwutils.getExt(filename) return self._FILE_HANDLERS.get(fileExt, self._DEFAULT_HANDLER)
[docs] def getObjectInfo(self, obj): filename = obj.getFileName() fileHandler = self.getFileHandler(obj) icon = fileHandler.getFileIcon(obj) info = {'key': filename, 'text': filename, 'values': (obj.getSizeStr(), obj.getDateStr()), 'image': icon } return info
[docs] def getObjectPreview(self, obj): fileHandler = self.getFileHandler(obj) return fileHandler.getFilePreview(obj)
[docs] def getObjectActions(self, obj): fileHandler = self.getFileHandler(obj) actions = fileHandler.getFileActions(obj) # Always allow the option to open as text # specially useful for unknown formats fn = obj.getPath() actions.append(("Open external Editor", lambda: openTextFileEditor(fn), pwutils.Icon.ACTION_REFERENCES)) return actions
[docs] def getObjects(self): fileInfoList = [] if not self._currentDir == pwutils.ROOT: fileInfoList.append(FileInfo(self._currentDir, PARENT_FOLDER)) try: # This might fail if there is not granted files = os.listdir(self._currentDir) for f in files: fullPath = os.path.join(self._currentDir, f) # If f is a file and only need folders if self._onlyFolders and not os.path.isdir(fullPath): continue # Do not add hidden files if not requested if not self._showHidden and f.startswith('.'): continue # All ok...add item. fileInfoList.append(FileInfo(self._currentDir, f)) except Exception as e: print("Can't list files at " + self._currentDir, e) # Sort objects fileInfoList.sort(key=self.fileKey, reverse=not self.isSortingAscending()) return fileInfoList
[docs] def fileKey(self, f): sortDict = {self.FILE_COLUMN: 'getFileName', self.SIZE_COLUMN: 'getSize'} return getattr(f, sortDict.get(self._sortingColumnName, 'getDate'))()
[docs] def getDir(self): return self._currentDir
[docs] def setDir(self, newPath): self._currentDir = newPath
# Some constants for the type of selection # when the file browser is opened SELECT_NONE = 0 # No selection, just browse files SELECT_FILE = 1 SELECT_FOLDER = 2 SELECT_PATH = 3 # Can be either file or folder
[docs]class FileBrowser(ObjectBrowser): """ The FileBrowser is a particular class of ObjectBrowser where the "objects" are just files and directories. """ def __init__(self, parent, initialDir='.', selectionType=SELECT_FILE, selectionSingle=True, allowFilter=True, filterFunction=None, previewDim=144, showHidden=False, # Show hidden files or not? selectButton='Select', # Change the Select button text entryLabel=None, # Display an entry for some input entryValue='', # Display a value in the entry field showInfo=None, # Used to notify errors or messages shortCuts=None, # Shortcuts to common locations/paths onlyFolders=False ): self.pathVar = tk.StringVar() self.pathVar.set(os.path.abspath(initialDir)) self.pathEntry = None self.previousSearch = None self.previousSearchTS = None self.shortCuts = shortCuts self._provider = FileTreeProvider(initialDir, showHidden, onlyFolders) self.selectButton = selectButton self.entryLabel = entryLabel self.entryVar = tk.StringVar() self.entryVar.set(entryValue) self.showInfo = showInfo or self._showInfo ObjectBrowser.__init__(self, parent, self._provider) # focuses on the browser in order to allow to move with the keyboard self._goDir(os.path.abspath(initialDir)) if selectionType == SELECT_NONE: selectButton = None buttonsFrame = tk.Frame(self) self._fillButtonsFrame(buttonsFrame) buttonsFrame.grid(row=1, column=0) def _showInfo(self, msg): """ Default way (print to console) to show a message with a given info. """ print(msg) def _fillLeftPanel(self, frame): """ Redefine this method to include a buttons toolbar and also include a filter bar at the bottom of the Tree. """ # Tree with files frame.columnconfigure(0, weight=1) treeFrame = tk.Frame(frame) ObjectBrowser._fillLeftPanel(self, treeFrame) # Register the double-click event self.tree.itemDoubleClick = self._itemDoubleClick # Register keypress event self.tree.itemKeyPressed = self._itemKeyPressed treeRow = 3 treeFrame.grid(row=treeRow, column=0, sticky='news') # Toolbar frame toolbarFrame = tk.Frame(frame) self._fillToolbar(toolbarFrame) toolbarFrame.grid(row=0, column=0, sticky='new') pathFrame = tk.Frame(frame) pathLabel = tk.Label(pathFrame, text='Path') pathLabel.grid(row=0, column=0, padx=0, pady=3) pathEntry = tk.Entry(pathFrame, bg='white', width=65, textvariable=self.pathVar, font=gui.getDefaultFont()) pathEntry.grid(row=0, column=1, sticky='new', pady=3) pathEntry.bind("<Return>", self._onEnterPath) pathEntry.bind("<KP_Enter>", self._onEnterPath) self.pathEntry = pathEntry pathFrame.grid(row=1, column=0, sticky='new') # Entry frame, could be used for filter if self.entryLabel: entryFrame = tk.Frame(frame) entryFrame.grid(row=2, column=0, sticky='new') tk.Label(entryFrame, text=self.entryLabel).grid(row=0, column=0, sticky='nw', pady=3) tk.Entry(entryFrame, textvariable=self.entryVar, bg='white', width=65).grid(row=0, column=1, sticky='nw', pady=3) frame.rowconfigure(treeRow, weight=1) def _addButton(self, frame, text, image, command): btn = tk.Label(frame, text=text, image=self.getImage(image), compound=tk.LEFT, cursor='hand2') btn.bind('<Button-1>', command) btn.grid(row=0, column=self._col, sticky='nw', padx=(0, 5), pady=5) self._col += 1 def _fillToolbar(self, frame): """ Fill the toolbar frame with some buttons. """ self._col = 0 self._addButton(frame, 'Refresh', pwutils.Icon.ACTION_REFRESH, self._actionRefresh) self._addButton(frame, 'Home', pwutils.Icon.HOME, self._actionHome) self._addButton(frame, 'Launch folder', pwutils.Icon.ROCKET, self._actionLaunchFolder) self._addButton(frame, 'Working dir', pwutils.Icon.ACTION_BROWSE, self._actionWorkingDir) self._addButton(frame, 'Up', pwutils.Icon.ARROW_UP, self._actionUp) # Add shortcuts self._addShortCuts(frame) def _addShortCuts(self, frame): """ Add shortcuts if available""" if self.shortCuts: for shortCut in self.shortCuts: self._addButton(frame, shortCut.name, shortCut.icon, lambda e: self._goDir(shortCut.path)) def _fillButtonsFrame(self, frame): """ Add button to the bottom frame if the selectMode is distinct from SELECT_NONE. """ Button(frame, "Close", pwutils.Icon.BUTTON_CLOSE, command=self._close).grid(row=0, column=0, padx=(0, 5)) if self.selectButton: HotButton(frame, self.selectButton, pwutils.Icon.BUTTON_SELECT, command=self._select).grid(row=0, column=1) def _actionRefresh(self, e=None): self.tree.update() def _goDir(self, newDir): newDir = os.path.abspath(newDir) # Add a final "/" to the path: abspath is removing it except for "/" if not newDir.endswith(os.path.sep): newDir += os.path.sep self.pathVar.set(newDir) self.pathEntry.icursor(len(newDir)) self.treeProvider.setDir(newDir) self.tree.update() self.tree.focus_set() itemKeyToFocus = PARENT_FOLDER if PARENT_FOLDER not in self.tree._objDict: itemKeyToFocus = self.tree.get_children()[0] # Focusing on a item, but nothing is selected # Current dir remains in _lastSelected self._lastSelected = FileInfo(os.path.dirname(newDir), os.path.basename(newDir)) self.tree.focus(itemKeyToFocus) def _actionUp(self, e=None): parentFolder = pwutils.getParentFolder(self.treeProvider.getDir()) self._goDir(parentFolder) def _actionHome(self, e=None): self._goDir(pwutils.getHomePath()) def _actionRoot(self, e=None): self._goDir("/") def _actionLaunchFolder(self, e=None): self._goDir(Config.SCIPION_CWD) def _actionWorkingDir(self, e=None): self._goDir(os.getcwd()) def _itemDoubleClick(self, obj): if obj.isDir(): self._goDir(obj.getPath()) else: actions = self._provider.getObjectActions(obj) if actions: # actions[0] = first Action, [1] = the action callback actions[0][1]() def _itemKeyPressed(self, obj, e=None): if e.keysym in [pwutils.KEYSYM.RETURN]: self._itemDoubleClick(obj) return textToSearch = self._composeTextToSearch(e.char) # locate an item in starting with that letter. self._searchItem(textToSearch) def _composeTextToSearch(self, newChar): currentMiliseconds = time.time() if (self.previousSearchTS is not None) and \ ((currentMiliseconds - self.previousSearchTS) < 0.3): newChar = self.previousSearch + newChar self.previousSearch = newChar self.previousSearchTS = currentMiliseconds return newChar def _searchItem(self, char): """ locate an item in starting with that letter.""" try: self.tree.search(char) except Exception as e: # seems to raise an exception but selects things right. pass def _onEnterPath(self, e=None): path = os.path.abspath(self.pathVar.get()) if os.path.exists(path): self._goDir(path) else: self.showInfo("Path '%s' does not exists. " % path) self.pathEntry.focus()
[docs] def onClose(self): pass
[docs] def onSelect(self, obj): print(obj, "type: ", type(obj))
def _close(self, e=None): self.onClose() def _select(self, e=None): _lastSelected = self.getSelected() if _lastSelected is not None: self.onSelect(_lastSelected) else: print('Select a valid file/folder')
[docs] def getEntryValue(self): return self.entryVar.get()
[docs] def getCurrentDir(self): return self.treeProvider.getDir()
[docs]class ShortCut: """ Shortcuts to paths to be displayed in the file browser"""
[docs] @staticmethod def factory(path, name, icon=None, toolTip=""): """ Factory method to create shortcuts""" return ShortCut(path, name, icon, toolTip)
def __init__(self, path, name, icon=None, toolTip=""): self.path = path self.name = name self.icon = icon self.toolTip = toolTip
[docs]class BrowserWindow(gui.Window): """ Windows to hold a browser frame inside. """ def __init__(self, title, master=None, **kwargs): if 'minsize' not in kwargs: kwargs['minsize'] = (800, 400) gui.Window.__init__(self, title, master, **kwargs)
[docs] def setBrowser(self, browser, row=0, column=0): self.browser = browser browser.grid(row=row, column=column, sticky='news') self.itemConfig = browser.tree.itemConfig
STANDARD_IMAGE_EXTENSIONS = ['.png', '.jpg', '.jpeg']
[docs]def isStandardImage(filename): """ Check if a filename have an standard image extension. """ fnLower = filename.lower() return any(fnLower.endswith(ext) for ext in STANDARD_IMAGE_EXTENSIONS)
[docs]class FileBrowserWindow(BrowserWindow): """ Windows to hold a file browser frame inside. """ def __init__(self, title, master=None, path=None, onSelect=None, shortCuts=None, **kwargs): BrowserWindow.__init__(self, title, master, **kwargs) self.registerHandlers() browser = FileBrowser(self.root, path, showInfo=lambda msg: self.showInfo(msg, "Info"), shortCuts=shortCuts, **kwargs) if onSelect: def selected(obj): onSelect(obj) self.close() browser.onSelect = selected browser.onClose = self.close self.setBrowser(browser)
[docs] def getEntryValue(self): return self.browser.getEntryValue()
[docs] def getCurrentDir(self): return self.browser.getCurrentDir()
[docs] def registerHandlers(self): register = FileTreeProvider.registerFileHandler # shortcut register(TextFileHandler('file_text.gif'), '.txt', '.log', '.out', '.err', '.stdout', '.stderr', '.emx', '.json', '.xml', '.pam') register(TextFileHandler('file_python.gif'), '.py') register(TextFileHandler('file_java.gif'), '.java') register(SqlFileHandler(), '.sqlite', '.db')
# register(MdFileHandler(), '.xmd', '.star', '.pos', '.ctfparam', '.doc') # register(ParticleFileHandler(), # '.xmp', '.tif', '.tiff', '.spi', '.mrc', '.map', '.raw', # '.inf', '.dm3', '.em', '.pif', '.psd', '.spe', '.ser', '.img', # '.hed', *STANDARD_IMAGE_EXTENSIONS) # register(VolFileHandler(), '.vol') # register(StackHandler(), '.stk', '.mrcs', '.st', '.pif', '.dm4') # register(ChimeraHandler(), '.bild')