Fix deprecated click get_terminal_size usage (#71)

CERT-Polska · Apr 18, 2022 · ad08161 · ad08161
1 parent 04c2fba
commit ad08161
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/mwdblib/cli/formatters/tabular.py b/mwdblib/cli/formatters/tabular.py
@@ -1,4 +1,5 @@
 import json
+import shutil
 
 import beautifultable
 import click
@@ -17,7 +18,7 @@
 
 class TabularFormatter(ObjectFormatter):
     def format_attr_table(self, data):
-        term_width, term_height = click.get_terminal_size()
+        term_width, term_height = shutil.get_terminal_size()
         table = beautifultable.BeautifulTable(
             maxwidth=term_width, default_alignment=beautifultable.ALIGN_LEFT
         )
@@ -30,7 +31,7 @@ def format_attr_table(self, data):
         return str(table)
 
     def format_table(self, headers, widths, row_formatter, rows):
-        term_width, term_height = click.get_terminal_size()
+        term_width, term_height = shutil.get_terminal_size()
         table = beautifultable.BeautifulTable(
             maxwidth=term_width, default_alignment=beautifultable.ALIGN_LEFT
         )