conditionally import sasl vs puresasl

dbt/adapters/spark/connections.py

@@ -1,3 +1,4 @@
+import sys
 from contextlib import contextmanager
 
 import dbt.exceptions
@@ -31,8 +32,11 @@
     from thrift.transport.TSSLSocket import TSSLSocket
     import thrift
     import ssl
-    import sasl
     import thrift_sasl
+    if sys.version_info < (3, 11):
+        import sasl
+    else:
+        import puresasl as sasl
 except ImportError:
     pass  # done deliberately: setting modules to None explicitly violates MyPy contracts by degrading type semantics
 

dev-requirements.txt

@@ -31,3 +31,5 @@ wheel~=0.40
 # Adapter specific dependencies
 mock~=5.0
 thrift_sasl~=0.4.3
+sasl~=0.3.1;python_version<"3.11"
+puresasl;python_version>="3.11"