Fix type inference issues by adding type ignores in various files and refactor dictionary creation for clarity

Author: daizutabi

src/xlviews/core/index.py

@@ -62,7 +62,7 @@ def __len__(self) -> int:
 
     @property
     def names(self) -> list[str]:
-        return self.index.names
+        return self.index.names  # type: ignore
 
     @property
     def nlevels(self) -> int:

src/xlviews/dataframes/dist_frame.py

@@ -199,7 +199,7 @@ def get_init_data(index: Index, columns: list[str]) -> DataFrame:
 
 def get_dist_func(dist: str | dict[str, str], columns: list[str]) -> dict[str, str]:
     if isinstance(dist, str):
-        return {column: dist for column in columns}
+        return dict.fromkeys(columns, dist)
 
     dist = dist.copy()
     for column in columns:

src/xlviews/dataframes/heat_frame.py

@@ -42,7 +42,7 @@ def __init__(
 
         super().__init__(row, column, data, sheet)
 
-        self.columns = data.columns
+        self.columns = data.columns  # type: ignore
 
         start = self.row + 1, self.column + 1
         end = start[0] + self.shape[0] - 1, start[1] + self.shape[1] - 1

src/xlviews/dataframes/sheet_frame.py

@@ -612,8 +612,8 @@ def number_format(
                     continue
 
                 for pattern, number_format in columns_format.items():
-                    if re.match(pattern, column):
-                        rng = self.get_range(column).impl
+                    if re.match(pattern, column):  # type: ignore
+                        rng = self.get_range(column).impl  # type: ignore
                         rng.number_format = number_format
                         if autofit:
                             rng.autofit()

src/xlviews/dataframes/stats_frame.py

@@ -81,7 +81,7 @@ def get_func(func: str | list[str] | None) -> list[str]:
 
 def get_by(sf: SheetFrame, by: str | list[str] | None) -> list[str]:
     if not by:
-        return sf.index.names
+        return sf.index.names  # type: ignore
 
     return list(iter_columns(sf.index.names, by))
 
@@ -142,7 +142,7 @@ def set_style(sf: SheetFrame, parent: SheetFrame, func_column_name: str) -> None
     idx = [sf.column + i for i in range(sf.index.nlevels + len(sf.columns))]
 
     columns = (*parent.index.names, *parent.columns)
-    formats = [None, *[parent.get_number_format(column) for column in columns]]
+    formats = [None, *[parent.get_number_format(column) for column in columns]]  # type: ignore
 
     for (func,), rows in sf.groupby(func_column_name).items():
         for col, fmt in zip(idx, formats, strict=True):

src/xlviews/dataframes/table.py

@@ -118,7 +118,7 @@ def auto_filter(self, *args, clear: bool = False, **field_criteria) -> None:
            - other: specify the value.
         """
         if clear:
-            clear_filter = {x: None for x in self.columns}
+            clear_filter = dict.fromkeys(self.columns)
             clear_filter.update(field_criteria)
             field_criteria = clear_filter
 

src/xlviews/figure/grid.py

@@ -98,9 +98,6 @@ def __getitem__(self, key: int) -> Series: ...
     @overload
     def __getitem__(self, key: tuple[int, int]) -> Axes: ...
 
-    @overload
-    def __getitem__(self, key: tuple[int, int]) -> Axes: ...
-
     @overload
     def __getitem__(self, key: tuple[slice, int]) -> Series: ...
 

src/xlviews/figure/palette.py

@@ -186,8 +186,8 @@ def get_palette(
 
     if isinstance(style, Callable):
         if isinstance(data.index, MultiIndex):
-            return FunctionPalette(data.index.names, style)
-        return FunctionPalette(data.index.name, style)
+            return FunctionPalette(data.index.names, style)  # type: ignore
+        return FunctionPalette(data.index.name, style)  # type: ignore
 
     if data.index.name is not None or isinstance(data.index, MultiIndex):
         data = data.index.to_frame(index=False)

src/xlviews/figure/plot.py

@@ -58,7 +58,7 @@ def keys(self) -> Iterator[dict[str, Hashable]]:
         names = self.data.index.names
 
         for index in self.index:
-            yield dict(zip(names, index, strict=True))
+            yield dict(zip(names, index, strict=True))  # type: ignore
 
     def set(
         self,

src/xlviews/utils.py

@@ -128,6 +128,8 @@ def suspend_screen_updates(func: Callable[P, R]) -> Callable[P, R]:
 
     @wraps(func)
     def _func(*args: P.args, **kwargs: P.kwargs) -> R:
+        is_updating = False
+
         if app := xlwings.apps.active:
             is_updating = app.screen_updating
             app.screen_updating = False