From e02dd268f4022ef7a11dd2cb0c29eb5da1b945a9 Mon Sep 17 00:00:00 2001
From: MrTyton <MrTyton@users.noreply.github.com>
Date: Mon, 27 May 2024 23:20:43 -0400
Subject: [PATCH] Fix the logging disabler

---
 release-versions/latest.txt |  2 +-
 root/app/url_ingester.py    | 20 ++++++++------------
 2 files changed, 9 insertions(+), 13 deletions(-)

diff --git a/release-versions/latest.txt b/release-versions/latest.txt
index 49ac87b..b0e53de 100644
--- a/release-versions/latest.txt
+++ b/release-versions/latest.txt
@@ -1 +1 @@
-2024.05.27-12
\ No newline at end of file
+2024.05.27-13
\ No newline at end of file
diff --git a/root/app/url_ingester.py b/root/app/url_ingester.py
index a7bfaa6..98e29c8 100644
--- a/root/app/url_ingester.py
+++ b/root/app/url_ingester.py
@@ -18,16 +18,6 @@ def set_timeout(time):
         yield
     finally:
         socket.setdefaulttimeout(old_timeout)
-        
-@contextmanager
-def set_logging_level():
-    """Set the logging level to CRITICAL."""
-    old_level = logging.root.manager.disable
-    logging.disable(logging.CRITICAL)
-    try:
-        yield
-    finally:
-        logging.disable(old_level)
 
 class EmailInfo:
     """
@@ -68,14 +58,20 @@ def get_urls(self) -> set[str]:
         """
         urls = set()
                 # Save the current logging level
+        old_level = logging.root.manager.disable
 
+        # Set the logging level to CRITICAL
+        logging.disable(logging.CRITICAL)
         with set_timeout(55):
             try:
                 # Get URLs from the email account
-                with set_logging_level():
-                    urls = geturls.get_urls_from_imap(self.server, self.email, self.password, self.mailbox)
+                urls = geturls.get_urls_from_imap(self.server, self.email, self.password, self.mailbox)
             except Exception as e:
+                logging.disable(old_level)
                 ff_logging.log_failure(f"Failed to get URLs: {e}")
+            finally:
+                # Restore the old logging level
+                logging.disable(old_level)
         return urls