lamin_cli 1.5.1__py2.py3-none-any.whl → 1.5.3__py2.py3-none-any.whl

lamin_cli/__init__.py

@@ -1,3 +1,3 @@
 """Lamin CLI."""
 
-__version__ = "1.5.1"
+__version__ = "1.5.3"

lamin_cli/__main__.py

@@ -9,8 +9,9 @@ from collections import OrderedDict
 from functools import wraps
 from importlib.metadata import PackageNotFoundError, version
 from pathlib import Path
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Literal
 
+from lamin_utils import logger
 from lamindb_setup._init_instance import (
     DOC_DB,
     DOC_INSTANCE_NAME,
@@ -54,11 +55,11 @@ else:
             },
             {
                 "name": "Read & write data",
-                "commands": ["load", "save", "get", "delete"],
+                "commands": ["load", "save", "get", "delete", "create", "list"],
             },
             {
                 "name": "Configure",
-                "commands": ["cache", "settings", "migrate"],
+                "commands": ["checkout", "switch", "cache", "settings", "migrate"],
             },
             {
                 "name": "Auth",
@@ -209,6 +210,66 @@ def disconnect():
     return close_()
 
 
+# fmt: off
+@main.command()
+@click.argument("entity", type=str)
+@click.option("--name", type=str, default=None, help="A name.")
+# fmt: on
+def create(entity: Literal["branch"], name: str | None = None):
+    """Create a record for an entity.
+
+    Currently only supports creating a branch.
+
+    ```
+    lamin create branch --name my_branch
+    ```
+    """
+    assert entity == "branch", "Currently only supports creating a branch."
+
+    from lamindb.models import Branch
+
+    branch = Branch(name=name).save()
+    logger.important(f"created branch: {branch.name}")
+
+
+# fmt: off
+@main.command(name="list")
+@click.argument("entity", type=str)
+@click.option("--name", type=str, default=None, help="A name.")
+# fmt: on
+def list_(entity: Literal["branch"], name: str | None = None):
+    """List records for an entity.
+
+    ```
+    lamin list branch
+    lamin list space
+    ```
+    """
+    assert entity in {"branch", "space"}, "Currently only supports listing branches and spaces."
+
+    from lamindb.models import Branch, Space
+
+    if entity == "branch":
+        print(Branch.df())
+    else:
+        print(Space.df())
+
+
+# fmt: off
+@main.command()
+@click.option("--branch", type=str, default=None, help="A valid branch name or uid.")
+@click.option("--space", type=str, default=None, help="A valid branch name or uid.")
+# fmt: on
+def switch(branch: str | None = None, space: str | None = None):
+    """Switch between branches or spaces.
+
+    Currently only supports creating a branch.
+    """
+    from lamindb.setup import switch as switch_
+
+    switch_(branch=branch, space=space)
+
+
 @main.command()
 @click.option("--schema", is_flag=True, help="View database schema.")
 def info(schema: bool):
@@ -226,17 +287,30 @@ def info(schema: bool):
 
 # fmt: off
 @main.command()
-@click.argument("instance", type=str, default=None)
-@click.option("--force", is_flag=True, default=False, help="Do not ask for confirmation.")
+@click.argument("entity", type=str)
+@click.option("--name", type=str, default=None)
+@click.option("--slug", type=str, default=None)
+@click.option("--force", is_flag=True, default=False, help="Do not ask for confirmation (only relevant for instance).")
 # fmt: on
-def delete(instance: str, force: bool = False):
+def delete(entity: str, name: str | None = None, slug: str | None = None, force: bool = False):
     """Delete an entity.
 
-    Currently only supports instance deletion.
+    Currently supported: `branch` and `instance`.
+
+    ```
+    lamin delete instance --slug account/name
+    lamin delete branch --name my_branch
+    ```
     """
+    from lamindb import Branch
     from lamindb_setup._delete import delete
 
-    return delete(instance, force=force)
+    if entity == "branch":
+        Branch.get(name=name).delete()
+    elif entity == "instance":
+        return delete(slug, force=force)
+    else:  # backwars compatibility
+        return delete(entity, force=force)
 
 
 @main.command()

lamin_cli/_load.py

@@ -163,10 +163,11 @@ def load(
 
             # we don't use .get here because DoesNotExist is hard to catch
             # due to private django API
+            # we use `.objects` here because we don't want to exclude kind = __lamindb_run__ artifacts
             if query_by_uid:
-                entities = EntityClass.filter(uid__startswith=uid)
+                entities = EntityClass.objects.filter(uid__startswith=uid)
             else:
-                entities = EntityClass.filter(key=key)
+                entities = EntityClass.objects.filter(key=key)
 
             if (n_entities := len(entities)) == 0:
                 err_msg = f"uid={uid}" if query_by_uid else f"key={key}"

lamin_cli/_save.py

@@ -6,6 +6,7 @@ from typing import TYPE_CHECKING
 
 import click
 from lamin_utils import logger
+from lamindb_setup.core.hashing import hash_file
 
 if TYPE_CHECKING:
     from pathlib import Path
@@ -155,6 +156,10 @@ def save_from_path_cli(
         return None
 
     if registry == "transform":
+        if key is not None:
+            logger.warning(
+                "key is ignored for transforms, the transform key is determined by the filename"
+            )
         if is_cloud_path:
             logger.error("Can not register a transform from a cloud path")
             return "transform-with-cloud-path"
@@ -201,8 +206,19 @@ def save_from_path_cli(
                 if transform is None:
                     uid = f"{stem_uid}0000"
         else:
-            # TODO: account for folders and hash equivalence as we do in ln.track()
+            # TODO: account for folders as we do in ln.track()
+            transform_hash, _ = hash_file(path)
             transform = ln.Transform.filter(key=path.name, is_latest=True).one_or_none()
+            if transform is not None and transform.hash is not None:
+                if transform.hash == transform_hash:
+                    logger.important(
+                        f"found existing Transform('{uid}') with matching hash"
+                    )
+                    return None
+                else:
+                    # we need to create a new version
+                    stem_uid = transform.uid[:12]
+                    transform = None
         revises = None
         if stem_uid is not None:
             revises = (

{lamin_cli-1.5.1.dist-info → lamin_cli-1.5.3.dist-info}/METADATA

@@ -1,10 +1,9 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.3
 Name: lamin_cli
-Version: 1.5.1
+Version: 1.5.3
 Summary: Lamin CLI.
 Author-email: Lamin Labs <open-source@lamin.ai>
 Description-Content-Type: text/markdown
-License-File: LICENSE
 Requires-Dist: rich-click>=1.7
 Project-URL: Home, https://github.com/laminlabs/lamin-cli
 

lamin_cli-1.5.3.dist-info/RECORD

@@ -0,0 +1,14 @@
+lamin_cli/__init__.py,sha256=LBeGwsZuICY7sdEYZSugLgtHOU3Wh7heajX7B22wO3I,40
+lamin_cli/__main__.py,sha256=1hYODyzurDvq198STsIZ2slp_t_xq_c7QS0oyts33PA,15528
+lamin_cli/_cache.py,sha256=oplwE8AcS_9PYptQUZxff2qTIdNFS81clGPkJNWk098,800
+lamin_cli/_load.py,sha256=QqiFxGYmvFz2RjhvwKO5r1fmPWMZ5Ai4y1pJytdYKak,8301
+lamin_cli/_migration.py,sha256=xQi6mwnpBzY5wcv1-TJhveD7a3XJIlpiYx6Z3AJ1NF0,1063
+lamin_cli/_save.py,sha256=W4sNGXGFWy9ilFozVYbmxROCD8kkD0jDBB51cTLgZWA,10769
+lamin_cli/_settings.py,sha256=O2tecCf5EIZu98ima4DTJujo4KuywckOLgw8c-Ke3dY,1142
+lamin_cli/compute/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lamin_cli/compute/modal.py,sha256=QnR7GyyvWWWkLnou95HxS9xxSQfw1k-SiefM_qRVnU0,6010
+lamin_cli-1.5.3.dist-info/entry_points.txt,sha256=Qms85i9cZPlu-U7RnVZhFsF7vJ9gaLZUFkCjcGcXTpg,49
+lamin_cli-1.5.3.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+lamin_cli-1.5.3.dist-info/WHEEL,sha256=ssQ84EZ5gH1pCOujd3iW7HClo_O_aDaClUbX4B8bjKY,100
+lamin_cli-1.5.3.dist-info/METADATA,sha256=nKdmSG7KwhZG3c1l_G8FkROI9xFdy_zdwmg6leWo7_Q,337
+lamin_cli-1.5.3.dist-info/RECORD,,

{lamin_cli-1.5.1.dist-info → lamin_cli-1.5.3.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: flit 3.12.0
+Generator: flit 3.10.1
 Root-Is-Purelib: true
 Tag: py2-none-any
 Tag: py3-none-any

lamin_cli-1.5.1.dist-info/RECORD

@@ -1,14 +0,0 @@
-lamin_cli/__init__.py,sha256=LmyJRrP1UTrfQvwJPwpd_N8S8yv0IA87nvbNVtF82Iw,40
-lamin_cli/__main__.py,sha256=dHR3SNbuzg04ceVmTKdIPz2iXFCSdd2XsIb3OMmuvE0,13401
-lamin_cli/_cache.py,sha256=oplwE8AcS_9PYptQUZxff2qTIdNFS81clGPkJNWk098,800
-lamin_cli/_load.py,sha256=ow9IPQog_mPvjoGj1Bn08l4AG1YyRNnEyBpQvQkfLZs,8182
-lamin_cli/_migration.py,sha256=xQi6mwnpBzY5wcv1-TJhveD7a3XJIlpiYx6Z3AJ1NF0,1063
-lamin_cli/_save.py,sha256=wdTMmALkkNUpleUfXkNbh2Ila_pUIi5vhVbr30wLhEA,10069
-lamin_cli/_settings.py,sha256=O2tecCf5EIZu98ima4DTJujo4KuywckOLgw8c-Ke3dY,1142
-lamin_cli/compute/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lamin_cli/compute/modal.py,sha256=QnR7GyyvWWWkLnou95HxS9xxSQfw1k-SiefM_qRVnU0,6010
-lamin_cli-1.5.1.dist-info/entry_points.txt,sha256=Qms85i9cZPlu-U7RnVZhFsF7vJ9gaLZUFkCjcGcXTpg,49
-lamin_cli-1.5.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-lamin_cli-1.5.1.dist-info/WHEEL,sha256=Dyt6SBfaasWElUrURkknVFAZDHSTwxg3PaTza7RSbkY,100
-lamin_cli-1.5.1.dist-info/METADATA,sha256=Id-yCqUTghn94D3gvLc69Mhv2qxi1sPfCH3Oz4KPxsE,359
-lamin_cli-1.5.1.dist-info/RECORD,,