lamin_cli 0.19.0__py2.py3-none-any.whl → 0.20.0__py2.py3-none-any.whl

lamin_cli/__init__.py

@@ -1,3 +1,3 @@
 """Lamin CLI."""
 
-__version__ = "0.19.0"
+__version__ = "0.20.0"

lamin_cli/__main__.py

@@ -98,13 +98,13 @@ def main():
 
 @main.command()
 @click.argument("user", type=str, default=None, required=False)
-@click.option("--key", type=str, default=None, help="The API key.")
+@click.option("--key", type=str, default=None, help="The legacy API key.")
 def login(user: str, key: Optional[str]):
     """Log into LaminHub.
 
     `lamin login` prompts for your API key unless you set it via environment variable `LAMIN_API_KEY`.
 
-    You find your API key in your account settings on LaminHub (top right corner).
+    You can create your API key in your account settings on LaminHub (top right corner).
 
     After authenticating once, you can re-authenticate and switch between accounts via `lamin login myhandle`.
     """
@@ -118,6 +118,12 @@ def login(user: str, key: Optional[str]):
     else:
         api_key = None
 
+    if key is not None:
+        click.echo(
+            "--key is deprecated and will be removed in the future, "
+            "use `lamin login` and enter your API key."
+        )
+
     return login_(user, key=key, api_key=api_key)
 
 

lamin_cli/_load.py

@@ -46,16 +46,32 @@ def load(entity: str, uid: str = None, key: str = None, with_env: bool = False):
         notebook = jupytext.reads(py_content, fmt="py:percent")
         jupytext.write(notebook, notebook_path)
 
+    query_by_uid = uid is not None
+
     if entity == "transform":
-        transform = (
-            ln.Transform.objects.get(uid=uid)
-            if uid is not None
+        if query_by_uid:
+            # we don't use .get here because DoesNotExist is hard to catch
+            # due to private django API
+            # here full uid is not expected anymore as before
+            # via ln.Transform.objects.get(uid=uid)
+            transforms = ln.Transform.objects.filter(uid__startswith=uid)
+        else:
             # if below, we take is_latest=True as the criterion, we might get draft notebooks
             # hence, we use source_code__isnull=False and order by created_at instead
-            else ln.Transform.objects.filter(key=key, source_code__isnull=False)
-            .order_by("-created_at")
-            .first()
-        )
+            transforms = ln.Transform.objects.filter(key=key, source_code__isnull=False)
+
+        if (n_transforms := len(transforms)) == 0:
+            err_msg = (
+                f"uid strating with {uid}"
+                if query_by_uid
+                else f"key={key} and source_code"
+            )
+            raise SystemExit(f"Transform with {err_msg} does not exist.")
+
+        if n_transforms > 1:
+            transforms = transforms.order_by("-created_at")
+        transform = transforms.first()
+
         target_filename = transform.key
         if Path(target_filename).exists():
             response = input(f"! {target_filename} exists: replace? (y/n)")
@@ -99,13 +115,22 @@ def load(entity: str, uid: str = None, key: str = None, with_env: bool = False):
         import lamindb as ln
 
         ln.settings.track_run_inputs = False
-        artifact = (
-            ln.Artifact.get(uid)
-            if uid is not None
-            else ln.Artifact.filter(key=key, source_code__isnull=False)
-            .order_by("-created_at")
-            .first()
-        )
+
+        if query_by_uid:
+            # we don't use .get here because DoesNotExist is hard to catch
+            # due to private django API
+            artifacts = ln.Artifact.filter(uid__startswith=uid)
+        else:
+            artifacts = ln.Artifact.filter(key=key)
+
+        if (n_artifacts := len(artifacts)) == 0:
+            err_msg = f"uid strating with {uid}" if query_by_uid else f"key={key}"
+            raise SystemExit(f"Artifact with {err_msg} does not exist.")
+
+        if n_artifacts > 1:
+            artifacts = artifacts.order_by("-created_at")
+        artifact = artifacts.first()
+
         cache_path = artifact.cache()
         logger.important(f"artifact is here: {cache_path}")
         return cache_path

