SNOW-848033: backend prerelease version compatibility (#1641)

Author: sfc-gh-aling

setup.cfg

@@ -9,7 +9,7 @@ author_email = [email protected]
 license = Apache-2.0
 license_files = LICENSE.txt, NOTICE
 classifiers =
-    Development Status :: 5 - Production/Stable
+    Development Status :: 4 - Beta
     Environment :: Console
     Environment :: Other Environment
     Intended Audience :: Developers

src/snowflake/connector/connection.py

@@ -98,6 +98,7 @@
 from .telemetry_oob import TelemetryService
 from .time_util import HeartBeatTimer, get_time_millis
 from .util_text import construct_hostname, parse_account, split_statements
+from .version import VERSION
 
 DEFAULT_CLIENT_PREFETCH_THREADS = 4
 MAX_CLIENT_PREFETCH_THREADS = 10
@@ -350,9 +351,19 @@ def __init__(
         self.__set_error_attributes()
         self.connect(**kwargs)
         self._telemetry = TelemetryClient(self._rest)
-
         # get the imported modules from sys.modules
         self._log_telemetry_imported_packages()
+        # in the future we won't need this if the backend supports prerelease versions
+        self._log_telemetry(
+            TelemetryData.from_telemetry_data_dict(
+                from_dict={
+                    TelemetryField.KEY_IS_PRERELEASE.value: TelemetryData.FALSE
+                    if str(VERSION[2]).isdigit()
+                    else TelemetryData.TRUE
+                },
+                timestamp=get_time_millis(),
+            )
+        )
 
     def __del__(self) -> None:  # pragma: no cover
         try:

src/snowflake/connector/description.py

@@ -8,6 +8,7 @@
 from __future__ import annotations
 
 import platform
+import re
 import sys
 
 from .version import VERSION
@@ -20,4 +21,10 @@
 COMPILER = platform.python_compiler()
 
 CLIENT_NAME = "PythonConnector"  # don't change!
-CLIENT_VERSION = ".".join([str(v) for v in VERSION[:3]])
+
+# This is a short-term workaround for the backend to enable client side features for preview version, e.g. 3.1.0a1
+CLIENT_VERSION = (
+    ".".join([str(v) for v in VERSION[:3]])
+    if str(VERSION[2]).isdigit()
+    else f"{str(VERSION[0])}.{str(VERSION[1])}.{re.split('[ab]', str(VERSION[2]))[0]}"
+)

src/snowflake/connector/telemetry.py

@@ -50,6 +50,7 @@ class TelemetryField(Enum):
     KEY_SQLSTATE = "sql_state"
     KEY_DRIVER_TYPE = "driver_type"
     KEY_DRIVER_VERSION = "driver_version"
+    KEY_IS_PRERELEASE = "is_prerelease"
     KEY_REASON = "reason"
     KEY_VALUE = "value"
     KEY_EXCEPTION = "exception"