Iic tdZddlmZddlmZddlZddlmZGddZGd d eejZ y) a Selection management for wxPython tree controls. Provides a mixin and ready-to-use control that automatically manages selection state for tree controls: - Auto-focus first item when control gains focus with no selection - Helper to maintain valid selection after item deletion Includes: - TreeSelectionManagerMixin: For wx.TreeCtrl-style controls - ManagedTreeCtrl: Ready-to-use tree with selection management Requires: wxPython ) annotations)AnyN)FocusAfterDeletecleZdZdZej d dfdZddZ ddZxZ S) TreeSelectionManagerMixina Mixin that manages selection for wx.TreeCtrl-style controls. Features: - Auto-selects first visible item when focused with no selection - Provides select_after_delete() to maintain valid selection after removal Args: focus_after_delete: Where to focus after deleting an item. FocusAfterDelete.PREVIOUS (default) or FocusAfterDelete.NEXT )focus_after_deletect||i|||_|jtj |j y)N)super__init__r Bindwx EVT_SET_FOCUS_on_tree_focus_ensure_selection)selfr argskwargs __class__s KC:\Users\dbart\PlayPalace11\clients\desktop\ui\enhance_wx\tree_selection.pyr z"TreeSelectionManagerMixin.__init__+s9 $)&)"4 """D$H$HIc|j|j}|jry|j}|jsy|j |\}}|jr|j |yy)zGAuto-select first visible item when tree gains focus with no selection.N)Skip GetSelectionIsOk GetRootItem GetFirstChild SelectItem)revtselroot first_child_s rrz9TreeSelectionManagerMixin._on_tree_focus_ensure_selection5sp  ! 88: !yy{ ++D1 Q     OOK ( rc|jtjk(r||}}n||}}d}|r|jr|}n<|r|jr|}n'|r%|jr||j k7r|}|r|j |yy)aSelect an appropriate item after deletion. Call this AFTER deleting a tree item to move selection to the previous sibling, next sibling, or parent -- in order of preference based on the focus_after_delete setting. Args: parent: Parent of the deleted item. prev_sibling: Sibling before the deleted item, or None. next_sibling: Sibling after the deleted item, or None. N)r rPREVIOUSrrr)rparent prev_sibling next_siblingprimaryfallbacktargets rselect_after_deletez-TreeSelectionManagerMixin.select_after_deleteBs"  " "&6&?&? ? ,lXG ,lXG w||~F (--/F  &D4D4D4F*FF  OOF # r)rrr rrrreturnNone)rz wx.FocusEventr,r-)r%z wx.TreeItemIdr&wx.TreeItemId | Noner'r.r,r-) __name__ __module__ __qualname____doc__rr$r rr+ __classcell__)rs@rrrst 0@/H/HJJ-J J  J )$$+$+ $  $rrceZdZdZy)ManagedTreeCtrlz2A wx.TreeCtrl with automatic selection management.N)r/r0r1r2rrr5r5is<rr5) r2 __future__rtypingrrlist_selectionrrTreeCtrlr5r6rrr;s8 # ,C$C$V / r