TYP: Return annotations for io/{formats,json}

Author: twoertwein

pandas/core/frame.py

@@ -1036,7 +1036,8 @@ def _repr_fits_horizontal_(self, ignore_width: bool = False) -> bool:
         value = buf.getvalue()
         repr_width = max(len(line) for line in value.split("\n"))
 
-        return repr_width < width
+        # error: Unsupported operand types for < ("int" and "None")
+        return repr_width < width  # type: ignore[operator]
 
     def _info_repr(self) -> bool:
         """

pandas/io/formats/console.py

@@ -1,11 +1,12 @@
 """
 Internal module for console introspection
 """
+from __future__ import annotations
 
 from shutil import get_terminal_size
 
 
-def get_console_size():
+def get_console_size() -> tuple[int | None, int | None]:
     """
     Return console size as tuple = (width, height).
 
@@ -43,14 +44,14 @@ def get_console_size():
     # Note if the User sets width/Height to None (auto-detection)
     # and we're in a script (non-inter), this will return (None,None)
     # caller needs to deal.
-    return (display_width or terminal_width, display_height or terminal_height)
+    return display_width or terminal_width, display_height or terminal_height
 
 
 # ----------------------------------------------------------------------
 # Detect our environment
 
 
-def in_interactive_session():
+def in_interactive_session() -> bool:
     """
     Check if we're running in an interactive shell.
 
@@ -75,7 +76,7 @@ def check_main():
         return check_main()
 
 
-def in_ipython_frontend():
+def in_ipython_frontend() -> bool:
     """
     Check if we're inside an IPython zmq frontend.
 

pandas/io/formats/css.py

@@ -7,6 +7,7 @@
 from typing import (
     Callable,
     Generator,
+    Iterator,
 )
 import warnings
 
@@ -369,7 +370,7 @@ def atomize(self, declarations) -> Generator[tuple[str, str], None, None]:
     expand_margin = _side_expander("margin-{:s}")
     expand_padding = _side_expander("padding-{:s}")
 
-    def parse(self, declarations_str: str):
+    def parse(self, declarations_str: str) -> Iterator[tuple[str, str]]:
         """
         Generates (prop, value) pairs from declarations.
 

pandas/io/formats/excel.py

@@ -197,7 +197,7 @@ def build_xlstyle(self, props: Mapping[str, str]) -> dict[str, dict[str, str]]:
 
         # TODO: handle cell width and height: needs support in pandas.io.excel
 
-        def remove_none(d: dict[str, str]) -> None:
+        def remove_none(d: dict[str, str | None]) -> None:
             """Remove key where value is None, through nested dicts"""
             for k, v in list(d.items()):
                 if v is None:
@@ -528,7 +528,7 @@ def __init__(
         self.inf_rep = inf_rep
 
     @property
-    def header_style(self):
+    def header_style(self) -> dict[str, dict[str, str | bool]]:
         return {
             "font": {"bold": True},
             "borders": {
@@ -850,7 +850,7 @@ def write(
         freeze_panes=None,
         engine=None,
         storage_options: StorageOptions = None,
-    ):
+    ) -> None:
         """
         writer : path-like, file-like, or ExcelWriter object
             File path or existing ExcelWriter

pandas/io/formats/format.py

@@ -22,10 +22,12 @@
     Callable,
     Hashable,
     Iterable,
+    Iterator,
     List,
     Mapping,
     Sequence,
     cast,
+    overload,
 )
 from unicodedata import east_asian_width
 
