Don't unconditionally import CryptographyDeprecationWarning.

Nextpertise · Apr 10, 2020 · 2c035b8 · 2c035b8
1 parent 9d5b829
commit 2c035b8
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 11 deletions.
diff --git a/src/lib/command.py b/src/lib/command.py
@@ -1,18 +1,24 @@
 import socket
 import subprocess
 import time
-import warnings
 
 from cryptography.utils import CryptographyDeprecationWarning
 import paramiko
 
 from .logger import logger
 
 
-warnings.filterwarnings(
-                         action='ignore',
-                         category=CryptographyDeprecationWarning,
-                       )
+try:
+    import warnings
+
+    from cryptography.utils import CryptographyDeprecationWarning
+
+    warnings.filterwarnings(
+                             action='ignore',
+                             category=CryptographyDeprecationWarning,
+                           )
+except Exception:
+    pass
 
 
 class command():

diff --git a/src/lib/rsync.py b/src/lib/rsync.py
@@ -1,19 +1,24 @@
 import socket
 import subprocess
 import time
-import warnings
 
-from cryptography.utils import CryptographyDeprecationWarning
 import paramiko
 
 from models.config import config
 from lib.logger import logger
 
 
-warnings.filterwarnings(
-                         action='ignore',
-                         category=CryptographyDeprecationWarning,
-                       )
+try:
+    import warnings
+
+    from cryptography.utils import CryptographyDeprecationWarning
+
+    warnings.filterwarnings(
+                             action='ignore',
+                             category=CryptographyDeprecationWarning,
+                           )
+except Exception:
+    pass
 
 
 class rsync():