Skip to content

Commit d5f6434

Browse files
Refactor CLI structure and update imports
- Updated the entry point for the `auralogger` CLI in `pyproject.toml` to reflect new module structure. - Refactored imports in `__init__.py` and `client_log.py` to align with the new CLI command organization. - Removed deprecated files related to CLI commands and authentication to streamline the codebase. - Enhanced modularity by moving command implementations to the `cli` namespace.
1 parent 6a4b237 commit d5f6434

35 files changed

Lines changed: 125 additions & 112 deletions

auralogger/__init__.py

Lines changed: 17 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -1,35 +1,34 @@
11
"""Public API barrel (mirrors node/src/index.ts)."""
22

3-
from auralogger.aura_log import AuraServer, aura_log, close_aura_log_socket
4-
from auralogger.backend_origin import (
5-
build_create_browser_logs_url,
6-
build_proj_auth_url,
7-
build_project_logs_url,
8-
)
93
from auralogger.client.client_log import (
104
AuraClient,
115
ClientLogInputs,
126
auralog,
137
client_log,
148
close_client_log_socket,
159
)
16-
from auralogger.commands.client_check import run_client_check
17-
from auralogger.commands.init import run_init
18-
from auralogger.commands.server_check import run_server_check
19-
from auralogger.commands.test_clientlog import run_test_clientlog
20-
from auralogger.commands.test_serverlog import run_test_serverlog
21-
from auralogger.get_logs import run_get_logs
22-
from auralogger.get_logs_filters import ApiLogFilter, normalize_and_validate_filters
23-
from auralogger.http_utils import parse_error_body
24-
from auralogger.log import log
25-
from auralogger.log_styles import (
10+
from auralogger.cli.commands.client_check import run_client_check
11+
from auralogger.cli.commands.init import run_init
12+
from auralogger.cli.commands.server_check import run_server_check
13+
from auralogger.cli.commands.test_clientlog import run_test_clientlog
14+
from auralogger.cli.commands.test_serverlog import run_test_serverlog
15+
from auralogger.cli.get_logs import run_get_logs
16+
from auralogger.cli.get_logs_filters import ApiLogFilter, normalize_and_validate_filters
17+
from auralogger.cli.log_styles import (
2618
DEFAULT_LOG_STYLE_SPEC,
2719
build_style_entries_from_api,
2820
resolve_log_style_spec,
2921
style_map_from_config_entries,
3022
)
31-
from auralogger.parser import ParsedFilter, ParsedGetLogsCommand, parse_command
32-
from auralogger.proj_auth import fetch_proj_auth_config, fetch_proj_auth_payload
23+
from auralogger.server.aura_log import AuraServer, aura_log, close_aura_log_socket
24+
from auralogger.server.proj_auth import fetch_proj_auth_config, fetch_proj_auth_payload
25+
from auralogger.utils.backend_origin import (
26+
build_create_browser_logs_url,
27+
build_proj_auth_url,
28+
build_project_logs_url,
29+
)
30+
from auralogger.utils.http_utils import parse_error_body
31+
from auralogger.utils.parser import ParsedFilter, ParsedGetLogsCommand, parse_command
3332

3433
__all__ = [
3534
"aura_log",
@@ -40,7 +39,6 @@
4039
"auralog",
4140
"client_log",
4241
"close_client_log_socket",
43-
"log",
4442
"run_get_logs",
4543
"normalize_and_validate_filters",
4644
"ApiLogFilter",

auralogger/cli/__init__.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
"""CLI package. Entrypoint: ``auralogger.cli.cli`` (``main``, ``print_usage``)."""
2+
3+
from __future__ import annotations
4+
5+
from typing import Any
6+
7+
__all__ = ["main", "print_usage"]
8+
9+
10+
def __getattr__(name: str) -> Any:
11+
if name == "main":
12+
from auralogger.cli.cli import main
13+
14+
return main
15+
if name == "print_usage":
16+
from auralogger.cli.cli import print_usage
17+
18+
return print_usage
19+
raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -3,13 +3,13 @@
33
import sys
44
from typing import TextIO
55

6-
from auralogger.cli_load_env import load_cli_env_files
7-
from auralogger.commands.client_check import run_client_check
8-
from auralogger.commands.get_logs_cmd import run_get_logs_command
9-
from auralogger.commands.init import run_init
10-
from auralogger.commands.server_check import run_server_check
11-
from auralogger.commands.test_clientlog import run_test_clientlog
12-
from auralogger.commands.test_serverlog import run_test_serverlog
6+
from auralogger.cli.cli_load_env import load_cli_env_files
7+
from auralogger.cli.commands.client_check import run_client_check
8+
from auralogger.cli.commands.get_logs_cmd import run_get_logs_command
9+
from auralogger.cli.commands.init import run_init
10+
from auralogger.cli.commands.server_check import run_server_check
11+
from auralogger.cli.commands.test_clientlog import run_test_clientlog
12+
from auralogger.cli.commands.test_serverlog import run_test_serverlog
1313

1414

1515
def print_usage(stream: TextIO = sys.stdout) -> None:
Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,13 +5,13 @@
55
from dataclasses import dataclass
66
from typing import Any, Dict, cast
77

8-
from auralogger.env_config import ENV_PROJECT_TOKEN
9-
from auralogger.env_config import (
8+
from auralogger.server.proj_auth import fetch_proj_auth_payload
9+
from auralogger.utils.env_config import ENV_PROJECT_TOKEN
10+
from auralogger.utils.env_config import (
1011
ENV_USER_SECRET,
1112
get_resolved_project_token,
1213
get_resolved_user_secret,
1314
)
14-
from auralogger.proj_auth import fetch_proj_auth_payload
1515

1616

1717
@dataclass(frozen=True)
Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,11 +9,11 @@
99
import websocket
1010
from websocket import create_connection
1111

12-
from auralogger.backend_origin import (
12+
from auralogger.cli.cli_auth import resolve_project_context_for_cli_checks
13+
from auralogger.utils.backend_origin import (
1314
build_create_browser_logs_url,
1415
resolve_ws_base_url,
1516
)
16-
from auralogger.cli_auth import resolve_project_context_for_cli_checks
1717

1818
CONNECT_TIMEOUT_S = 5
1919

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
from typing import List
44

5-
from auralogger.get_logs import run_get_logs
5+
from auralogger.cli.get_logs import run_get_logs
66

77

88
def run_get_logs_command(argv: List[str]) -> None:
Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,10 @@
55
import os
66
from typing import Any, Dict, List, cast
77

8-
from auralogger.cli_auth import resolve_project_token_for_init, resolve_user_secret_for_init
9-
from auralogger.env_config import (
8+
from auralogger.cli.cli_auth import resolve_project_token_for_init, resolve_user_secret_for_init
9+
from auralogger.cli.log_styles import build_style_entries_from_api
10+
from auralogger.server.proj_auth import fetch_proj_auth_payload
11+
from auralogger.utils.env_config import (
1012
ENV_NEXT_PUBLIC_PROJECT_TOKEN,
1113
ENV_PROJECT_SESSION,
1214
ENV_PROJECT_TOKEN,
@@ -17,8 +19,6 @@
1719
get_resolved_session,
1820
is_full_runtime_env_configured,
1921
)
20-
from auralogger.log_styles import build_style_entries_from_api
21-
from auralogger.proj_auth import fetch_proj_auth_payload
2222

2323

2424
def _user_secret_explicitly_in_env() -> bool:
Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,8 @@
99
import websocket
1010
from websocket import create_connection
1111

12-
from auralogger.backend_origin import resolve_ws_base_url
13-
from auralogger.cli_auth import resolve_project_context_for_cli_checks
12+
from auralogger.cli.cli_auth import resolve_project_context_for_cli_checks
13+
from auralogger.utils.backend_origin import resolve_ws_base_url
1414

1515
CONNECT_TIMEOUT_S = 5
1616

0 commit comments

Comments
 (0)