feat: limit divider width to terminal size (#41)

* limit divider width to terminal size * format api.py
jkerola · Jan 29, 2024 · 52496c1 · 52496c1
1 parent 1f14ab2
commit 52496c1
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/src/jmenu/api.py b/src/jmenu/api.py
@@ -10,7 +10,6 @@
     * API_URL
 """
 
-
 import requests
 from datetime import datetime
 from .classes import Restaurant, MenuItem, SKIPPED_ITEMS

diff --git a/src/jmenu/main.py b/src/jmenu/main.py
@@ -12,6 +12,7 @@
 import argparse
 import time
 import sys
+from os import get_terminal_size
 
 from importlib.metadata import version, PackageNotFoundError
 
@@ -146,9 +147,15 @@ def _print_highlight(items: list[MenuItem], allergens: list[str]):
 
 
 def _print_header(fetch_date: datetime):
-    print("-" * 79)
+    try:
+        width = get_terminal_size()[0]
+        if width is None or width > 79:
+            width = 79
+    except OSError:
+        width = 79
+    print("-" * width)
     print("Menu for", fetch_date.strftime("%d.%m"))
-    print("-" * 79)
+    print("-" * width)
 
 
 def get_version() -> str: