Fix unsupported wurlitzer on Windows

kinuax · May 7, 2024 · ad884c7 · ad884c7
1 parent 4187dcb
commit ad884c7
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/rolabesti/views/utils.py b/rolabesti/views/utils.py
@@ -1,9 +1,17 @@
-import contextlib
+from contextlib import contextmanager, redirect_stdout
 from pathlib import Path
-from wurlitzer import pipes
+
+try:
+    from wurlitzer import pipes
+except ModuleNotFoundError:
+    # wurlitzer is not supported on Windows.
+    @contextmanager
+    def pipes():
+        yield
+
 
 # Avoid messages from pygame while importing.
-with contextlib.redirect_stdout(None):
+with redirect_stdout(None):
     import pygame