{lamin_cli-0.19.0.dist-info → lamin_cli-0.20.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lamin_cli
-Version: 0.19.0
+Version: 0.20.0
 Summary: Lamin CLI.
 Author-email: Lamin Labs <open-source@lamin.ai>
 Description-Content-Type: text/markdown

lamin_cli-0.20.0.dist-info/RECORD

@@ -0,0 +1,12 @@
+lamin_cli/__init__.py,sha256=LLbtdktouioVHbc3LHIcHFkIqK6tGoeT0wiW37J2Rjk,41
+lamin_cli/__main__.py,sha256=-QAfgrYKcPjl0qkiWr8Tfm5wRaaZ99rW2atjIBnBn9Y,9900
+lamin_cli/_cache.py,sha256=kW8rqlMwQeOngm9uq2gjzPVl3EBrwh6W2F2AvyBFABY,799
+lamin_cli/_load.py,sha256=n0_qHTCXsZQIUy87rVdPvT89fx60N-KZgn_dL8nsvEA,5523
+lamin_cli/_migration.py,sha256=xTbad6aDUwuK0QvWCmDW3f8-xaqwisS-Cqf-LbD1WRI,1071
+lamin_cli/_save.py,sha256=KpGB1ZeKyE0Of0PkCX8j2UPtj4FQPS5wwKIYUFkTeio,4962
+lamin_cli/_settings.py,sha256=iS37mcQUHKRWxi2sHnAojEI6sWk3w232qwG-GeY2_Qc,1141
+lamin_cli-0.20.0.dist-info/entry_points.txt,sha256=Qms85i9cZPlu-U7RnVZhFsF7vJ9gaLZUFkCjcGcXTpg,49
+lamin_cli-0.20.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+lamin_cli-0.20.0.dist-info/WHEEL,sha256=Sgu64hAMa6g5FdzHxXv9Xdse9yxpGGMeagVtPMWpJQY,99
+lamin_cli-0.20.0.dist-info/METADATA,sha256=BmVqxVC-ZK09XH83pqa_Mr7MBnu56gqxGzh1SEnWKw8,375
+lamin_cli-0.20.0.dist-info/RECORD,,

lamin_cli-0.19.0.dist-info/RECORD

@@ -1,12 +0,0 @@
-lamin_cli/__init__.py,sha256=lIL1Y076MI8xywLMtnXwDm2sbJ543xy8YeHZu83ZJWU,41
-lamin_cli/__main__.py,sha256=-auo60kaPHZB-BuQbzGJNDl9ne3OZ4T_GhFweOAr-74,9706
-lamin_cli/_cache.py,sha256=kW8rqlMwQeOngm9uq2gjzPVl3EBrwh6W2F2AvyBFABY,799
-lamin_cli/_load.py,sha256=2TkNQviRrvsp5wFweMqD5je0t1-BQYiWrBN7ygyxmuQ,4508
-lamin_cli/_migration.py,sha256=xTbad6aDUwuK0QvWCmDW3f8-xaqwisS-Cqf-LbD1WRI,1071
-lamin_cli/_save.py,sha256=KpGB1ZeKyE0Of0PkCX8j2UPtj4FQPS5wwKIYUFkTeio,4962
-lamin_cli/_settings.py,sha256=iS37mcQUHKRWxi2sHnAojEI6sWk3w232qwG-GeY2_Qc,1141
-lamin_cli-0.19.0.dist-info/entry_points.txt,sha256=Qms85i9cZPlu-U7RnVZhFsF7vJ9gaLZUFkCjcGcXTpg,49
-lamin_cli-0.19.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-lamin_cli-0.19.0.dist-info/WHEEL,sha256=Sgu64hAMa6g5FdzHxXv9Xdse9yxpGGMeagVtPMWpJQY,99
-lamin_cli-0.19.0.dist-info/METADATA,sha256=BlyOjcxJ2_nbB-0hjMkeIPFEkRKTZUy08JP1LE5-QBE,375
-lamin_cli-0.19.0.dist-info/RECORD,,