@@ -1203,12 +1205,15 @@ def save_to_buffer(
     with get_buffer(buf, encoding=encoding) as f:
         f.write(string)
         if buf is None:
-            return f.getvalue()
+            # error: "WriteBuffer[str]" has no attribute "getvalue"
+            return f.getvalue()  # type: ignore[attr-defined]
         return None
 
 
 @contextmanager
-def get_buffer(buf: FilePath | WriteBuffer[str] | None, encoding: str | None = None):
+def get_buffer(
+    buf: FilePath | WriteBuffer[str] | None, encoding: str | None = None
+) -> Iterator[WriteBuffer[str]] | Iterator[StringIO]:
     """
     Context manager to open, yield and close buffer for filenames or Path-like
     objects, otherwise yield buf unchanged.

pandas/io/formats/html.py

@@ -89,7 +89,7 @@ def render(self) -> list[str]:
         return self.elements
 
     @property
-    def should_show_dimensions(self):
+    def should_show_dimensions(self) -> bool:
         return self.fmt.should_show_dimensions
 
     @property

pandas/io/formats/info.py

@@ -566,7 +566,7 @@ def dtypes(self) -> Iterable[Dtype]:
         return [self.data.dtypes]
 
     @property
-    def dtype_counts(self):
+    def dtype_counts(self) -> Mapping[str, int]:
         from pandas.core.frame import DataFrame
 
         return _get_dataframe_dtype_counts(DataFrame(self.data))
@@ -1087,7 +1087,7 @@ def _fill_non_empty_info(self) -> None:
         if self.display_memory_usage:
             self.add_memory_usage_line()
 
-    def add_series_name_line(self):
+    def add_series_name_line(self) -> None:
         self._lines.append(f"Series name: {self.data.name}")
 
     @property

pandas/io/formats/style.py

@@ -12,6 +12,7 @@
     Callable,
     Hashable,
     Sequence,
+    overload,
 )
 import warnings
 
@@ -591,6 +592,52 @@ def to_excel(
             engine=engine,
         )
 
+    @overload
+    def to_latex(
+        self,
+        buf: FilePath | WriteBuffer[str],
+        *,
+        column_format: str | None = ...,
+        position: str | None = ...,
+        position_float: str | None = ...,
+        hrules: bool | None = ...,
+        clines: str | None = ...,
+        label: str | None = ...,
+        caption: str | tuple | None = ...,
+        sparse_index: bool | None = ...,
+        sparse_columns: bool | None = ...,
+        multirow_align: str | None = ...,
+        multicol_align: str | None = ...,
+        siunitx: bool = ...,
+        environment: str | None = ...,
+        encoding: str | None = ...,
+        convert_css: bool = ...,
+    ) -> None:
+        ...
+
+    @overload
+    def to_latex(
+        self,
+        buf: None = ...,
+        *,
+        column_format: str | None = ...,
+        position: str | None = ...,
+        position_float: str | None = ...,
+        hrules: bool | None = ...,
+        clines: str | None = ...,
+        label: str | None = ...,
+        caption: str | tuple | None = ...,
+        sparse_index: bool | None = ...,
+        sparse_columns: bool | None = ...,
+        multirow_align: str | None = ...,
+        multicol_align: str | None = ...,
+        siunitx: bool = ...,
+        environment: str | None = ...,
+        encoding: str | None = ...,
+        convert_css: bool = ...,
+    ) -> str:
+        ...
+
     def to_latex(
         self,
         buf: FilePath | WriteBuffer[str] | None = None,
@@ -610,7 +657,7 @@ def to_latex(
         environment: str | None = None,
         encoding: str | None = None,
         convert_css: bool = False,
-    ):
+    ) -> str | None:
         r"""
         Write Styler to a file, buffer or string in LaTeX format.
 
@@ -1161,6 +1208,46 @@ def to_latex(
         )
         return save_to_buffer(latex, buf=buf, encoding=encoding)
 
+    @overload
+    def to_html(
+        self,
+        buf: FilePath | WriteBuffer[str],
+        *,
+        table_uuid: str | None = ...,
+        table_attributes: str | None = ...,
+        sparse_index: bool | None = ...,
+        sparse_columns: bool | None = ...,
+        bold_headers: bool = ...,
+        caption: str | None = ...,
+        max_rows: int | None = ...,
+        max_columns: int | None = ...,
+        encoding: str | None = ...,
+        doctype_html: bool = ...,
+        exclude_styles: bool = ...,
+        **kwargs,
+    ) -> None:
+        ...
+
+    @overload
+    def to_html(
+        self,
+        buf: None = ...,
+        *,
+        table_uuid: str | None = ...,
+        table_attributes: str | None = ...,
+        sparse_index: bool | None = ...,
+        sparse_columns: bool | None = ...,
+        bold_headers: bool = ...,
+        caption: str | None = ...,
+        max_rows: int | None = ...,
+        max_columns: int | None = ...,
+        encoding: str | None = ...,
+        doctype_html: bool = ...,
+        exclude_styles: bool = ...,
+        **kwargs,
+    ) -> str:
+        ...
+
     @Substitution(buf=buf, encoding=encoding)
     def to_html(
         self,
@@ -1178,7 +1265,7 @@ def to_html(
         doctype_html: bool = False,
         exclude_styles: bool = False,
         **kwargs,
-    ):
+    ) -> str | None:
         """
         Write Styler to a file, buffer or string in HTML-CSS format.
 
@@ -1292,18 +1379,46 @@ def to_html(
             html, buf=buf, encoding=(encoding if buf is not None else None)
         )
 
+    @overload
+    def to_string(
+        self,
+        buf: FilePath | WriteBuffer[str],
+        *,
+        encoding=...,
+        sparse_index: bool | None = ...,
+        sparse_columns: bool | None = ...,
+        max_rows: int | None = ...,
+        max_columns: int | None = ...,
+        delimiter: str = ...,
+    ) -> None:
+        ...
+
+    @overload
+    def to_string(
+        self,
+        buf: None = ...,
+        *,
+        encoding=...,
+        sparse_index: bool | None = ...,
+        sparse_columns: bool | None = ...,
+        max_rows: int | None = ...,
+        max_columns: int | None = ...,
+        delimiter: str = ...,
+    ) -> str:
+        ...
+
     @Substitution(buf=buf, encoding=encoding)
     def to_string(
         self,
-        buf=None,
+        buf: FilePath | WriteBuffer[str] | None = None,
         *,
         encoding=None,
         sparse_index: bool | None = None,
         sparse_columns: bool | None = None,
         max_rows: int | None = None,
         max_columns: int | None = None,
         delimiter: str = " ",
-    ):
+    ) -> str | None:
         """
         Write Styler to a file, buffer or string in text format.