cli-anything-drms 0.1.0__tar.gz

cli_anything_drms-0.1.0/PKG-INFO

@@ -0,0 +1,10 @@
+Metadata-Version: 2.4
+Name: cli-anything-drms
+Version: 0.1.0
+Summary: CLI-Anything harness for the DRMS / HTM FastAPI backend
+Requires-Python: >=3.10
+Requires-Dist: click>=8.0
+Requires-Dist: requests>=2.28
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary

cli_anything_drms-0.1.0/cli_anything/drms/__main__.py

@@ -0,0 +1,4 @@
+from cli_anything.drms.drms_cli import cli
+
+if __name__ == "__main__":
+    cli()

cli_anything_drms-0.1.0/cli_anything/drms/core/case_notes.py

@@ -0,0 +1,27 @@
+"""Case notes domain operations."""
+
+from typing import Any
+from cli_anything.drms.core.client import DrmsClient
+
+
+def list_case_notes(client: DrmsClient, patient_uuid: str) -> Any:
+    return client.get(f"/patient/case_notes/v2/{patient_uuid}")
+
+
+def get_case_note(client: DrmsClient, note_uuid: str) -> Any:
+    return client.get(f"/patient/case_notes/v2/note/{note_uuid}")
+
+
+def generate_case_note_report(client: DrmsClient, patient_id: str, design_type: str = "new") -> Any:
+    return client.post(
+        f"/patient/case_notes/v2/report/{patient_id}",
+        data={"designtype": design_type},
+    )
+
+
+def list_case_note_categories(client: DrmsClient) -> Any:
+    return client.get("/patient/case_notes/v2/categories")
+
+
+def list_case_note_templates(client: DrmsClient) -> Any:
+    return client.get("/patient/case_notes/v2/templates")

cli_anything_drms-0.1.0/cli_anything/drms/core/client.py

@@ -0,0 +1,45 @@
+"""HTTP client wrapper for the DRMS / HTM FastAPI backend."""
+
+import json
+from typing import Any, Optional
+import requests
+from cli_anything.drms.utils.session import require_token
+
+
+class DrmsClient:
+    def __init__(self, base_url: Optional[str] = None, token: Optional[str] = None):
+        if base_url and token:
+            self.base_url = base_url.rstrip("/")
+            self.token = token
+        else:
+            self.base_url, self.token = require_token()
+
+    def _headers(self) -> dict:
+        return {"Authorization": f"Bearer {self.token}"}
+
+    def get(self, path: str, params: Optional[dict] = None) -> Any:
+        url = f"{self.base_url}{path}"
+        r = requests.get(url, headers=self._headers(), params=params, timeout=30)
+        r.raise_for_status()
+        return r.json()
+
+    def post(self, path: str, data: Any = None, files: Optional[dict] = None) -> Any:
+        url = f"{self.base_url}{path}"
+        if files:
+            r = requests.post(url, headers=self._headers(), data=data, files=files, timeout=60)
+        else:
+            r = requests.post(url, headers=self._headers(), json=data, timeout=30)
+        r.raise_for_status()
+        return r.json()
+
+    def put(self, path: str, data: Any = None) -> Any:
+        url = f"{self.base_url}{path}"
+        r = requests.put(url, headers=self._headers(), json=data, timeout=30)
+        r.raise_for_status()
+        return r.json()
+
+    def delete(self, path: str) -> Any:
+        url = f"{self.base_url}{path}"
+        r = requests.delete(url, headers=self._headers(), timeout=30)
+        r.raise_for_status()
+        return r.json()

cli_anything_drms-0.1.0/cli_anything/drms/core/lab_reports.py

@@ -0,0 +1,16 @@
+"""Lab report domain operations."""
+
+from typing import Any
+from cli_anything.drms.core.client import DrmsClient
+
+
+def get_lab_reports(client: DrmsClient, patient_uuid: str) -> Any:
+    return client.get(f"/health_info/lab_report/patient/{patient_uuid}")
+
+
+def get_lab_report(client: DrmsClient, report_uuid: str) -> Any:
+    return client.get(f"/health_info/lab_report/get/{report_uuid}")
+
+
+def get_lab_trends(client: DrmsClient, patient_uuid: str) -> Any:
+    return client.get(f"/health_info/lab_report/patient/{patient_uuid}/trends")

cli_anything_drms-0.1.0/cli_anything/drms/core/medication.py

@@ -0,0 +1,20 @@
+"""Medication domain operations."""
+
+from typing import Any
+from cli_anything.drms.core.client import DrmsClient
+
+
+def get_medications(client: DrmsClient, patient_id: str) -> Any:
+    return client.get(f"/patient/v3/medication/{patient_id}")
+
+
+def get_daily_medication(client: DrmsClient, patient_id: str) -> Any:
+    return client.get(f"/daily-medication/{patient_id}")
+
+
+def get_medication_monitoring(client: DrmsClient, patient_id: str) -> Any:
+    return client.get(f"/medication-monitoring/{patient_id}")
+
+
+def get_health_summary(client: DrmsClient, patient_id: str) -> Any:
+    return client.get(f"/patient/v5/health-summary/{patient_id}")

