lamindb_setup 1.2.0__tar.gz → 1.3.0__tar.gz

{lamindb_setup-1.2.0 → lamindb_setup-1.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: lamindb_setup
-Version: 1.2.0
+Version: 1.3.0
 Summary: Setup & configure LaminDB.
 Author-email: Lamin Labs <open-source@lamin.ai>
 Requires-Python: >=3.10
@@ -40,7 +40,7 @@ Provides-Extra: gcp
 
 [![codecov](https://codecov.io/gh/laminlabs/lamindb-setup/branch/main/graph/badge.svg)](https://codecov.io/gh/laminlabs/lamindb-setup)
 
-# lamindb-setup: Setup LaminDB
+# lamindb-setup: Setting up `lamindb`
 
 - User [docs](https://lamin.ai/docs)
 - Developer [docs](https://lamindb-setup-htry.netlify.app/)

{lamindb_setup-1.2.0 → lamindb_setup-1.3.0}/README.md

@@ -1,6 +1,6 @@
 [![codecov](https://codecov.io/gh/laminlabs/lamindb-setup/branch/main/graph/badge.svg)](https://codecov.io/gh/laminlabs/lamindb-setup)
 
-# lamindb-setup: Setup LaminDB
+# lamindb-setup: Setting up `lamindb`
 
 - User [docs](https://lamin.ai/docs)
 - Developer [docs](https://lamindb-setup-htry.netlify.app/)

lamindb_setup-1.2.0/docs/hub-prod/test-connect-anonymously.ipynb → lamindb_setup-1.3.0/docs/hub-prod/test-connect-anonymously-set-token.ipynb

@@ -13,7 +13,10 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import lamindb_setup as ln_setup"
+    "import pytest\n",
+    "import lamindb_setup as ln_setup\n",
+    "from lamindb_setup.core.django import set_token\n",
+    "from psycopg2.errors import UndefinedFunction"
    ]
   },
   {
@@ -44,6 +47,20 @@
     "assert ln_setup.settings.storage._root is None"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# this just tests that the function is setup properly\n",
+    "# the actual functionality is tested in lamindb\n",
+    "set_token(None)\n",
+    "# set_token sql function is not defined\n",
+    "with pytest.raises(UndefinedFunction):\n",
+    "    set_token(\"test\")"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -79,7 +96,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.17"
+   "version": "3.10.16"
   }
  },
  "nbformat": 4,

{lamindb_setup-1.2.0 → lamindb_setup-1.3.0}/lamindb_setup/__init__.py

@@ -33,7 +33,7 @@ Modules & settings:
 
 """
 
-__version__ = "1.2.0"  # denote a release candidate for 0.1.0 with 0.1rc1
+__version__ = "1.3.0"  # denote a release candidate for 0.1.0 with 0.1rc1
 
 import os
 

{lamindb_setup-1.2.0 → lamindb_setup-1.3.0}/lamindb_setup/core/django.py

@@ -14,6 +14,31 @@ IS_MIGRATING = False
 CONN_MAX_AGE = 299
 
 
+def set_token(token: str | None, connection_name: str = "default"):
+    # None to reset
+    from django.db import connections
+    from django.db.backends.base.base import BaseDatabaseWrapper
+
+    connection = connections[connection_name]
+    assert connection.vendor == "postgresql"
+
+    if token is not None:
+
+        def connect(self):
+            BaseDatabaseWrapper.connect(self)
+            # now the connection should be set
+            # Use a psycopg cursor directly, bypassing Django's utilities.
+            with self.connection.cursor() as cursor:
+                cursor.execute("SELECT set_token(%s, false);", (token,))
+    else:
+        connect = BaseDatabaseWrapper.connect
+
+    connection.connect = connect.__get__(connection, connection.__class__)
+    # execute this to initially set or reset the token
+    connection.close()
+    connection.connect()
+
+
 def close_if_health_check_failed(self) -> None:
     if self.close_at is not None:
         if time.monotonic() >= self.close_at: