starbash 0.1.4__tar.gz → 0.1.5__tar.gz

{starbash-0.1.4 → starbash-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: starbash
-Version: 0.1.4
+Version: 0.1.5
 Summary: A tool for automating/standardizing/sharing astrophotography workflows.
 License-File: LICENSE
 Author: Kevin Hester

{starbash-0.1.4 → starbash-0.1.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "starbash"
-version = "0.1.4"
+version = "0.1.5"
 description = "A tool for automating/standardizing/sharing astrophotography workflows."
 authors = ["Kevin Hester <kevinh@geeksville.com>"]
 readme = "README.md"

{starbash-0.1.4 → starbash-0.1.5}/src/starbash/analytics.py

@@ -60,7 +60,7 @@ def is_development_environment() -> bool:
     """Detect if running in a development environment."""
 
     # Check for explicit environment variable
-    if os.getenv("STARBASH_ENV") == "development":
+    if os.getenv("SENTRY_ENVIRONMENT") == "development":
         return True
 
     # Check if running under VS Code

starbash-0.1.5/src/starbash/commands/__init__.py

@@ -0,0 +1,15 @@
+"""Shared utilities for starbash commands."""
+
+
+def format_duration(seconds: int | float) -> str:
+    """Format seconds as a human-readable duration string."""
+    if seconds < 60:
+        return f"{int(seconds)}s"
+    elif seconds < 120:
+        minutes = int(seconds // 60)
+        secs = int(seconds % 60)
+        return f"{minutes}m {secs}s" if secs else f"{minutes}m"
+    else:
+        hours = int(seconds // 3600)
+        minutes = int((seconds % 3600) // 60)
+        return f"{hours}h {minutes}m" if minutes else f"{hours}h"

starbash-0.1.5/src/starbash/commands/info.py

@@ -0,0 +1,92 @@
+"""Info commands for displaying system and data information."""
+
+import typer
+from typing_extensions import Annotated
+
+from starbash.app import Starbash
+from starbash import console
+from starbash.database import Database
+from starbash.paths import get_user_config_dir, get_user_data_dir
+from starbash.commands import format_duration
+
+app = typer.Typer()
+
+
+@app.command()
+def target():
+    """List targets (filtered based on the current selection)."""
+    with Starbash("info.target") as sb:
+        console.print("[yellow]Not yet implemented[/yellow]")
+        console.print(
+            "This command will list all unique targets in the current selection."
+        )
+
+
+@app.command()
+def telescope():
+    """List telescopes/instruments (filtered based on the current selection)."""
+    with Starbash("info.telescope") as sb:
+        console.print("[yellow]Not yet implemented[/yellow]")
+        console.print(
+            "This command will list all unique telescopes in the current selection."
+        )
+
+
+@app.command()
+def filter():
+    """List all filters found in current selection."""
+    with Starbash("info.filter") as sb:
+        console.print("[yellow]Not yet implemented[/yellow]")
+        console.print(
+            "This command will list all unique filters in the current selection."
+        )
+
+
+@app.callback(invoke_without_command=True)
+def main_callback(ctx: typer.Context):
+    """Show user preferences location and other app info.
+
+    This is the default command when no subcommand is specified.
+    """
+    if ctx.invoked_subcommand is None:
+        with Starbash("info") as sb:
+            from rich.table import Table
+
+            table = Table(title="Starbash Information")
+            table.add_column("Setting", style="cyan", no_wrap=True)
+            table.add_column("Value", style="green")
+
+            # Show config and data directories
+            table.add_row("Config Directory", str(get_user_config_dir()))
+            table.add_row("Data Directory", str(get_user_data_dir()))
+
+            # Show user preferences if set
+            user_name = sb.user_repo.get("user.name")
+            if user_name:
+                table.add_row("User Name", str(user_name))
+
+            user_email = sb.user_repo.get("user.email")
+            if user_email:
+                table.add_row("User Email", str(user_email))
+
+            # Show analytics setting
+            analytics_enabled = sb.user_repo.get("analytics.enabled", True)
+            table.add_row("Analytics", "Enabled" if analytics_enabled else "Disabled")
+
+            # Show number of repos
+            table.add_row("Total Repositories", str(len(sb.repo_manager.repos)))
+            table.add_row("User Repositories", str(len(sb.repo_manager.regular_repos)))
+
+            # Show database stats
+            table.add_row(
+                "Sessions Indexed", str(sb.db.len_table(Database.SESSIONS_TABLE))
+            )
+
+            table.add_row("Images Indexed", str(sb.db.len_table(Database.IMAGES_TABLE)))
+
+            total_exptime = sb.db.sum_column(Database.SESSIONS_TABLE, "exptime_total")
+            table.add_row(
+                "Total image time",
+                format_duration(total_exptime),
+            )
+            console.print(table)

{starbash-0.1.4 → starbash-0.1.5}/src/starbash/commands/select.py

@@ -10,6 +10,7 @@ from rich.table import Table
 from starbash.app import Starbash, copy_images_to_dir
 from starbash.database import Database
 from starbash import console
+from starbash.commands import format_duration
 
 app = typer.Typer()
 
@@ -114,20 +115,6 @@ def date(
             raise typer.Exit(1)
 
 
-def format_duration(seconds: int):
-    """Format seconds as a human-readable duration string."""
-    if seconds < 60:
-        return f"{int(seconds)}s"
-    elif seconds < 120:
-        minutes = int(seconds // 60)
-        secs = int(seconds % 60)
-        return f"{minutes}m {secs}s" if secs else f"{minutes}m"
-    else:
-        hours = int(seconds // 3600)
-        minutes = int((seconds % 3600) // 60)
-        return f"{hours}h {minutes}m" if minutes else f"{hours}h"
-
-
 @app.command(name="list")
 def list_sessions():
     """List sessions (filtered based on the current selection)"""
@@ -135,7 +122,7 @@ def list_sessions():
     with Starbash("selection.list") as sb:
         sessions = sb.search_session()
         if sessions and isinstance(sessions, list):
-            len_all = sb.db.len_session()
+            len_all = sb.db.len_table(Database.SESSIONS_TABLE)
             table = Table(title=f"Sessions ({len(sessions)} selected out of {len_all})")
             sb.analytics.set_data("session.num_selected", len(sessions))
             sb.analytics.set_data("session.num_total", len_all)

{starbash-0.1.4 → starbash-0.1.5}/src/starbash/database.py

@@ -33,6 +33,9 @@ class Database:
     OBJECT_KEY = "OBJECT"
     TELESCOP_KEY = "TELESCOP"
 
+    SESSIONS_TABLE = "sessions"
+    IMAGES_TABLE = "images"
+
     def __init__(
         self,
         base_dir: Optional[Path] = None,
@@ -59,8 +62,8 @@ class Database:
 
         # Create images table with DATE-OBS and DATE as indexed columns
         cursor.execute(
-            """
-            CREATE TABLE IF NOT EXISTS images (
+            f"""
+            CREATE TABLE IF NOT EXISTS {self.IMAGES_TABLE} (
                 id INTEGER PRIMARY KEY AUTOINCREMENT,
                 path TEXT UNIQUE NOT NULL,
                 date_obs TEXT,
@@ -72,29 +75,29 @@ class Database:
 
         # Create index on path for faster lookups
         cursor.execute(
-            """
-            CREATE INDEX IF NOT EXISTS idx_images_path ON images(path)
+            f"""
+            CREATE INDEX IF NOT EXISTS idx_images_path ON {self.IMAGES_TABLE}(path)
         """
         )
 
         # Create index on date_obs for efficient date range queries
         cursor.execute(
-            """
-            CREATE INDEX IF NOT EXISTS idx_images_date_obs ON images(date_obs)
+            f"""
+            CREATE INDEX IF NOT EXISTS idx_images_date_obs ON {self.IMAGES_TABLE}(date_obs)
         """
         )
 
         # Create index on date for queries using DATE field
         cursor.execute(
-            """
-            CREATE INDEX IF NOT EXISTS idx_images_date ON images(date)
+            f"""
+            CREATE INDEX IF NOT EXISTS idx_images_date ON {self.IMAGES_TABLE}(date)
         """
         )
 
         # Create sessions table
         cursor.execute(
-            """
-            CREATE TABLE IF NOT EXISTS sessions (
+            f"""
+            CREATE TABLE IF NOT EXISTS {self.SESSIONS_TABLE} (
                 id INTEGER PRIMARY KEY AUTOINCREMENT,
                 start TEXT NOT NULL,
                 end TEXT NOT NULL,
@@ -111,9 +114,9 @@ class Database:
 
         # Create index on session attributes for faster queries
         cursor.execute(
-            """
+            f"""
             CREATE INDEX IF NOT EXISTS idx_sessions_lookup
-            ON sessions(filter, imagetyp, object, telescop, start, end)
+            ON {self.SESSIONS_TABLE}(filter, imagetyp, object, telescop, start, end)
         """
         )
 
@@ -141,8 +144,8 @@ class Database:
 
         cursor = self._db.cursor()
         cursor.execute(
-            """
-            INSERT INTO images (path, date_obs, date, metadata) VALUES (?, ?, ?, ?)
+            f"""
+            INSERT INTO {self.IMAGES_TABLE} (path, date_obs, date, metadata) VALUES (?, ?, ?, ?)
             ON CONFLICT(path) DO UPDATE SET
                 date_obs = excluded.date_obs,
                 date = excluded.date,
@@ -154,7 +157,7 @@ class Database:
         self._db.commit()
 
         # Get the rowid of the inserted/updated record
-        cursor.execute("SELECT id FROM images WHERE path = ?", (path,))
+        cursor.execute(f"SELECT id FROM {self.IMAGES_TABLE} WHERE path = ?", (path,))
         result = cursor.fetchone()
         if result:
             return result[0]
@@ -190,7 +193,7 @@ class Database:
             params.append(date_end)
 
         # Build the query
-        query = "SELECT id, path, date_obs, date, metadata FROM images"
+        query = f"SELECT id, path, date_obs, date, metadata FROM {self.IMAGES_TABLE}"
         if where_clauses:
             query += " WHERE " + " AND ".join(where_clauses)
 
@@ -236,10 +239,10 @@ class Database:
 
         cursor = self._db.cursor()
         cursor.execute(
-            """
+            f"""
             SELECT id, start, end, filter, imagetyp, object, telescop,
                    num_images, exptime_total, image_doc_id
-            FROM sessions
+            FROM {self.SESSIONS_TABLE}
         """
         )
 
@@ -286,17 +289,17 @@ class Database:
 
         return results if results else None
 
-    def len_session(self) -> int:
-        """Return the total number of sessions."""
+    def len_table(self, table_name: str) -> int:
+        """Return the total number of rows in the specified table."""
         cursor = self._db.cursor()
-        cursor.execute("SELECT COUNT(*) FROM sessions")
+        cursor.execute(f"SELECT COUNT(*) FROM {table_name}")
         result = cursor.fetchone()
         return result[0] if result else 0
 
-    def get_column(self, column_name: str, table_name: str = "sessions") -> list[Any]:
+    def get_column(self, table_name: str, column_name: str) -> list[Any]:
         """Return all values from a specific column in the specified table."""
         cursor = self._db.cursor()
-        cursor.execute(f"SELECT {column_name} FROM {table_name}")
+        cursor.execute(f'SELECT "{column_name}" FROM {table_name}')
 
         results = []
         for row in cursor.fetchall():
@@ -304,11 +307,18 @@ class Database:
 
         return results
 
+    def sum_column(self, table_name: str, column_name: str) -> float:
+        """Return the SUM of all values in a specific column in the specified table."""
+        cursor = self._db.cursor()
+        cursor.execute(f'SELECT SUM("{column_name}") FROM {table_name}')
+        result = cursor.fetchone()
+        return result[0] if result and result[0] is not None else 0
+
     def get_image(self, path: str) -> dict[str, Any] | None:
         """Get an image record by path."""
         cursor = self._db.cursor()
         cursor.execute(
-            "SELECT id, path, date_obs, date, metadata FROM images WHERE path = ?",
+            f"SELECT id, path, date_obs, date, metadata FROM {self.IMAGES_TABLE} WHERE path = ?",
             (path,),
         )
         row = cursor.fetchone()
@@ -331,7 +341,9 @@ class Database:
     def all_images(self) -> list[dict[str, Any]]:
         """Return all image records."""
         cursor = self._db.cursor()
-        cursor.execute("SELECT id, path, date_obs, date, metadata FROM images")
+        cursor.execute(
+            f"SELECT id, path, date_obs, date, metadata FROM {self.IMAGES_TABLE}"
+        )
 
         results = []
         for row in cursor.fetchall():
@@ -353,10 +365,10 @@ class Database:
         """Return all session records."""
         cursor = self._db.cursor()
         cursor.execute(
-            """
+            f"""
             SELECT id, start, end, filter, imagetyp, object, telescop,
                    num_images, exptime_total, image_doc_id
-            FROM sessions
+            FROM {self.SESSIONS_TABLE}
         """
         )
 
@@ -389,10 +401,10 @@ class Database:
         """
         cursor = self._db.cursor()
         cursor.execute(
-            """
+            f"""
             SELECT id, start, end, filter, imagetyp, object, telescop,
                    num_images, exptime_total, image_doc_id
-            FROM sessions
+            FROM {self.SESSIONS_TABLE}
             WHERE id = ?
         """,
             (session_id,),
@@ -443,10 +455,10 @@ class Database:
         # comparison aligns with chronological ordering for a uniform format.
         cursor = self._db.cursor()
         cursor.execute(
-            """
+            f"""
             SELECT id, start, end, filter, imagetyp, object, telescop,
                    num_images, exptime_total, image_doc_id
-            FROM sessions
+            FROM {self.SESSIONS_TABLE}
             WHERE filter = ? AND imagetyp = ? AND object = ? AND telescop = ?
               AND start >= ? AND start <= ?
             LIMIT 1
@@ -489,8 +501,8 @@ class Database:
             ) + new.get(Database.EXPTIME_TOTAL_KEY, 0)
 
             cursor.execute(
-                """
-                UPDATE sessions
+                f"""
+                UPDATE {self.SESSIONS_TABLE}
                 SET start = ?, end = ?, num_images = ?, exptime_total = ?
                 WHERE id = ?
             """,
@@ -505,8 +517,8 @@ class Database:
         else:
             # Insert new session
             cursor.execute(
-                """
-                INSERT INTO sessions
+                f"""
+                INSERT INTO {self.SESSIONS_TABLE}
                 (start, end, filter, imagetyp, object, telescop, num_images, exptime_total, image_doc_id)
                 VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
             """,

{starbash-0.1.4 → starbash-0.1.5}/src/starbash/main.py

@@ -6,7 +6,7 @@ import starbash.url as url
 import starbash
 
 from .app import Starbash, get_user_config_path, setup_logging
-from .commands import repo, select, user
+from .commands import info, repo, select, user
 from . import console
 
 app = typer.Typer(
@@ -16,6 +16,7 @@ app = typer.Typer(
 app.add_typer(user.app, name="user", help="Manage user settings.")
 app.add_typer(repo.app, name="repo", help="Manage Starbash repositories.")
 app.add_typer(select.app, name="select", help="Manage session and target selection.")
+app.add_typer(info.app, name="info", help="Display system and data information.")
 
 
 @app.callback(invoke_without_command=True)