Redefining print to flush always.

ministryofjustice · Nov 15, 2023 · ac895ed · ac895ed
1 parent f734933
commit ac895ed
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/app/app.py b/app/app.py
@@ -27,6 +27,9 @@
 from geventwebsocket.handler import WebSocketHandler
 from geventwebsocket import WebSocketError
 import websocket
+import functools
+
+print = functools.partial(print, flush=True) # redefine to flush the buffer always
 
 # Load secrets from a JSON file
 with open("secrets.json") as f:

diff --git a/app/azure_active_directory.py b/app/azure_active_directory.py
@@ -1,5 +1,8 @@
 import requests
 from flask import flash
+import functools
+
+print = functools.partial(print, flush=True) # redefine to flush the buffer always
 
 
 def create_aad_group(group_name, description, user_id, access_token, dry_run=False):