cli_anything_drms-0.1.0/cli_anything/drms/core/patients.py

@@ -0,0 +1,20 @@
+"""Patient domain operations."""
+
+from typing import Any
+from cli_anything.drms.core.client import DrmsClient
+
+
+def search_patients(client: DrmsClient, term: str) -> Any:
+    return client.get(f"/patient/v2/search/{term}")
+
+
+def list_patients(client: DrmsClient, page: int = 1) -> Any:
+    return client.get(f"/patient/v2/all/{page}")
+
+
+def get_patient(client: DrmsClient, patient_id: str) -> Any:
+    return client.get(f"/patient/v2/get/{patient_id}")
+
+
+def list_patients_by_condition(client: DrmsClient, condition: str, page: int = 1) -> Any:
+    return client.get("/patient-list-by-conditions/by-condition", params={"condition": condition, "page": page})

cli_anything_drms-0.1.0/cli_anything/drms/core/staff.py

@@ -0,0 +1,12 @@
+"""Staff/organisation domain operations."""
+
+from typing import Any
+from cli_anything.drms.core.client import DrmsClient
+
+
+def list_staff(client: DrmsClient) -> Any:
+    return client.get("/admin/staffs")
+
+
+def get_staff(client: DrmsClient, firebase_uid: str) -> Any:
+    return client.get(f"/admin/staff/{firebase_uid}")

cli_anything_drms-0.1.0/cli_anything/drms/core/stats.py

@@ -0,0 +1,40 @@
+"""Organisation statistics — action logs, health conditions, CPD analytics."""
+
+from typing import Any, List, Optional
+from cli_anything.drms.core.client import DrmsClient
+
+
+def get_action_stats(client: DrmsClient, start_date: Optional[str] = None, end_date: Optional[str] = None) -> Any:
+    params = {}
+    if start_date:
+        params["start_date"] = start_date
+    if end_date:
+        params["end_date"] = end_date
+    return client.get("/admin/action_stats_v2", params=params or None)
+
+
+def get_condition_stats(client: DrmsClient, start_date: Optional[str] = None, end_date: Optional[str] = None) -> Any:
+    params = {}
+    if start_date:
+        params["start_date"] = start_date
+    if end_date:
+        params["end_date"] = end_date
+    return client.get("/admin/condition_stats", params=params or None)
+
+
+def get_cpd_analytics(
+    client: DrmsClient,
+    start_date: Optional[str] = None,
+    end_date: Optional[str] = None,
+    premise_ids: Optional[List[str]] = None,
+    include_inactive: bool = False,
+    top_n: int = 100,
+) -> Any:
+    params: dict = {"top_n": top_n, "include_inactive_staff": str(include_inactive).lower()}
+    if start_date:
+        params["start_date"] = start_date
+    if end_date:
+        params["end_date"] = end_date
+    if premise_ids:
+        params["premise_ids"] = premise_ids
+    return client.get("/admin/cpd_analytics", params=params)

cli_anything_drms-0.1.0/cli_anything/drms/drms_cli.py

