Typing fixes in core.py

Improve python compatability

camelot/core.py

@@ -5,15 +5,24 @@
 import math
 import os
 import sqlite3
+import sys
 import tempfile
 import zipfile
 from operator import itemgetter
+from typing import Any
 from typing import Iterable
 from typing import Iterator
 
 import cv2
 import pandas as pd
 
+
+if sys.version_info >= (3, 11):
+ from typing import TypedDict # pylint: disable=no-name-in-module
+ from typing import Unpack
+else:
+ from typing_extensions import TypedDict, Unpack
+
 from .backends import ImageConversionBackend
 from .utils import build_file_path_in_temp_dir
 from .utils import compute_whitespace
@@ -783,6 +792,13 @@ def to_sqlite(self, path, **kwargs):
  conn.close()
 
 
+class Kw(TypedDict):
+ path: os.PathLike[Any] | str
+ dirname: str
+ root: str
+ ext: str
+
+
 class TableList:
  """Defines a list of camelot.core.Table objects.
 
@@ -804,7 +820,7 @@ def __repr__(self): # noqa D105
  def __len__(self): # noqa D105
  return len(self._tables)
 
- def __getitem__(self, idx) -> Table: # noqa D105
+ def __getitem__(self, idx): # noqa D105
  return self._tables[idx]
 
  def __iter__(self) -> Iterator[Table]: # noqa D105
@@ -822,29 +838,29 @@ def n(self) -> int:
  """The number of tables in the list."""
  return len(self)
 
- def _write_file(self, f=None, **kwargs) -> None:
- dirname = kwargs.get("dirname")
- root = kwargs.get("root")
- ext = kwargs.get("ext")
+ def _write_file(self, f=None, **kwargs: Unpack[Kw]) -> None:
+ dirname = kwargs["dirname"]
+ root = kwargs["root"]
+ ext = kwargs["ext"]
  for table in self._tables:
  filename = f"{root}-page-{table.page}-table-{table.order}{ext}"
  filepath = os.path.join(dirname, filename)
  to_format = self._format_func(table, f)
  to_format(filepath)
 
- def _compress_dir(self, **kwargs) -> None:
- path = kwargs.get("path")
- dirname = kwargs.get("dirname")
- root = kwargs.get("root")
- ext = kwargs.get("ext")
+ def _compress_dir(self, **kwargs: Unpack[Kw]) -> None:
+ path = kwargs["path"]
+ dirname = kwargs["dirname"]
+ root = kwargs["root"]
+ ext = kwargs["ext"]
  zipname = os.path.join(os.path.dirname(path), root) + ".zip"
  with zipfile.ZipFile(zipname, "w", allowZip64=True) as z:
  for table in self._tables:
  filename = f"{root}-page-{table.page}-table-{table.order}{ext}"
  filepath = os.path.join(dirname, filename)
  z.write(filepath, os.path.basename(filepath))
 
- def export(self, path, f="csv", compress=False):
+ def export(self, path: str, f="csv", compress=False):
  """Export the list of tables to specified file format.
 
  Parameters
@@ -863,7 +879,7 @@ def export(self, path, f="csv", compress=False):
  if compress:
  dirname = tempfile.mkdtemp()
 
- kwargs = {"path": path, "dirname": dirname, "root": root, "ext": ext}
+ kwargs: Kw = {"path": path, "dirname": dirname, "root": root, "ext": ext}
 
  if f in ["csv", "html", "json", "markdown"]:
  self._write_file(f=f, **kwargs)