docs(transaction_control): use a different code for Python 3.10 to check other libraries' versions

Author: Thibault Bétrémieux

demos/transaction_control.ipynb

@@ -20,7 +20,6 @@
    "outputs": [],
    "source": [
     "import pandas as pd\n",
-    "from distutils.version import LooseVersion\n",
     "from pangres import upsert\n",
     "from sqlalchemy import create_engine, inspect as sqla_inspect, text\n",
     "from typing import Optional\n",
@@ -39,6 +38,37 @@
     "# Helpers"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "id": "32d537c1-3a4d-4e0d-a467-43be15c91773",
+   "metadata": {},
+   "source": [
+    "## Generic"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "id": "46fcfb29-6ef0-4a25-8af8-969ce9c0582c",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "def version_equal_or_greater_than(version_string, minimal_version_string) -> bool:\n",
+    "    # different implementations of this function depending on the python version\n",
+    "    # check if Python >= 3.10\n",
+    "    import sys\n",
+    "    py_v = sys.version_info\n",
+    "    py_gt3_10 = (py_v.major > 3) or ((py_v.major >= 3) and (py_v.minor >= 10))\n",
+    "\n",
+    "    # for Python >= 3.10 use packaging.version otherwise use distutils.version.LooseVersion\n",
+    "    if py_gt3_10:\n",
+    "        from packaging import version\n",
+    "        return version.parse(version_string) >= version.parse(minimal_version_string)\n",
+    "    else:\n",
+    "        from distutils.version import LooseVersion\n",
+    "        return LooseVersion(version_string) >= LooseVersion(minimal_version_string)"
+   ]
+  },
   {
    "cell_type": "markdown",
    "id": "832b8ab1-1356-4e7f-bd5a-cfdfe4087cbe",
@@ -49,15 +79,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": 3,
    "id": "normal-material",
    "metadata": {},
    "outputs": [],
    "source": [
     "def table_exists(connection, table_name, schema=None) -> bool:\n",
     "    insp = sqla_inspect(connection)\n",
     "    import sqlalchemy\n",
-    "    if LooseVersion(sqlalchemy.__version__) >= LooseVersion(\"1.4.0\"):\n",
+    "    if version_equal_or_greater_than(sqlalchemy.__version__, '1.4.0'):\n",
     "        return insp.has_table(table_name=table_name, schema=schema)\n",
     "    else:\n",
     "        return table_name in insp.get_table_names(schema=schema)\n",
@@ -97,7 +127,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 4,
    "id": "d8a3f3d2-fa54-41d4-a070-25c34b760db1",
    "metadata": {},
    "outputs": [],
@@ -149,7 +179,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": 5,
    "id": "strategic-sugar",
    "metadata": {},
    "outputs": [],
@@ -205,7 +235,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 6,
    "id": "seventh-reference",
    "metadata": {},
    "outputs": [
@@ -300,7 +330,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": 7,
    "id": "spiritual-spider",
    "metadata": {},
    "outputs": [
@@ -386,7 +416,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": 8,
    "id": "617a1ec1-b39c-45dd-903d-4520f6de71ec",
    "metadata": {},
    "outputs": [
@@ -484,7 +514,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": 9,
    "id": "204330eb-ce97-4fb4-8176-5fac3241ef0e",
    "metadata": {},
    "outputs": [
@@ -601,7 +631,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": 10,
    "id": "5e43fe2a-41f2-44b5-92ec-2d6828dbf08f",
    "metadata": {},
    "outputs": [],
@@ -611,7 +641,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": 11,
    "id": "55e4b5b6-af9c-4ba8-bc37-27ed9ec4a3c3",
    "metadata": {},
    "outputs": [
@@ -662,7 +692,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 11,
+   "execution_count": 12,
    "id": "88ecb1b5-7960-4d60-8e7f-3c5cd3774005",
    "metadata": {},
    "outputs": [],
@@ -672,7 +702,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 12,
+   "execution_count": 13,
    "id": "b0eeb2dd-7ab8-4f91-ae8b-91abe324a8f8",
    "metadata": {},
    "outputs": [
@@ -760,9 +790,9 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python [conda env:pangres-dev]",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
-   "name": "conda-env-pangres-dev-py"
+   "name": "python3"
   },
   "language_info": {
    "codemirror_mode": {
@@ -774,7 +804,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.7"
+   "version": "3.10.1"
   }
  },
  "nbformat": 4,