from functools import lru_cache import logging from pathlib import Path from types import MappingProxyType from typing import List from PySide6.QtCore import ( QFileSystemWatcher, QObject, Qt, QTimer, Signal, Slot, ) from PySide6.QtWidgets import ( QMenu, QTreeWidget, ) from datalad.interface.base import Interface from datalad.utils import get_dataset_root from datalad.dataset.gitrepo import GitRepo from .cmd_actions import add_cmd_actions_to_menu from .fsbrowser_item import FSBrowserItem from .lsdir import GooeyLsDir from .status_light import GooeyStatusLight lgr = logging.getLogger('datalad.gooey.fsbrowser') class GooeyFilesystemBrowser(QObject): # TODO Establish ENUM for columns # FSBrowserItem item_requires_annotation = Signal(FSBrowserItem) # DONE def __init__(self, app, path: Path, treewidget: QTreeWidget): super().__init__() tw = treewidget # TODO must setColumnNumber() self._app = app self._fswatcher = QFileSystemWatcher(parent=app) self.item_requires_annotation.connect( self._queue_item_for_annotation) tw.setHeaderLabels(['Name', 'Type', 'State']) # established defined sorting order of the tree tw.sortItems(1, Qt.AscendingOrder) # establish the root item, based on a fake lsdir result # the info needed is so simple, it is not worth a command # execution root = FSBrowserItem.from_lsdir_result( dict( path=path, type='dataset' if GitRepo.is_valid(path) else 'directory', ), parent=tw, ) # set the tooltip to the full path, otherwise only names are shown root.setToolTip(0, str(path)) tw.addTopLevelItem(root) self._root_item = root tw.customContextMenuRequested.connect( self._custom_context_menu) self._tree = tw # whenever a treeview node is expanded, add the path to the fswatcher tw.itemExpanded.connect(self._watch_dir) # and also populate it with items for contained paths tw.itemExpanded.connect(self._populate_item) tw.itemCollapsed.connect(self._unwatch_dir) self._fswatcher.directoryChanged.connect(self._inspect_changed_dir) # items of directories to be annotated, populated by # _queue_item_for_annotation() self._annotation_queue = set() # msec self._annotation_timer_interval = 3000 self._annotation_timer = QTimer(self) self._annotation_timer.timeout.connect( self._process_item_annotation_queue) self._annotation_timer.start(self._annotation_timer_interval) self._app._cmdexec.results_received.connect( self._cmdexec_results_handler) def _populate_item(self, item): if item.childCount(): return # only parse, if there are no children yet # kick off lsdir command in the background self._populate_and_annotate(item, no_existing_children=True) def _populate_and_annotate(self, item, no_existing_children): self._app.execute_dataladcmd.emit( 'gooey_lsdir', MappingProxyType(dict( path=item.pathobj, result_renderer='disabled', on_failure='ignore', return_type='generator', )), MappingProxyType(dict( preferred_result_interval=0.2, result_override=dict( gooey_parent_item=item, gooey_no_existing_item=no_existing_children, ), )), ) # for now we register the parent for an annotation update # but we could also report the specific path and let the # annotation code figure out the optimal way. # at present however, we get here for items of a whole dir # being reported at once. self._queue_item_for_annotation(item) @Slot(Interface, list) def _cmdexec_results_handler(self, cls, res): res_handler = None if cls == GooeyLsDir: res_handler = self._lsdir_result_receiver elif cls == GooeyStatusLight: res_handler = self._status_result_receiver else: lgr.debug('FSBrowser has no handler for result from %s', cls) return for r in res: res_handler(r) def _lsdir_result_receiver(self, res): if res.get('action') != 'gooey-lsdir': # no what we are looking for return target_item = None target_item_parent = res.get('gooey_parent_item') no_existing_item = res.get('gooey_no_existing_item', False) ipath = Path(res['path']) if target_item_parent is None: # we did not get it delivered in the result, search for it try: target_item_parent = self._get_item_from_path(ipath.parent) except ValueError: # ok, now we have no clue what this lsdir result is about # its parent is no in the tree return if (no_existing_item and target_item_parent and target_item_parent.pathobj == ipath): # sender claims that the item does not exist and provided a parent # item. reject a result if it matches the parent to avoid # duplicating the item as a child, and to also prevent an unintended # item update return if not no_existing_item: # we have no indication that the item this is about does not # already exist, search for it try: # give the parent as a starting item, to speed things up target_item = self._get_item_from_path( ipath, target_item_parent) except ValueError: # it is quite possible that the item does not exist yet. # but such cases are expensive, and the triggering code could # consider sending the 'gooey_no_existing_item' flag pass if target_item is None: # we don't have such an item yet -> make one target_item = FSBrowserItem.from_lsdir_result( res, target_item_parent) else: # we do have this already, good occasion to update it target_item.update_from_lsdir_result(res) @lru_cache(maxsize=1000) def _get_item_from_path(self, path: Path, root: FSBrowserItem = None): # this is a key function in terms of result UI snappiness # it must be as fast as anyhow possible item = self._root_item if root is None else root ipath = item.pathobj if path == ipath: return item # otherwise look for the item with the right name at the # respective level try: return self._get_item_from_trace( item, path.relative_to(ipath).parts) except ValueError as e: raise ValueError(f'Cannot find item for {path}') from e def _get_item_from_trace(self, root: FSBrowserItem, trace: List): item = root for p in trace: item = item[p] if item is None: raise ValueError(f'Cannot find item for {trace}') continue return item def _queue_item_for_annotation(self, item): """This is not thread-safe `item` should be of type 'directory' or 'dataset' for meaningful behavior. """ # wait for at least half a sec longer after a new request came in # to avoid DDOS'ing the facility? if self._annotation_timer.remainingTime() < 500: self._annotation_timer.start(500) self._annotation_queue.add(item) def _process_item_annotation_queue(self): if not self._annotation_queue: return if self._app._cmdexec.n_running: # stuff is still running # make sure the population of the tree items is done too! self._annotation_timer.start(1000) return # there is stuff to annotate, make sure we do not trigger more # annotations while this one is running self._annotation_timer.stop() print("ANNOTATE!", len(self._annotation_queue)) # TODO stuff could be optimized here: collapsing multiple # directories belonging to the same dataset into a single `status` # call... while self._annotation_queue: # process the queue in reverse order, assuming a user would be # interested in the last triggered directory first # (i.e., assumption is: expanding tree nodes one after # another, attention would be on the last expanded one, not the # first) item = self._annotation_queue.pop() print('->', item) ipath = item.pathobj dsroot = get_dataset_root(ipath) if dsroot is None: # no containing dataset, by definition everything is untracked for child in item.children_(): # get type, only annotate non-directory items if child.datalad_type != 'directory': child.update_from_status_result( dict(state='untracked')) else: # trigger datalad-gooey-status-light execution # giving the target directory as a `path` argument should # avoid undesired recursion into subDIRECTORIES paths_to_investigate = [ c.pathobj.relative_to(dsroot) for c in item.children_() if c.datalad_type != 'directory' ] if paths_to_investigate: # do not run, if there are no relevant paths to inspect self._app.execute_dataladcmd.emit( 'gooey_status_light', MappingProxyType(dict( dataset=dsroot, path=[ipath], #annex='basic', result_renderer='disabled', on_failure='ignore', return_type='generator', )), MappingProxyType(dict( preferred_result_interval=3.0, result_override=dict( gooey_parent_item=item, ), )), ) # restart annotation watcher self._annotation_timer.start(self._annotation_timer_interval) def _status_result_receiver(self, res): if res.get('action') != 'status': # no what we are looking for return path = res.get('path') if path is None: # nothing that we could handle return try: target_item = self._get_item_from_trace( res['gooey_parent_item'], # the parent will only ever be the literal parent directory [Path(path).name], ) except ValueError: # the corersponding item is no longer around return target_item.update_from_status_result(res) # DONE def _watch_dir(self, item): path = item.pathobj lgr.log( 9, "GooeyFilesystemBrowser._watch_dir(%r)", path, ) self._fswatcher.addPath(str(path)) if item.datalad_type == 'dataset': # for a repository, also watch its .git to become aware of more # Git operation outcomes. specifically watch the HEADS to catch # updates on any branch self._fswatcher.addPath(str(path / '.git' / 'refs' / 'heads')) # DONE # https://github.com/datalad/datalad-gooey/issues/50 def _unwatch_dir(self, item): path = str(item.pathobj) lgr.log( 9, "GooeyFilesystemBrowser._unwatch_dir(%r) -> %r", path, self._fswatcher.removePath(path), ) # DONE def _inspect_changed_dir(self, path: str): pathobj = Path(path) # look for special case of the internals of a dataset having changed path_parts = pathobj.parts if len(path_parts) > 3 \ and path_parts[-3:] == ('.git', 'refs', 'heads'): # yep, happened -- inspect corresponding dataset root self._inspect_changed_dir(pathobj.parent.parent.parent) return lgr.log(9, "GooeyFilesystemBrowser._inspect_changed_dir(%r)", pathobj) # we need to know the item in the tree corresponding # to the changed directory try: item = self._get_item_from_path(pathobj) except ValueError: # the changed dir has no (longer) a matching entry in the # tree model. make sure to take it off the watch list self._fswatcher.removePath(path) lgr.log(9, "_inspect_changed_dir() -> not in view (anymore), " "removed from watcher") return parent = item.parent() if not pathobj.exists(): if parent is None: # TODO we could have lost the root dir -> special action raise NotImplementedError parent.removeChild(item) lgr.log(8, "-> _inspect_changed_dir() -> item removed") return # we will kick off a `lsdir` run to update the widget, but it could # no detect item that no longer have a file system counterpart # so we remove them here and now for child in item.children_(): try: # same as lexists() but with pathlib child.pathobj.lstat() except (OSError, ValueError): item.removeChild(child) # now re-explore self._populate_and_annotate(item, no_existing_children=False) lgr.log(9, "_inspect_changed_dir() -> requested update") # DONE def _custom_context_menu(self, onpoint): """Present a context menu for the item click in the directory browser """ # get the tree item for the coordinate that received the # context menu request item = self._tree.itemAt(onpoint) if not item: # prevent context menus when the request did not actually # land on an item return # what kind of path is this item representing path_type = item.datalad_type if path_type is None: # we don't know what to do with this (but it also is not expected # to happen) return ipath = item.pathobj cmdkwargs = dict() context = QMenu(parent=self._tree) if path_type == 'dataset': from .active_suite import dataset_api as cmdapi submenu = context.addMenu('Dataset commands') cmdkwargs['dataset'] = ipath elif path_type == 'directory': dsroot = get_dataset_root(ipath) # path the directory path to the command's `path` argument cmdkwargs['path'] = ipath if dsroot: from .active_suite import directory_in_ds_api as cmdapi # also pass dsroot cmdkwargs['dataset'] = dsroot else: from .active_suite import directory_api as cmdapi submenu = context.addMenu('Directory commands') elif path_type in ('file', 'symlink', 'annexed-file'): dsroot = get_dataset_root(ipath) cmdkwargs['path'] = ipath if dsroot: if path_type == 'annexed-file': from .active_suite import annexed_file_api as cmdapi else: from .active_suite import file_in_ds_api as cmdapi cmdkwargs['dataset'] = dsroot else: from .active_suite import file_api as cmdapi submenu = context.addMenu('File commands') # TODO context menu for annex'ed files add_cmd_actions_to_menu( self._tree, self._app._cmdui.configure, cmdapi, submenu, cmdkwargs, ) if not context.isEmpty(): # present the menu at the clicked point context.exec(self._tree.viewport().mapToGlobal(onpoint))