@@ -0,0 +1,516 @@
+"""DRMS / HTM CLI — member search, discovery, recommendations, and patient data."""
+
+import json
+import sys
+from typing import Optional
+
+import click
+import requests
+
+from cli_anything.drms.core.client import DrmsClient
+from cli_anything.drms.core import patients, lab_reports, case_notes, medication, staff, stats
+from cli_anything.drms.drms_feature import member, discovery, recommend
+from cli_anything.drms.utils.session import save_token, save_session, clear_session, get_session
+from cli_anything.drms.utils import auth as _auth
+from cli_anything.drms.utils.repl_skin import ReplSkin
+
+
+# ── Helpers ───────────────────────────────────────────────────────────────────
+
+def _client(ctx: click.Context) -> DrmsClient:
+    return DrmsClient()
+
+
+def _out(ctx: click.Context, data) -> None:
+    if ctx.obj.get("json"):
+        click.echo(json.dumps(data, indent=2, default=str))
+    else:
+        if isinstance(data, (dict, list)):
+            click.echo(json.dumps(data, indent=2, default=str))
+        else:
+            click.echo(str(data))
+
+
+def _handle_error(e: Exception) -> None:
+    if isinstance(e, requests.HTTPError):
+        try:
+            detail = e.response.json().get("detail", str(e))
+        except Exception:
+            detail = str(e)
+        click.echo(f"API error {e.response.status_code}: {detail}", err=True)
+    else:
+        click.echo(f"Error: {e}", err=True)
+    sys.exit(1)
+
+
+# ── Root group ────────────────────────────────────────────────────────────────
+
+@click.group()
+@click.option("--json", "json_mode", is_flag=True, default=False, help="Output raw JSON.")
+@click.pass_context
+def cli(ctx: click.Context, json_mode: bool) -> None:
+    """DRMS / HTM CLI — member search, discovery, drug recommendations, and patient data."""
+    ctx.ensure_object(dict)
+    ctx.obj["json"] = json_mode
+
+
+# ── Auth ──────────────────────────────────────────────────────────────────────
+
+@cli.command()
+@click.option("--url", default="https://htm.aipharm.xyz", show_default=True, help="HTM API base URL.")
+@click.option("--token", default=None, hidden=True, help="Pre-issued token (CI/agent use only).")
+def login(url: str, token: str) -> None:
+    """Log in to DRMS / HTM. Prompts for email and password."""
+    if token:
+        save_token(url, token)
+        click.echo(f"Session saved (token mode). Base URL: {url}")
+        return
+
+    email = click.prompt("Email")
+    password = click.prompt("Password", hide_input=True)
+
+    click.echo("Authenticating…")
+    try:
+        resp = _auth.login(url, email, password)
+    except ValueError as e:
+        click.echo(str(e), err=True)
+        sys.exit(1)
+
+    id_token = resp["firebase_token"]
+    local_id = resp["user_id"]
+
+    save_session(url, id_token, resp["refresh_token"], local_id, email)
+    click.echo(f"Logged in as {email}")
+
+
+@cli.command()
+def logout() -> None:
+    """Log out and revoke tokens server-side."""
+    data = get_session()
+    if data.get("token") and data.get("base_url"):
+        _auth.logout_server(data["base_url"], data["token"])
+    clear_session()
+    click.echo("Logged out.")
+
+
+@cli.command()
+def whoami() -> None:
+    """Show current session info."""
+    data = get_session()
+    if not data or not data.get("token"):
+        click.echo("Not logged in.")
+        return
+    click.echo(f"URL  : {data.get('base_url', 'n/a')}")
+    click.echo(f"Email: {data.get('email', '(token mode)')}")
+    issued = data.get("issued_at")
+    if issued:
+        import time
+        age_min = int((time.time() - issued) / 60)
+        click.echo(f"Token: {age_min} min old (auto-refreshes at 55 min)")
+    else:
+        click.echo(f"Token: {data.get('token', '')[:24]}… (manual token mode)")
+
+
+# ── Patient commands ──────────────────────────────────────────────────────────
+
+@cli.group()
+def patient() -> None:
+    """Patient search and lookup."""
+
+
+@patient.command("search")
+@click.argument("term")
+@click.pass_context
+def patient_search(ctx, term: str) -> None:
+    """Search patients by name, IC, or ref ID."""
+    try:
+        _out(ctx, patients.search_patients(_client(ctx), term))
+    except Exception as e:
+        _handle_error(e)
+
+
+@patient.command("list")
+@click.option("--page", default=1, show_default=True)
+@click.pass_context
+def patient_list(ctx, page: int) -> None:
+    """List patients (paginated)."""
+    try:
+        _out(ctx, patients.list_patients(_client(ctx), page))
+    except Exception as e:
+        _handle_error(e)
+
+
+@patient.command("get")
+@click.argument("patient_id")
+@click.pass_context
+def patient_get(ctx, patient_id: str) -> None:
+    """Get a patient by ID."""
+    try:
+        _out(ctx, patients.get_patient(_client(ctx), patient_id))
+    except Exception as e:
+        _handle_error(e)
+
+
+@patient.command("by-condition")
+@click.argument("condition")
+@click.option("--page", default=1, show_default=True)
+@click.pass_context
+def patient_by_condition(ctx, condition: str, page: int) -> None:
+    """List patients by medical condition tag."""
+    try:
+        _out(ctx, patients.list_patients_by_condition(_client(ctx), condition, page))
+    except Exception as e:
+        _handle_error(e)
+
+
+# ── Lab report commands ───────────────────────────────────────────────────────
+
+@cli.group()
+def lab() -> None:
+    """Lab report data and trends."""
+
+
+@lab.command("list")
+@click.argument("patient_uuid")
+@click.pass_context
+def lab_list(ctx, patient_uuid: str) -> None:
+    """List all lab reports for a patient."""
+    try:
+        _out(ctx, lab_reports.get_lab_reports(_client(ctx), patient_uuid))
+    except Exception as e:
+        _handle_error(e)
+
+
+@lab.command("get")
+@click.argument("report_uuid")
+@click.pass_context
+def lab_get(ctx, report_uuid: str) -> None:
+    """Get a single lab report with all results."""
+    try:
+        _out(ctx, lab_reports.get_lab_report(_client(ctx), report_uuid))
+    except Exception as e:
+        _handle_error(e)
+
+
+@lab.command("trends")
+@click.argument("patient_uuid")
+@click.pass_context
+def lab_trends(ctx, patient_uuid: str) -> None:
+    """Show lab result trends for a patient."""
+    try:
+        _out(ctx, lab_reports.get_lab_trends(_client(ctx), patient_uuid))
+    except Exception as e:
+        _handle_error(e)
+
+
+# ── Case note commands ────────────────────────────────────────────────────────
+
+@cli.group()
+def note() -> None:
+    """Case note lookup and reference data."""
+
+
+@note.command("list")
+@click.argument("patient_uuid")
+@click.pass_context
+def note_list(ctx, patient_uuid: str) -> None:
+    """List all case notes for a patient."""
+    try:
+        _out(ctx, case_notes.list_case_notes(_client(ctx), patient_uuid))
+    except Exception as e:
+        _handle_error(e)
+
+
+@note.command("get")
+@click.argument("note_uuid")
+@click.pass_context
+def note_get(ctx, note_uuid: str) -> None:
+    """Get a single case note."""
+    try:
+        _out(ctx, case_notes.get_case_note(_client(ctx), note_uuid))
+    except Exception as e:
+        _handle_error(e)
+
+
+@note.command("report")
+@click.argument("patient_id")
+@click.option("--design", default="new", type=click.Choice(["old", "new", "markdown"]), show_default=True)
+@click.pass_context
+def note_report(ctx, patient_id: str, design: str) -> None:
+    """Generate a PDF case note report for a patient."""
+    try:
+        _out(ctx, case_notes.generate_case_note_report(_client(ctx), patient_id, design))
+    except Exception as e:
+        _handle_error(e)
+
+
+# ── Medication commands ───────────────────────────────────────────────────────
+
+@cli.group()
+def med() -> None:
+    """Medication and health summary."""
+
+
+@med.command("list")
+@click.argument("patient_id")
+@click.pass_context
+def med_list(ctx, patient_id: str) -> None:
+    """List medications for a patient."""
+    try:
+        _out(ctx, medication.get_medications(_client(ctx), patient_id))
+    except Exception as e:
+        _handle_error(e)
+
+
+@med.command("daily")
+@click.argument("patient_id")
+@click.pass_context
+def med_daily(ctx, patient_id: str) -> None:
+    """Show daily medication schedule for a patient."""
+    try:
+        _out(ctx, medication.get_daily_medication(_client(ctx), patient_id))
+    except Exception as e:
+        _handle_error(e)
+
+
+@med.command("summary")
+@click.argument("patient_id")
+@click.pass_context
+def med_summary(ctx, patient_id: str) -> None:
+    """Show health summary for a patient."""
+    try:
+        _out(ctx, medication.get_health_summary(_client(ctx), patient_id))
+    except Exception as e:
+        _handle_error(e)
+
+
+# ── Staff commands ────────────────────────────────────────────────────────────
+
+@cli.group("staff")
+def staff_cmd() -> None:
+    """Staff and organisation lookup."""
+
+
+@staff_cmd.command("list")
+@click.pass_context
+def staff_list(ctx) -> None:
+    """List all staff in the organisation."""
+    try:
+        _out(ctx, staff.list_staff(_client(ctx)))
+    except Exception as e:
+        _handle_error(e)
+
+
+@staff_cmd.command("get")
+@click.argument("firebase_uid")
+@click.pass_context
+def staff_get(ctx, firebase_uid: str) -> None:
+    """Get a staff member by Firebase UID."""
+    try:
+        _out(ctx, staff.get_staff(_client(ctx), firebase_uid))
+    except Exception as e:
+        _handle_error(e)
+
+
+# ── Stats commands ────────────────────────────────────────────────────────────
+
+@cli.group("stats")
+def stats_group() -> None:
+    """Organisation statistics (admin / branch manager only)."""
+
+
+@stats_group.command("actions")
+@click.option("--start", default=None, help="Start date ISO-8601 (default: 7 days ago).")
+@click.option("--end", default=None, help="End date ISO-8601 (default: now).")
+@click.pass_context
+def stats_actions(ctx, start, end) -> None:
+    """Action-log analytics: staff performance, branch activeness, behaviour insights."""
+    try:
+        _out(ctx, stats.get_action_stats(_client(ctx), start, end))
+    except Exception as e:
+        _handle_error(e)
+
+
+@stats_group.command("conditions")
+@click.option("--start", default=None, help="Start date ISO-8601 (default: 7 days ago).")
+@click.option("--end", default=None, help="End date ISO-8601 (default: now).")
+@click.pass_context
+def stats_conditions(ctx, start, end) -> None:
+    """Health condition frequency: counts by condition, branch, age group, and gender."""
+    try:
+        _out(ctx, stats.get_condition_stats(_client(ctx), start, end))
+    except Exception as e:
+        _handle_error(e)
+
+
+@stats_group.command("cpd")
+@click.option("--start", default=None)
+@click.option("--end", default=None)
+@click.option("--premise", "premise_ids", multiple=True)
+@click.option("--include-inactive", is_flag=True, default=False)
+@click.option("--top-n", default=100, show_default=True)
+@click.pass_context
+def stats_cpd(ctx, start, end, premise_ids, include_inactive, top_n) -> None:
+    """CPD analytics: enrolments, completions, leaderboard, at-risk cohort."""
+    try:
+        _out(ctx, stats.get_cpd_analytics(_client(ctx), start, end, list(premise_ids) or None, include_inactive, top_n))
+    except Exception as e:
+        _handle_error(e)
+
+
+# ── HTM Member commands ───────────────────────────────────────────────────────
+
+@cli.group("member")
+def member_group() -> None:
+    """HTM member search and purchase history."""
+
+
+@member_group.command("search")
+@click.argument("query")
+@click.option("--by-name", is_flag=True, default=False, help="Search by name instead of IC.")
+@click.pass_context
+def member_search(ctx, query: str, by_name: bool) -> None:
+    """Search HTM members by IC number (or name with --by-name)."""
+    try:
+        _out(ctx, member.search_members(_client(ctx), query, by_name))
+    except Exception as e:
+        _handle_error(e)
+
+
+@member_group.command("search-card")
+@click.argument("query")
+@click.option("--by-name", is_flag=True, default=False)
+@click.pass_context
+def member_search_card(ctx, query: str, by_name: bool) -> None:
+    """Search HTM members by loyalty card number."""
+    try:
+        _out(ctx, member.search_members_by_card(_client(ctx), query, by_name))
+    except Exception as e:
+        _handle_error(e)
+
+
+@member_group.command("purchases")
+@click.argument("member_id")
+@click.pass_context
+def member_purchases(ctx, member_id: str) -> None:
+    """Show purchase history for a member."""
+    try:
+        _out(ctx, member.get_purchases(_client(ctx), member_id))
+    except Exception as e:
+        _handle_error(e)
+
+
+@member_group.command("labs")
+@click.argument("patient_id")
+@click.pass_context
+def member_labs(ctx, patient_id: str) -> None:
+    """Show HTM lab reports for a patient."""
+    try:
+        _out(ctx, member.get_lab_reports(_client(ctx), patient_id))
+    except Exception as e:
+        _handle_error(e)
+
+
+@member_group.command("labs-all")
+@click.pass_context
+def member_labs_all(ctx) -> None:
+    """List all HTM lab reports for the organisation."""
+    try:
+        _out(ctx, member.list_lab_reports(_client(ctx)))
+    except Exception as e:
+        _handle_error(e)
+
+
+# ── Discovery commands ────────────────────────────────────────────────────────
+
+@cli.group("discovery")
+def discovery_group() -> None:
+    """DRMS illness discovery — symptoms, illness search, checker, analyses."""
+
+
+@discovery_group.command("complaints")
+@click.option("--invalidate", is_flag=True, default=False, help="Bust the server-side cache.")
+@click.pass_context
+def discovery_complaints(ctx, invalidate: bool) -> None:
+    """List all symptoms and risk factors."""
+    try:
+        _out(ctx, discovery.get_complaints(_client(ctx), invalidate))
+    except Exception as e:
+        _handle_error(e)
+
+
+@discovery_group.command("illnesses")
+@click.option("--q", default=None, help="Keyword search.")
+@click.option("--category", default=None, help="Filter by category.")
+@click.option("--limit", default=20, show_default=True)
+@click.option("--offset", default=0, show_default=True)
+@click.pass_context
+def discovery_illnesses(ctx, q, category, limit, offset) -> None:
+    """Search illnesses by keyword or category."""
+    try:
+        _out(ctx, discovery.search_illnesses(_client(ctx), q, category, limit, offset))
+    except Exception as e:
+        _handle_error(e)
+
+
+@discovery_group.command("recommendations")
+@click.argument("illness_id", type=int)
+@click.pass_context
+def discovery_recommendations(ctx, illness_id: int) -> None:
+    """Show health recommendations for a specific illness ID."""
+    try:
+        _out(ctx, discovery.get_illness_recommendations(_client(ctx), illness_id))
+    except Exception as e:
+        _handle_error(e)
+
+
+@discovery_group.command("analyses")
+@click.argument("patient_id")
+@click.pass_context
+def discovery_analyses(ctx, patient_id: str) -> None:
+    """Show all saved discovery analyses for a patient."""
+    try:
+        _out(ctx, discovery.get_patient_analyses(_client(ctx), patient_id))
+    except Exception as e:
+        _handle_error(e)
+
+
+@discovery_group.command("check")
+@click.argument("symptom_ids", nargs=-1, type=int, required=True)
+@click.pass_context
+def discovery_check(ctx, symptom_ids) -> None:
+    """Run illness checker for the given symptom IDs (space-separated)."""
+    try:
+        _out(ctx, discovery.check_illnesses(_client(ctx), list(symptom_ids)))
+    except Exception as e:
+        _handle_error(e)
+
+
+# ── Recommend commands ────────────────────────────────────────────────────────
+
+@cli.group("recommend")
+def recommend_group() -> None:
+    """HTM ML-based drug/product recommendations."""
+
+
+@recommend_group.command("get")
+@click.argument("query_items")
+@click.option("--top-n", default=3, show_default=True, help="Recommendations per item.")
+@click.pass_context
+def recommend_get(ctx, query_items: str, top_n: int) -> None:
+    """Get ML recommendations for comma-separated product/drug names."""
+    try:
+        _out(ctx, recommend.get_recommendations(_client(ctx), query_items, top_n))
+    except Exception as e:
+        _handle_error(e)
+
+
+# ── REPL ──────────────────────────────────────────────────────────────────────
+
+@cli.command("repl")
+@click.option("--json", "json_mode", is_flag=True, default=False)
+@click.pass_context
+def repl(ctx, json_mode: bool) -> None:
+    """Start interactive REPL session."""
+    skin = ReplSkin(cli, json_mode=json_mode or ctx.obj.get("json", False))
+    skin.run()

cli_anything_drms-0.1.0/cli_anything/drms/drms_feature/discovery.py

@@ -0,0 +1,57 @@
+"""DRMS Discovery v2 domain operations."""
+
+from typing import Any, List, Optional
+from cli_anything.drms.core.client import DrmsClient
+
+
+def get_complaints(client: DrmsClient, invalidate: bool = False) -> Any:
+    """GET /discovery/v2/complaints — all symptoms and risk factors."""
+    return client.get("/discovery/v2/complaints", params={"invalidate": invalidate})
+
+
+def get_screening_data(client: DrmsClient) -> Any:
+    """GET /discovery/v2/screening — full screening dataset (illnesses, symptoms, risk factors)."""
+    return client.get("/discovery/v2/screening")
+
+
+def search_illnesses(
+    client: DrmsClient,
+    q: Optional[str] = None,
+    category: Optional[str] = None,
+    limit: int = 20,
+    offset: int = 0,
+) -> Any:
+    """GET /discovery/v2/possible_illnesses_list — search illnesses by keyword or category."""
+    params: dict = {"limit": limit, "offset": offset}
+    if q:
+        params["q"] = q
+    if category:
+        params["category"] = category
+    return client.get("/discovery/v2/possible_illnesses_list", params=params)
+
+
+def get_illness_recommendations(client: DrmsClient, illness_id: int) -> Any:
+    """GET /discovery/v2/illness_recommendation/{illness_id} — health recommendations for an illness."""
+    return client.get(f"/discovery/v2/illness_recommendation/{illness_id}")
+
+
+def get_patient_analyses(client: DrmsClient, patient_id: str) -> Any:
+    """GET /discovery/v2/get_all_analyses/{patient_id} — all saved analyses for a patient."""
+    return client.get(f"/discovery/v2/get_all_analyses/{patient_id}")
+
+
+def get_analysis(client: DrmsClient, patient_id: str, analyzed_id: str) -> Any:
+    """GET /discovery/v2/get_analysis/{patient_id}/{analyzed_id} — single analysis record."""
+    return client.get(f"/discovery/v2/get_analysis/{patient_id}/{analyzed_id}")
+
+
+def check_illnesses(
+    client: DrmsClient,
+    symptom_ids: List[int],
+    known_illness_ids: Optional[List[int]] = None,
+) -> Any:
+    """POST /discovery/v2/checker — match illnesses from selected symptoms."""
+    return client.post("/discovery/v2/checker", data={
+        "main_sympt_ids": symptom_ids,
+        "known_illness_ids": known_illness_ids or [],
+    })

cli_anything_drms-0.1.0/cli_anything/drms/drms_feature/member.py

@@ -0,0 +1,29 @@
+"""HTM member search and purchase history operations."""
+
+from typing import Any
+from cli_anything.drms.core.client import DrmsClient
+
+
+def search_members(client: DrmsClient, query: str, by_name: bool = False) -> Any:
+    """GET /htm/search_members_v2 — search HTM members by IC or name."""
+    return client.get("/htm/search_members_v2", params={"query": query, "searchname": by_name})
+
+
+def search_members_by_card(client: DrmsClient, query: str, by_name: bool = False) -> Any:
+    """GET /htm/search_members_by_card_v2 — search members by loyalty card."""
+    return client.get("/htm/search_members_by_card_v2", params={"query": query, "searchname": by_name})
+
+
+def get_purchases(client: DrmsClient, member_id: str) -> Any:
+    """GET /htm/v3/search_purchases — purchase history for a member (v3 with remarks)."""
+    return client.get("/htm/v3/search_purchases", params={"member_id": member_id})
+
+
+def get_lab_reports(client: DrmsClient, patient_id: str) -> Any:
+    """GET /htm/htm_labreports/by_patient_id — HTM lab reports for a patient."""
+    return client.get("/htm/htm_labreports/by_patient_id", params={"PatientID": patient_id})
+
+
+def list_lab_reports(client: DrmsClient) -> Any:
+    """GET /htm/htm_labreports — all uploaded lab reports for the organisation."""
+    return client.get("/htm/htm_labreports")

