TYP: io (#36120)

Author: jbrockmendel

pandas/io/excel/_util.py

@@ -1,3 +1,5 @@
+from typing import List
+
 from pandas.compat._optional import import_optional_dependency
 
 from pandas.core.dtypes.common import is_integer, is_list_like
@@ -56,7 +58,7 @@ def get_writer(engine_name):
         raise ValueError(f"No Excel writer '{engine_name}'") from err
 
 
-def _excel2num(x):
+def _excel2num(x: str) -> int:
     """
     Convert Excel column name like 'AB' to 0-based column index.
 
@@ -88,7 +90,7 @@ def _excel2num(x):
     return index - 1
 
 
-def _range2cols(areas):
+def _range2cols(areas: str) -> List[int]:
     """
     Convert comma separated list of column names and ranges to indices.
 
@@ -109,12 +111,12 @@ def _range2cols(areas):
     >>> _range2cols('A,C,Z:AB')
     [0, 2, 25, 26, 27]
     """
-    cols = []
+    cols: List[int] = []
 
     for rng in areas.split(","):
         if ":" in rng:
-            rng = rng.split(":")
-            cols.extend(range(_excel2num(rng[0]), _excel2num(rng[1]) + 1))
+            rngs = rng.split(":")
+            cols.extend(range(_excel2num(rngs[0]), _excel2num(rngs[1]) + 1))
         else:
             cols.append(_excel2num(rng))
 

pandas/io/formats/css.py

@@ -3,6 +3,7 @@
 """
 
 import re
+from typing import Optional
 import warnings
 
 
@@ -93,6 +94,7 @@ def __call__(self, declarations_str, inherited=None):
                 props[prop] = val
 
         # 2. resolve relative font size
+        font_size: Optional[float]
         if props.get("font-size"):
             if "font-size" in inherited:
                 em_pt = inherited["font-size"]
@@ -173,10 +175,11 @@ def _error():
             warnings.warn(f"Unhandled size: {repr(in_val)}", CSSWarning)
             return self.size_to_pt("1!!default", conversions=conversions)
 
-        try:
-            val, unit = re.match(r"^(\S*?)([a-zA-Z%!].*)", in_val).groups()
-        except AttributeError:
+        match = re.match(r"^(\S*?)([a-zA-Z%!].*)", in_val)
+        if match is None:
             return _error()
+
+        val, unit = match.groups()
         if val == "":
             # hack for 'large' etc.
             val = 1

setup.cfg

@@ -279,9 +279,6 @@ check_untyped_defs=False
 [mypy-pandas.io.formats.console]
 check_untyped_defs=False
 
-[mypy-pandas.io.formats.css]
-check_untyped_defs=False
-
 [mypy-pandas.io.formats.csvs]
 check_untyped_defs=False