lamin_cli 1.5.0__py2.py3-none-any.whl → 1.5.2__py2.py3-none-any.whl

lamin_cli/__init__.py

@@ -1,3 +1,3 @@
 """Lamin CLI."""
 
-__version__ = "1.5.0"
+__version__ = "1.5.2"

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-1.5.0.dist-info → lamin_cli-1.5.2.dist-info}/METADATA

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

{lamin_cli-1.5.0.dist-info → lamin_cli-1.5.2.dist-info}/RECORD

@@ -1,14 +1,14 @@
-lamin_cli/__init__.py,sha256=zkMFZ7WjniomHHD8qNZEp3Qd4EepX7ahcE_jBnqWNuw,40
-lamin_cli/__main__.py,sha256=dHR3SNbuzg04ceVmTKdIPz2iXFCSdd2XsIb3OMmuvE0,13401
+lamin_cli/__init__.py,sha256=8S1h5xBXifK8y1USGZGBHskI7LFAjrYeMRmTQMXgAqU,40
+lamin_cli/__main__.py,sha256=1hYODyzurDvq198STsIZ2slp_t_xq_c7QS0oyts33PA,15528
 lamin_cli/_cache.py,sha256=oplwE8AcS_9PYptQUZxff2qTIdNFS81clGPkJNWk098,800
-lamin_cli/_load.py,sha256=ow9IPQog_mPvjoGj1Bn08l4AG1YyRNnEyBpQvQkfLZs,8182
+lamin_cli/_load.py,sha256=QqiFxGYmvFz2RjhvwKO5r1fmPWMZ5Ai4y1pJytdYKak,8301
 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.0.dist-info/entry_points.txt,sha256=Qms85i9cZPlu-U7RnVZhFsF7vJ9gaLZUFkCjcGcXTpg,49
-lamin_cli-1.5.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-lamin_cli-1.5.0.dist-info/WHEEL,sha256=ssQ84EZ5gH1pCOujd3iW7HClo_O_aDaClUbX4B8bjKY,100
-lamin_cli-1.5.0.dist-info/METADATA,sha256=iHeeye4BQLhOp0HCcqsogIfQbt95x5Dr5vEW3-t7ArM,337
-lamin_cli-1.5.0.dist-info/RECORD,,
+lamin_cli-1.5.2.dist-info/entry_points.txt,sha256=Qms85i9cZPlu-U7RnVZhFsF7vJ9gaLZUFkCjcGcXTpg,49
+lamin_cli-1.5.2.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+lamin_cli-1.5.2.dist-info/WHEEL,sha256=ssQ84EZ5gH1pCOujd3iW7HClo_O_aDaClUbX4B8bjKY,100
+lamin_cli-1.5.2.dist-info/METADATA,sha256=etOomDY2o5Hw6h41jnObSNjubozaPdLn8d8dndiz2AE,337
+lamin_cli-1.5.2.dist-info/RECORD,,