cli_anything_drms-0.1.0/cli_anything/drms/drms_feature/recommend.py

@@ -0,0 +1,14 @@
+"""HTM ML-based drug/product recommendation operations."""
+
+from typing import Any
+from cli_anything.drms.core.client import DrmsClient
+
+
+def get_recommendations(client: DrmsClient, query_items: str, top_n: int = 3) -> Any:
+    """GET /htm/recommendations — ML-based product/drug recommendations.
+
+    Args:
+        query_items: Comma-separated product/drug names or codes.
+        top_n: Number of recommendations to return per item.
+    """
+    return client.get("/htm/recommendations", params={"query_items": query_items, "top_n": top_n})

cli_anything_drms-0.1.0/cli_anything/drms/utils/auth.py

@@ -0,0 +1,42 @@
+"""Backend auth — calls /v1/auth/* endpoints instead of Firebase directly."""
+
+import requests
+
+
+def login(base_url: str, email: str, password: str) -> dict:
+    """POST /v1/auth/login → {firebase_token, refresh_token, user_id, email, display_name}"""
+    resp = requests.post(
+        f"{base_url}/v1/auth/login",
+        json={"email": email, "password": password, "origin": "drms-cli"},
+        timeout=15,
+    )
+    if resp.status_code == 401:
+        detail = resp.json().get("detail", "Authentication failed.")
+        raise ValueError(detail)
+    resp.raise_for_status()
+    return resp.json()
+
+
+def refresh(base_url: str, refresh_token: str) -> dict:
+    """POST /v1/auth/refresh → {firebase_token, refresh_token}"""
+    resp = requests.post(
+        f"{base_url}/v1/auth/refresh",
+        json={"refresh_token": refresh_token},
+        timeout=15,
+    )
+    if resp.status_code in (400, 401):
+        raise ValueError("Refresh token expired — please log in again.")
+    resp.raise_for_status()
+    return resp.json()
+
+
+def logout_server(base_url: str, token: str) -> None:
+    """POST /v1/auth/logout — revokes all refresh tokens server-side."""
+    try:
+        requests.post(
+            f"{base_url}/v1/auth/logout",
+            headers={"Authorization": f"Bearer {token}"},
+            timeout=10,
+        )
+    except Exception:
+        pass

cli_anything_drms-0.1.0/cli_anything/drms/utils/repl_skin.py

@@ -0,0 +1,68 @@
+"""Unified REPL skin for the DRMS CLI harness."""
+
+from typing import Optional
+import click
+
+
+BANNER = """\
+╔══════════════════════════════════════════╗
+║       DRMS / HTM CLI  •  type 'help'    ║
+╚══════════════════════════════════════════╝
+"""
+
+PROMPT = "drms> "
+
+
+class ReplSkin:
+    """Interactive REPL that delegates lines to a Click group."""
+
+    def __init__(self, cli_group: click.Group, json_mode: bool = False):
+        self.cli = cli_group
+        self.json_mode = json_mode
+        self._history: list[str] = []
+
+    def push_history(self, line: str) -> None:
+        self._history.append(line)
+
+    def undo(self) -> Optional[str]:
+        return self._history.pop() if self._history else None
+
+    def _dispatch(self, line: str) -> None:
+        args = line.strip().split()
+        if not args:
+            return
+        if self.json_mode and "--json" not in args:
+            args.append("--json")
+        try:
+            self.cli.main(args=args, standalone_mode=False)
+        except click.UsageError as e:
+            click.echo(f"Error: {e}", err=True)
+        except SystemExit:
+            pass
+        except Exception as e:
+            click.echo(f"Error: {e}", err=True)
+
+    def run(self) -> None:
+        click.echo(BANNER)
+        while True:
+            try:
+                line = input(PROMPT).strip()
+            except (EOFError, KeyboardInterrupt):
+                click.echo("\nBye.")
+                break
+
+            if not line:
+                continue
+            if line in ("exit", "quit", "q"):
+                click.echo("Bye.")
+                break
+            if line == "undo":
+                last = self.undo()
+                click.echo(f"Removed from history: {last}" if last else "Nothing to undo.")
+                continue
+            if line in ("help", "?"):
+                self._dispatch("--help")
+                continue
+
+            self.push_history(line)
+            self._dispatch(line)

cli_anything_drms-0.1.0/cli_anything/drms/utils/session.py

@@ -0,0 +1,108 @@
+"""
+Session management: persist credentials to ~/.drms_session.json.
+Locked writes prevent concurrent JSON corruption.
+Auto-refreshes the token when it is older than 55 minutes
+by calling the backend /v1/auth/refresh endpoint.
+"""
+
+import json
+import fcntl
+import os
+import time
+from pathlib import Path
+
+SESSION_FILE = Path.home() / ".drms_session.json"
+
+_TOKEN_TTL = 55 * 60  # refresh 5 min before Firebase's 1-hour expiry
+
+
+def _load_raw() -> dict:
+    if not SESSION_FILE.exists():
+        return {}
+    try:
+        with open(SESSION_FILE, "r") as f:
+            return json.load(f)
+    except (json.JSONDecodeError, OSError):
+        return {}
+
+
+def _save_raw(data: dict) -> None:
+    tmp = SESSION_FILE.with_suffix(".tmp")
+    fd = os.open(tmp, os.O_WRONLY | os.O_CREAT | os.O_TRUNC, 0o600)
+    with os.fdopen(fd, "w") as f:
+        fcntl.flock(f, fcntl.LOCK_EX)
+        json.dump(data, f, indent=2)
+        fcntl.flock(f, fcntl.LOCK_UN)
+    tmp.replace(SESSION_FILE)
+
+
+def get_session() -> dict:
+    return _load_raw()
+
+
+def save_session(
+    base_url: str,
+    token: str,
+    refresh_token: str,
+    local_id: str,
+    email: str,
+) -> None:
+    data = _load_raw()
+    data.update(
+        base_url=base_url.rstrip("/"),
+        token=token,
+        refresh_token=refresh_token,
+        local_id=local_id,
+        email=email,
+        issued_at=time.time(),
+    )
+    _save_raw(data)
+
+
+def save_token(base_url: str, token: str) -> None:
+    """Agent / CI mode: store only base_url + token, no auto-refresh."""
+    data = _load_raw()
+    data["base_url"] = base_url.rstrip("/")
+    data["token"] = token
+    data.pop("issued_at", None)
+    _save_raw(data)
+
+
+def clear_session() -> None:
+    if SESSION_FILE.exists():
+        SESSION_FILE.unlink()
+
+
+def require_token() -> tuple[str, str]:
+    """Return (base_url, token). Auto-refreshes via backend if token is near expiry."""
+    data = _load_raw()
+    base_url = data.get("base_url")
+    token = data.get("token")
+
+    if not base_url or not token:
+        raise RuntimeError("Not logged in. Run: drms-agent login")
+
+    issued_at = data.get("issued_at")
+    refresh_token = data.get("refresh_token")
+
+    if issued_at and refresh_token and (time.time() - issued_at) > _TOKEN_TTL:
+        try:
+            from cli_anything.drms.utils.auth import refresh as backend_refresh
+            new = backend_refresh(base_url, refresh_token)
+            token = new["firebase_token"]
+            data["token"] = token
+            data["refresh_token"] = new["refresh_token"]
+            data["issued_at"] = time.time()
+            _save_raw(data)
+        except Exception:
+            raise RuntimeError(
+                "Session expired and token refresh failed. Run: drms-agent login"
+            )
+
+    return base_url, token
+
+
+def update_session_key(key: str, value) -> None:
+    data = _load_raw()
+    data[key] = value
+    _save_raw(data)

cli_anything_drms-0.1.0/cli_anything_drms.egg-info/PKG-INFO

@@ -0,0 +1,10 @@
+Metadata-Version: 2.4
+Name: cli-anything-drms
+Version: 0.1.0
+Summary: CLI-Anything harness for the DRMS / HTM FastAPI backend
+Requires-Python: >=3.10
+Requires-Dist: click>=8.0
+Requires-Dist: requests>=2.28
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary

cli_anything_drms-0.1.0/cli_anything_drms.egg-info/SOURCES.txt

@@ -0,0 +1,22 @@
+setup.py
+cli_anything/drms/__main__.py
+cli_anything/drms/drms_cli.py
+cli_anything/drms/core/case_notes.py
+cli_anything/drms/core/client.py
+cli_anything/drms/core/lab_reports.py
+cli_anything/drms/core/medication.py
+cli_anything/drms/core/patients.py
+cli_anything/drms/core/staff.py
+cli_anything/drms/core/stats.py
+cli_anything/drms/drms_feature/discovery.py
+cli_anything/drms/drms_feature/member.py
+cli_anything/drms/drms_feature/recommend.py
+cli_anything/drms/utils/auth.py
+cli_anything/drms/utils/repl_skin.py
+cli_anything/drms/utils/session.py
+cli_anything_drms.egg-info/PKG-INFO
+cli_anything_drms.egg-info/SOURCES.txt
+cli_anything_drms.egg-info/dependency_links.txt
+cli_anything_drms.egg-info/entry_points.txt
+cli_anything_drms.egg-info/requires.txt
+cli_anything_drms.egg-info/top_level.txt

cli_anything_drms-0.1.0/cli_anything_drms.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

cli_anything_drms-0.1.0/cli_anything_drms.egg-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+drms = cli_anything.drms.drms_cli:cli
+drms-agent = cli_anything.drms.drms_cli:cli

cli_anything_drms-0.1.0/cli_anything_drms.egg-info/requires.txt

@@ -0,0 +1,2 @@
+click>=8.0
+requests>=2.28

cli_anything_drms-0.1.0/cli_anything_drms.egg-info/top_level.txt

@@ -0,0 +1 @@
+cli_anything

cli_anything_drms-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

cli_anything_drms-0.1.0/setup.py

@@ -0,0 +1,19 @@
+from setuptools import setup, find_namespace_packages
+
+setup(
+    name="cli-anything-drms",
+    version="0.1.0",
+    description="CLI-Anything harness for the DRMS / HTM FastAPI backend",
+    packages=find_namespace_packages(include=["cli_anything.*"]),
+    install_requires=[
+        "click>=8.0",
+        "requests>=2.28",
+    ],
+    entry_points={
+        "console_scripts": [
+            "drms-agent=cli_anything.drms.drms_cli:cli",
+            "drms=cli_anything.drms.drms_cli:cli",
+        ]
+    },
+    python_requires=">=3.10",
+)