fix: fix crashing when entering plain text file preview panel (#247)

hnicke · Aug 24, 2022 · 053ae44 · 053ae44
1 parent 8de2586
commit 053ae44
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/sodalite/ui/highlighting.py b/sodalite/ui/highlighting.py
@@ -44,19 +44,20 @@ def compute_highlighting(entry: Entry) -> Generator[HighlightedLine, None, None]
 
 
 def find_lexer(file: Path, content: str):
-    import pygments
+    from pygments.util import ClassNotFound
+    import pygments.lexers
     try:
         if file.name in ('.gitignore', '.dockerignore'):
             from pygments.lexers.shell import BashLexer
             lexer = BashLexer()
         else:
             lexer = pygments.lexers.guess_lexer_for_filename(str(file), content, stripnl=False, ensurenl=False)
         logger.debug('Detected lexer by filename')
-    except pygments.util.ClassNotFound:
+    except ClassNotFound:
         try:
             lexer = pygments.lexers.guess_lexer(content, stripnl=False, ensurenl=False)
             logger.debug('Detected lexer by file content')
-        except pygments.util.ClassNotFound:
+        except ClassNotFound:
             lexer = BashLexer(stripnl=False, ensurenl=False)
             logger.debug('Using fallback lexer')
     return lexer