py10x-universe 0.1.3__py3-none-any.whl

ui_10x/entity_stocker.py

@@ -0,0 +1,173 @@
+from typing import Any
+
+from core_10x.traitable import RT, Traitable
+
+from ui_10x.traitable_editor import TraitableEditor, TraitableView
+from ui_10x.utils import ux, ux_answer, ux_push_button, ux_success, ux_warning
+
+
+class StockerPlug(Traitable):
+    master: Traitable
+    # fmt: off
+    current_class_trait_name: str   = RT('current_class')
+    current_entity_trait_name: str  = RT('current_entity')
+    changed_entity_cb_name: str     = RT('on_changed_entity')        #-- f(ce: Traitable) - after accepting edits and reloading
+    deleted_entity_cb_name: str     = RT('on_deleted_entity')        #-- f(de: Traitable)
+    # fmt: on
+
+    new_entity_cb: Any
+    changed_entity_cb: Any
+    deleted_entity_cb: Any
+
+    current_class: type
+    current_entity: Traitable
+
+    def _cb(self, cb_name: str):
+        m = self.master
+        if m:
+            cls = m.__class__
+            method = getattr(cls, cb_name, None)
+            if method:
+                return lambda e: method(m, e)
+
+        return lambda e: None
+
+    # fmt: off
+    def changed_entity_cb_get(self):    return self._cb(self.changed_entity_cb_name)
+    def deleted_entity_cb_get(self):    return self._cb(self.deleted_entity_cb_name)
+    # fmt: on
+
+    def current_class_get(self) -> type:
+        m = self.master
+        if not m:
+            return None
+
+        cls = m.get_value(self.current_class_trait_name)
+        if not cls:
+            ce = self.current_entity
+            if ce:
+                cls = ce.__class__
+
+        return cls
+
+    def current_entity_get(self) -> Traitable:
+        m = self.master
+        if not m:
+            return None
+
+        return m.get_value(self.current_entity_trait_name)
+
+
+class EntityStocker(Traitable):
+    plug: StockerPlug
+    entity_viewer: TraitableEditor
+    buttons_spec: dict
+
+    def entity_viewer_get(self) -> TraitableEditor:
+        ce = self.plug.current_entity
+        if not ce:
+            return None
+
+        return TraitableEditor.editor(ce, view=TraitableView.default(ce.__class__, read_only=True))
+
+    def top_layout(self) -> ux.HBoxLayout:
+        lay = ux.HBoxLayout()
+        lay.set_spacing(0)
+
+        for name, (cb, icon) in self.buttons_spec.items():
+            lay.add_widget(ux_push_button(name, callback=cb, style_icon=icon))
+
+        return lay
+
+    def main_widget(self) -> ux.Widget:
+        entity_viewer = self.entity_viewer
+        if entity_viewer:
+            lay = ux.VBoxLayout()
+            lay.add_layout(self.top_layout(), stretch=0)
+            lay.add_widget(entity_viewer.main_widget())
+
+        else:
+            lay = self.top_layout()
+
+        w = ux.Widget()
+        w.set_layout(lay)
+        return w
+
+    # def on_new_entity(self):
+    #     cls = self.plug.current_class
+    #     if cls:
+    #         new_entity = cls()
+    #         ed = TraitableEditor.editor(new_entity)
+    #         if ed.popup(copy_entity = False, title = f'New Entity of {cls.__name__}', save = True):
+    #             rc = new_entity.share(False)    #-- not accepting existing entity values, if any
+    #             if not rc:
+    #                 ux_warning(rc.error(), parent = None)   #-- TODO: parent must be an existing appropriate widget
+    #                 #-- TODO: what to do with a temp new_entity?
+    #
+    #             else:
+    #                 self.plug.new_entity_cb(new_entity)
+
+    def on_edit_entity(self):
+        ce = self.plug.current_entity
+        if ce:
+            ed = TraitableEditor.editor(ce)
+            if ed.popup():
+                self.plug.changed_entity_cb(ce)
+
+    def on_reload_entity(self):
+        ce = self.plug.current_entity
+        if ce:
+            if not ce.reload():
+                ux_warning(f'Failed to reload entity {ce.__class__}/{ce.id()}', parent=None, on_close=lambda ctx: None)
+            else:
+                self.plug.changed_entity_cb(ce)
+
+    def on_save_entity(self):
+        ce = self.plug.current_entity
+        if ce:
+            try:
+                rc = ce.save()
+                if not rc:
+                    ux_warning(rc.error(), parent=None, on_close=lambda ctx: None)
+
+                return
+
+            except Exception:  # -- revision conflict?
+                # -- TODO: resolve revision conflict - MergingEditor
+                ...
+
+            # -- The conflict seems to be resolved, try again
+            try:
+                rc = ce.save()
+                if not rc:
+                    ux_warning(rc.error(), parent=None, on_close=lambda ctx: None)
+                    return
+
+            except Exception:
+                ux_warning('Failed to resolve revision conflict (most probably due to continuous updates from other session(s)')
+                return
+
+            ux_success(f'Conflict resolved, {ce.__class__}/{ce.id()} has been saved', on_close=lambda ctx: None)
+
+    def on_delete_entity(self):
+        ce = self.plug.current_entity
+        if ce:
+
+            def on_close(accepted: bool):
+                if accepted:
+                    if not ce.delete():
+                        ux_warning('Deletion failed: {ce.__class__}/{ce.id()', parent=None, on_close=lambda ctx: None)
+                    self.plug.deleted_entity_cb(ce)
+
+            ux_answer(f'Please confirm deletion of {ce.__class__}/{ce.id()}', parent=None, on_close=on_close)
+
+    def buttons_spec_get(self) -> dict:
+        # fmt: off
+        return dict(
+            #new     = (self.on_new_entity,      'FileIcon'),
+            edit    = (self.on_edit_entity,     'FileDialogDetailedView'),
+            reload  = (self.on_reload_entity,   'ArrowDown'),
+            save    = (self.on_save_entity,     'DriveNetIcon'),
+            delete  = (self.on_delete_entity,   'DialogDiscardButton')
+        )
+        # fmt: on