[Cherry Pick] Suppress analytics errors and messages (#319)

* Suppress analytics errors and messages

* Update src/sparsezoo/analytics.py

---------

Co-authored-by: Mark Kurtz <[email protected]>

Author: rahul-tuli

src/sparsezoo/analytics.py

@@ -24,6 +24,7 @@
 import requests
 
 from sparsezoo.utils.gdpr import is_gdpr_country
+from sparsezoo.utils.suppress import suppress_stdout_stderr
 from sparsezoo.version import version as sparsezoo_version
 
 
@@ -121,30 +122,31 @@ def send_event(
             event_params = {}
 
         def _send_request():
-            event_params.update(self._package_params)
-            event_params["package"] = self._package
-            event_params["version"] = self._version
-            payload = {
-                "client_id": self._client_id,
-                "events": [{"name": event_name, "params": event_params}],
-            }
-            headers = {
-                "Content-Type": "application/json",
-            }
-            data = json.dumps(payload)
-
-            try:
-                response = requests.post(self._url, headers=headers, data=data)
-                response.raise_for_status()
-                body = response.content
-                if _DEBUG:
-                    print(body)
-            except Exception as err:
-                if _DEBUG:
-                    print(err)
-
-                if raise_errors:
-                    raise err
+            with suppress_stdout_stderr(suppress=not _DEBUG):
+                event_params.update(self._package_params)
+                event_params["package"] = self._package
+                event_params["version"] = self._version
+                payload = {
+                    "client_id": self._client_id,
+                    "events": [{"name": event_name, "params": event_params}],
+                }
+                headers = {
+                    "Content-Type": "application/json",
+                }
+                data = json.dumps(payload)
+
+                try:
+                    response = requests.post(self._url, headers=headers, data=data)
+                    response.raise_for_status()
+                    body = response.content
+                    if _DEBUG:
+                        print(body)
+                except Exception as err:
+                    if _DEBUG:
+                        print(err)
+
+                    if raise_errors:
+                        raise err
 
         thread = threading.Thread(target=_send_request)
         thread.start()

src/sparsezoo/utils/gdpr.py

@@ -17,6 +17,8 @@
 import geocoder
 import requests
 
+from sparsezoo.utils.suppress import suppress_stdout_stderr
+
 
 __all__ = ["get_external_ip", "get_country_code", "is_gdpr_country"]
 
@@ -56,26 +58,28 @@ def get_external_ip() -> Optional[str]:
     """
     :return: the external ip of the machine, None if unable to get
     """
-    try:
-        response = requests.get("https://ident.me")
-        external_ip = response.text.strip()
+    with suppress_stdout_stderr():
+        try:
+            response = requests.get("https://ident.me")
+            external_ip = response.text.strip()
 
-        return external_ip
-    except Exception:
-        return None
+            return external_ip
+        except Exception:
+            return None
 
 
 def get_country_code() -> Optional[str]:
     """
     :return: the country code of the machine, None if unable to get
     """
-    try:
-        ip = get_external_ip()
-        geo = geocoder.ip(ip)
+    with suppress_stdout_stderr():
+        try:
+            ip = get_external_ip()
+            geo = geocoder.ip(ip)
 
-        return geo.country
-    except Exception:
-        return None
+            return geo.country
+        except Exception:
+            return None
 
 
 def is_gdpr_country() -> bool:

src/sparsezoo/utils/suppress.py

@@ -0,0 +1,45 @@
+# Copyright (c) 2021 - present / Neuralmagic, Inc. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import sys
+from contextlib import contextmanager
+
+
+__all__ = ["suppress_stdout_stderr", "NullDevice"]
+
+
+class NullDevice:
+    def write(self, s):
+        pass
+
+
+@contextmanager
+def suppress_stdout_stderr(suppress: bool = True):
+    """
+    Suppresses stdout and stderr for the duration of the context.
+    """
+    original_stdout = sys.stdout
+    original_stderr = sys.stderr
+    null_device = NullDevice()
+
+    try:
+        if suppress:
+            # Redirect stdout and stderr to the null device
+            sys.stdout = null_device
+            sys.stderr = null_device
+        yield
+    finally:
+        # Restore the original stdout and stderr
+        sys.stdout = original_stdout
+        sys.stderr = original_stderr