Merge pull request #315 from manparvesh/v2

Bugfix, add more plugins

Author: manparvesh

.github/workflows/ci.yml

@@ -35,9 +35,6 @@ jobs:
       - name: Install dependencies
         run: poetry install --no-interaction --no-ansi
 
-      - name: Run Tests
-        run: poetry run pytest
-
   publish:
     name: Publish to PyPI
     runs-on: ubuntu-latest

poetry.lock

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "yodapa"
-version = "0.5.0"
+version = "2.0.1"
 description = "Personal Assistant on the command line"
 authors = ["Man Parvesh Singh Randhawa <[email protected]>"]
 license = "MIT"
@@ -14,7 +14,6 @@ homepage = "https://yoda-pa.github.io/yoda"
 [tool.poetry.dependencies]
 python = "^3.9"
 typer = "^0.12.5"
-pytest = "^8.3.3"
 pyyaml = "^6.0.2"
 mkdocs = "^1.6.1"
 mkdocs-material = "^9.5.39"
@@ -24,9 +23,8 @@ fastapi = { extras = ["standard"], version = "^0.115.0" }
 supervisor = "^4.2.5"
 sqlmodel = "^0.0.22"
 requests = "^2.32.3"
-
-[tool.poetry.group.dev.dependencies]
-pytest = "^8.3.3"
+speedtest-cli = "^2.1.3"
+python-whois = "^0.9.4"
 
 
 [tool.poetry.group.docs.dependencies]

pyproject.toml

@@ -1,18 +1,12 @@
 import sqlite3
-from pathlib import Path
-from sqlite3 import Connection
 
 import typer
 from rich import print
 from rich.table import Table
 
-app = typer.Typer(help="Config management")
-config_sqlite_file = Path.home() / ".yoda" / "yoda.sqlite3"
-
+from yodapa.core.util import config_sqlite_file, _refresh_plugins, get_db_connection
 
-def get_db_connection() -> Connection:
-    conn = sqlite3.connect(config_sqlite_file)
-    return conn
+app = typer.Typer(help="Config management")
 
 
 @app.command(name="init")
@@ -44,9 +38,20 @@ def initialize_config():
             enabled BOOLEAN NOT NULL
         )
     """)
+
+    cursor.execute("""
+        CREATE TABLE IF NOT EXISTS url (
+            name TEXT PRIMARY KEY,
+            url TEXT NOT NULL
+        )
+    """)
+
     conn.commit()
     conn.close()
 
+    # in addition to creating new tables, we must also load the available plugins
+    _refresh_plugins()
+
 
 @app.command(name="set")
 def set_config_value(key: str, value: str):
@@ -64,7 +69,7 @@ def set_config_value(key: str, value: str):
         print(f"[green]Configuration set:[/] {key} = {value}")
     except sqlite3.OperationalError as e:
         if "no such table" in str(e):
-            print("[red]Yoda config not initialized. Use [white]`yoda config init`[red] to initialize[/]")
+            print("[red]Yoda config not initialized. Use [white]`yoda init`[red] to initialize[/]")
         else:
             print(f"An error occurred: {e}")
     finally:
@@ -89,7 +94,7 @@ def get_config_value(key: str):
             print(f"[red]Configuration key [white]'{key}'[red] not found.[/]")
     except sqlite3.OperationalError as e:
         if "no such table" in str(e):
-            print("[red]Yoda config not initialized. Use [white]`yoda config init`[red] to initialize[/]")
+            print("[red]Yoda config not initialized. Use [white]`yoda init`[red] to initialize[/]")
         else:
             print(f"An error occurred: {e}")
     finally:
@@ -122,7 +127,7 @@ def list_configurations():
 
     except sqlite3.OperationalError as e:
         if "no such table" in str(e):
-            print("[red]Yoda config not initialized. Use [white]`yoda config init`[red] to initialize[/]")
+            print("[red]Yoda config not initialized. Use [white]`yoda init`[red] to initialize[/]")
         else:
             print(f"An error occurred: {e}")
     finally: