starbash 0.1.9__py3-none-any.whl → 0.1.11__py3-none-any.whl

starbash/commands/info.py

@@ -76,6 +76,75 @@ def filter():
         dump_column(sb, "Filter", Database.FILTER_KEY)
 
 
+kind_arg = typer.Argument(
+    help="Optional image type to filter by (e.g., BIAS, DARK, FLAT, LIGHT)",
+)
+
+
+@app.command()
+def master(
+    kind: Annotated[
+        str | None,
+        kind_arg,
+    ] = None,
+):
+    """List all precalculated master images (darks, biases, flats)."""
+    with Starbash("info.master") as sb:
+        # Get the master repo
+        images = sb.get_master_images(kind)
+
+        if not images:
+            kind_msg = f" of type '{kind}'" if kind else ""
+            console.print(f"[yellow]No master images{kind_msg} found.[/yellow]")
+            return
+
+        # Create table to display results
+        title = f"Master Images ({len(images)} total)"
+        if kind:
+            title = f"Master {kind} Images ({len(images)} total)"
+        table = Table(title=title)
+        table.add_column("Date", style=TABLE_COLUMN_STYLE, no_wrap=True)
+        table.add_column("Type", style=TABLE_COLUMN_STYLE, no_wrap=True)
+        table.add_column("Filename", style=TABLE_VALUE_STYLE, no_wrap=False)
+
+        # Sort by date, then by type
+        sorted_images = sorted(
+            images,
+            key=lambda img: (
+                img.get(Database.DATE_OBS_KEY) or img.get(Database.DATE_KEY) or "",
+                img.get(Database.IMAGETYP_KEY) or "",
+            ),
+        )
+
+        for image in sorted_images:
+            date = (
+                image.get(Database.DATE_OBS_KEY)
+                or image.get(Database.DATE_KEY)
+                or "Unknown"
+            )
+            # Extract just the date part (YYYY-MM-DD) if it's a full ISO timestamp
+            if "T" in date:
+                date = date.split("T")[0]
+
+            kind = image.get(Database.IMAGETYP_KEY) or "Unknown"
+            filename = image.get("path") or "Unknown"
+
+            table.add_row(date, kind, filename)
+
+        console.print(table)
+
+
+@app.command(hidden=True)
+def masters(
+    kind: Annotated[
+        str | None,
+        kind_arg,
+    ] = None,
+):
+    """Alias for 'info master' command."""
+    master(kind)
+
+
 @app.callback(invoke_without_command=True)
 def main_callback(ctx: typer.Context):
     """Show user preferences location and other app info.

starbash/commands/repo.py

@@ -7,6 +7,7 @@ import starbash
 from repo import repo_suffix, Repo
 from starbash.app import Starbash
 from starbash import console, log_filter_level
+from starbash.paths import get_user_documents_dir
 from starbash.toml import toml_from_template
 
 app = typer.Typer(invoke_without_command=True)
@@ -87,10 +88,15 @@ def main(
 
 @app.command()
 def add(
-    path: str,
+    path: Annotated[str | None, typer.Argument(help="Path to the respository")] = None,
     master: bool = typer.Option(
         False, "--master", help="Mark this new repository for master files."
     ),
+    processed: bool = typer.Option(
+        False,
+        "--processed",
+        help="Mark this new repository for processed output files.",
+    ),
 ):
     """
     Add a repository. path is either a local path or a remote URL.
@@ -98,6 +104,17 @@ def add(
     repo_type = None
     if master:
         repo_type = "master"
+    elif processed:
+        repo_type = "processed"
+
+    if path is None:
+        if repo_type is not None:
+            # if we know the repo type we can auto create a default path
+            path = str(get_user_documents_dir() / "repos" / repo_type)
+        else:
+            console.print("[red]Error: path is required for input repositories[/red]")
+            raise typer.Exit(1)
+
     with Starbash("repo.add") as sb:
         p = Path(path)
 
@@ -115,7 +132,7 @@ def add(
                     overrides={
                         "REPO_TYPE": repo_type,
                         "REPO_PATH": str(p),
-                        "DEFAULT_RELATIVE": "{instrument}/{date}/{imagetyp}/master_{imagetyp}.fits",
+                        "DEFAULT_RELATIVE": "{instrument}/{date}/{imagetyp}/master_{session_config}.fit",
                     },
                 )
             else:
@@ -134,7 +151,6 @@ def add(
             # we don't yet always write default config files at roots of repos, but it would be easy to add here
             # r.write_config()
             sb.user_repo.write_config()
-            # FIXME, we also need to index the newly added repo!!!
 
 
 def repo_url_to_repo(sb: Starbash, repo_url: str | None) -> Repo | None:

starbash/commands/select.py

@@ -71,7 +71,7 @@ def complete_column(incomplete: str, column_name: str):
         c = get_column(sb, column_name)
 
         for item, count in c.items():
-            if item.startswith(incomplete):
+            if item.lower().startswith(incomplete.lower()):
                 yield (item, f"{count} sessions")
 
 
@@ -253,6 +253,13 @@ def list_sessions(
                 telescope = get_key(Database.TELESCOP_KEY)
                 telescopes.add(telescope)
 
+                # Show the non normalized target name
+                metadata = sess.get("metadata")
+                if metadata:
+                    long_name = metadata.get("OBJECT")
+                    if long_name:
+                        object = long_name
+
                 # Format total exposure time as integer seconds
                 exptime_raw = get_key(Database.EXPTIME_TOTAL_KEY)
                 try: