tksheet 7.0.6__py3-none-any.whl → 7.1.0__py3-none-any.whl

tksheet/other_classes.py

@@ -4,14 +4,11 @@ import pickle
 from collections import namedtuple
 from collections.abc import Callable, Generator, Hashable, Iterator
 from functools import partial
+from typing import Literal
 
 pickle_obj = partial(pickle.dumps, protocol=pickle.HIGHEST_PROTOCOL)
 
 FontTuple = namedtuple("FontTuple", "family size style")
-CurrentlySelectedClass = namedtuple(
-    "CurrentlySelectedClass",
-    "row column type_ tags",
-)
 Box_nt = namedtuple(
     "Box_nt",
     "from_r from_c upto_r upto_c",
@@ -219,6 +216,21 @@ class Span(dict):
             self["widget"].clear(self, redraw=redraw)
         return self
 
+    def tag(self, *tags) -> Span:
+        self["widget"].tag(self, tags=tags)
+        return self
+
+    def untag(self) -> Span:
+        if self.kind == "cell":
+            for r in self.rows:
+                for c in self.columns:
+                    self["widget"].untag(cell=(r, c))
+        elif self.kind == "row":
+            self["widget"].untag(rows=self.rows)
+        elif self.kind == "column":
+            self["widget"].untag(columns=self.columns)
+        return self
+
     def options(
         self,
         type_: str | None = None,
@@ -344,3 +356,116 @@ class GeneratedMouseEvent:
     def __init__(self):
         self.keycode = "??"
         self.num = 1
+
+
+class Node:
+    __slots__ = ("text", "iid", "parent", "children")
+
+    def __init__(
+        self,
+        text: str,
+        iid: str,
+        parent: Node | Literal[""] | None = None,
+    ) -> None:
+        self.text = text
+        self.iid = iid
+        self.parent = parent
+        self.children = []
+
+    def __str__(self) -> str:
+        return self.text
+
+
+class DropdownStorage:
+    __slots__ = ("canvas_id", "window", "open")
+
+    def __init__(self) -> None:
+        self.canvas_id = None
+        self.window = None
+        self.open = False
+
+    def get_coords(self) -> int | tuple[int, int] | None:
+        """
+        Returns None if not open or window is None
+        """
+        if self.open and self.window is not None:
+            return self.window.get_coords()
+        return None
+
+
+class TextEditorStorage:
+    __slots__ = ("canvas_id", "window", "open")
+
+    def __init__(self) -> None:
+        self.canvas_id = None
+        self.window = None
+        self.open = False
+
+    def focus(self) -> None:
+        if self.window:
+            self.window.tktext.focus_set()
+
+    def get(self) -> str:
+        if self.window:
+            return self.window.get()
+        return ""
+
+    @property
+    def tktext(self) -> object:
+        if self.window:
+            return self.window.tktext
+        return self.window
+
+    @property
+    def coords(self) -> tuple[int, int]:
+        return self.window.r, self.window.c
+
+    @property
+    def row(self) -> int:
+        return self.window.r
+
+    @property
+    def column(self) -> int:
+        return self.window.c
+
+
+class SelectionBox:
+    __slots__ = ("fill_iid", "bd_iid", "index", "header", "coords", "type_")
+
+    def __init__(
+        self,
+        fill_iid: int | None = None,
+        bd_iid: int | None = None,
+        index: int | None = None,
+        header: int | None = None,
+        coords: tuple[int, int, int, int] = None,
+        type_: Literal["cells", "rows", "columns"] = "cells",
+    ) -> None:
+        self.fill_iid = fill_iid
+        self.bd_iid = bd_iid
+        self.index = index
+        self.header = header
+        self.coords = coords
+        self.type_ = type_
+
+
+Selected = namedtuple(
+    "Selected",
+    (
+        "row",
+        "column",
+        "type_",
+        "box",
+        "iid",
+        "fill_iid",
+    ),
+    defaults=(
+        None,
+        None,
+        None,
+        None,
+        None,
+        None,
+    ),
+)
+