Revert "TYP/CLN: cleanup _openpyxl.py, add type annotation #36021 (#36022)"

This reverts commit ecc5015.

Author: Sarthak

ci/deps/azure-37-locale_slow.yaml

@@ -18,7 +18,7 @@ dependencies:
   - lxml
   - matplotlib=3.0.0
   - numpy=1.16.*
-  - openpyxl=2.6.0
+  - openpyxl=2.5.7
   - python-dateutil
   - python-blosc
   - pytz=2017.3

ci/deps/azure-37-minimum_versions.yaml

@@ -19,7 +19,7 @@ dependencies:
   - numba=0.46.0
   - numexpr=2.6.8
   - numpy=1.16.5
-  - openpyxl=2.6.0
+  - openpyxl=2.5.7
   - pytables=3.4.4
   - python-dateutil=2.7.3
   - pytz=2017.3

doc/source/getting_started/install.rst

@@ -274,7 +274,7 @@ html5lib                  1.0.1              HTML parser for read_html (see :ref
 lxml                      4.3.0              HTML parser for read_html (see :ref:`note <optional_html>`)
 matplotlib                2.2.3              Visualization
 numba                     0.46.0             Alternative execution engine for rolling operations
-openpyxl                  2.6.0              Reading / writing for xlsx files
+openpyxl                  2.5.7              Reading / writing for xlsx files
 pandas-gbq                0.12.0             Google Big Query access
 psycopg2                  2.7                PostgreSQL engine for sqlalchemy
 pyarrow                   0.15.0             Parquet, ORC, and feather reading / writing

doc/source/whatsnew/v1.2.0.rst

@@ -109,7 +109,7 @@ Optional libraries below the lowest tested version may still work, but are not c
 +-----------------+-----------------+---------+
 | numba           | 0.46.0          |         |
 +-----------------+-----------------+---------+
-| openpyxl        | 2.6.0           |    X    |
+| openpyxl        | 2.5.7           |         |
 +-----------------+-----------------+---------+
 | pyarrow         | 0.15.0          |    X    |
 +-----------------+-----------------+---------+

pandas/io/excel/_openpyxl.py

@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING, Dict, List, Optional
+from typing import List
 
 import numpy as np
 
@@ -8,9 +8,6 @@
 from pandas.io.excel._base import ExcelWriter, _BaseExcelReader
 from pandas.io.excel._util import _validate_freeze_panes
 
-if TYPE_CHECKING:
-    from openpyxl.descriptors.serialisable import Serialisable
-
 
 class _OpenpyxlWriter(ExcelWriter):
     engine = "openpyxl"
@@ -25,22 +22,53 @@ def __init__(self, path, engine=None, mode="w", **engine_kwargs):
         if self.mode == "a":  # Load from existing workbook
             from openpyxl import load_workbook
 
-            self.book = load_workbook(self.path)
+            book = load_workbook(self.path)
+            self.book = book
         else:
             # Create workbook object with default optimized_write=True.
             self.book = Workbook()
 
             if self.book.worksheets:
-                self.book.remove(self.book.worksheets[0])
+                try:
+                    self.book.remove(self.book.worksheets[0])
+                except AttributeError:
+
+                    # compat - for openpyxl <= 2.4
+                    self.book.remove_sheet(self.book.worksheets[0])
 
     def save(self):
         """
         Save workbook to disk.
         """
-        self.book.save(self.path)
+        return self.book.save(self.path)
+
+    @classmethod
+    def _convert_to_style(cls, style_dict):
+        """
+        Converts a style_dict to an openpyxl style object.
+
+        Parameters
+        ----------
+        style_dict : style dictionary to convert
+        """
+        from openpyxl.style import Style
+
+        xls_style = Style()
+        for key, value in style_dict.items():
+            for nk, nv in value.items():
+                if key == "borders":
+                    (
+                        xls_style.borders.__getattribute__(nk).__setattr__(
+                            "border_style", nv
+                        )
+                    )
+                else:
+                    xls_style.__getattribute__(key).__setattr__(nk, nv)
+
+        return xls_style
 
     @classmethod
-    def _convert_to_style_kwargs(cls, style_dict: dict) -> Dict[str, "Serialisable"]:
+    def _convert_to_style_kwargs(cls, style_dict):
         """
         Convert a style_dict to a set of kwargs suitable for initializing
         or updating-on-copy an openpyxl v2 style object.
@@ -65,7 +93,7 @@ def _convert_to_style_kwargs(cls, style_dict: dict) -> Dict[str, "Serialisable"]
         """
         _style_key_map = {"borders": "border"}
 
-        style_kwargs: Dict[str, Serialisable] = {}
+        style_kwargs = {}
         for k, v in style_dict.items():
             if k in _style_key_map:
                 k = _style_key_map[k]
@@ -376,7 +404,7 @@ def write_cells(
         # Write the frame cells using openpyxl.
         sheet_name = self._get_sheet_name(sheet_name)
 
-        _style_cache: Dict[str, Dict[str, Serialisable]] = {}
+        _style_cache = {}
 
         if sheet_name in self.sheets:
             wks = self.sheets[sheet_name]
@@ -398,7 +426,7 @@ def write_cells(
             if fmt:
                 xcell.number_format = fmt
 
-            style_kwargs: Optional[Dict[str, Serialisable]] = {}
+            style_kwargs = {}
             if cell.style:
                 key = str(cell.style)
                 style_kwargs = _style_cache.get(key)
@@ -487,17 +515,16 @@ def get_sheet_by_index(self, index: int):
 
     def _convert_cell(self, cell, convert_float: bool) -> Scalar:
 
-        from openpyxl.cell.cell import TYPE_BOOL, TYPE_ERROR, TYPE_NUMERIC
-
+        # TODO: replace with openpyxl constants
         if cell.is_date:
             return cell.value
-        elif cell.data_type == TYPE_ERROR:
+        elif cell.data_type == "e":
             return np.nan
-        elif cell.data_type == TYPE_BOOL:
+        elif cell.data_type == "b":
             return bool(cell.value)
         elif cell.value is None:
             return ""  # compat with xlrd
-        elif cell.data_type == TYPE_NUMERIC:
+        elif cell.data_type == "n":
             # GH5394
             if convert_float:
                 val = int(cell.value)

setup.cfg

@@ -217,6 +217,9 @@ check_untyped_defs=False
 [mypy-pandas.io.excel._base]
 check_untyped_defs=False
 
+[mypy-pandas.io.excel._openpyxl]
+check_untyped_defs=False
+
 [mypy-pandas.io.excel._util]
 check_untyped_defs=False