From fd43d33e3a142a3b8964260122a5e74e596efe6d Mon Sep 17 00:00:00 2001
From: Julien Palard <julien@palard.fr>
Date: Wed, 22 Jan 2020 17:57:12 +0100
Subject: [PATCH] Avoid starting curses when not needed (like for --help), it
 blinks my screen (#5)

---
 pogrep.py | 19 ++++---------------
 1 file changed, 4 insertions(+), 15 deletions(-)

diff --git a/pogrep.py b/pogrep.py
index 1acff26..85e42d7 100644
--- a/pogrep.py
+++ b/pogrep.py
@@ -13,6 +13,7 @@ from textwrap import fill
 import regex
 import polib
 from tabulate import tabulate
+from shutil import get_terminal_size
 
 
 def get_colors():
@@ -34,19 +35,6 @@ def get_colors():
 RED, GREEN, MAGENTA, NO_COLOR = get_colors()
 
 
-def get_term_width():
-    scr = curses.initscr()
-    width = scr.getmaxyx()[1]
-    try:
-        curses.endwin()
-    except curses.error:
-        pass
-    return width
-
-
-WIDTH = get_term_width()
-
-
 def colorize(text, pattern, prefixes):
     result = regex.sub(pattern, RED + r"\g<0>" + NO_COLOR, text)
     for pnum, pfile in prefixes:
@@ -62,6 +50,7 @@ def colorize(text, pattern, prefixes):
 def find_in_po(pattern, path, linenum, file_match, no_messages):
     table = []
     prefixes = []
+    term_width = get_terminal_size()[0]
     for filename in path:
         try:
             pofile = polib.pofile(filename)
@@ -85,8 +74,8 @@ def find_in_po(pattern, path, linenum, file_match, no_messages):
                     prefixes.append((pnum, pfile))
                 table.append(
                     [
-                        fill(left, width=(WIDTH - 7) // 2),
-                        fill(entry.msgstr, width=(WIDTH - 7) // 2),
+                        fill(left, width=(term_width - 7) // 2),
+                        fill(entry.msgstr, width=(term_width - 7) // 2),
                     ]
                 )
     if not file_match: