Fix DeprecationWarnings: use TableStyle instead of MARKDOWN and SINGL…

…E_BORDER (#456)
hugovk · Nov 2, 2024 · 1eaefde · 1eaefde
2 parents 62d581a + 76d132d
commit 1eaefde
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -38,7 +38,7 @@ dynamic = [ "version" ]
 dependencies = [
   "httpx>=0.19",
   "platformdirs",
-  "prettytable>=2.4",
+  "prettytable>=3.12",
   "pytablewriter[html]>=0.63",
   "python-dateutil",
   "python-slugify",

diff --git a/src/pypistats/__init__.py b/src/pypistats/__init__.py
@@ -373,10 +373,12 @@ def _tabulate(data: dict | list, format_: str = "markdown", color: str = "yes")
 def _prettytable(
     headers: list[str], data: dict | list, format_: str, color: str = "yes"
 ) -> str:
-    from prettytable import MARKDOWN, SINGLE_BORDER, PrettyTable
+    from prettytable import PrettyTable, TableStyle
 
     x = PrettyTable()
-    x.set_style(MARKDOWN if format_ == "markdown" else SINGLE_BORDER)
+    x.set_style(
+        TableStyle.MARKDOWN if format_ == "markdown" else TableStyle.SINGLE_BORDER
+    )
 
     if isinstance(data, dict):
         data = [data]