snapshot of https://github.com/ActivityWatch/aw-core.git (master, 2026-09-16; blobs > 131072B omitted)

aw_cli/__main__.py

@@ -0,0 +1,107 @@
+"""
+The idea behind this `aw` or `aw-cli` wrapper script is to act as a collection of helper tools,
+and perhaps even as a way to list and run ActivityWatch modules on a system (a bit like aw-qt, but without the GUI).
+"""
+
+from pathlib import Path
+from datetime import datetime
+import os
+import subprocess
+
+import click
+
+from aw_cli.log import find_oldest_log, print_log, LOGLEVELS
+from typing import Optional
+
+
+def _resolve_profile(profile: Optional[str], testing: bool) -> str:
+    """Return the active profile name, resolving --testing as an alias for 'testing'."""
+    if profile:
+        return profile
+    if testing:
+        return "testing"
+    return ""
+
+
+@click.group()
+@click.option(
+    "--profile",
+    default=None,
+    help="Named instance profile (e.g. 'testing', 'research').",
+)
+@click.option("--testing", is_flag=True, help="Alias for --profile testing.")
+def main(profile: Optional[str] = None, testing: bool = False):
+    active = _resolve_profile(profile, testing)
+    if active:
+        os.environ["AW_PROFILE"] = active
+    # If no explicit flag, preserve any inherited AW_PROFILE from the environment.
+
+
+@main.command()
+@click.pass_context
+def qt(ctx):
+    profile = ctx.parent.params.get("profile")
+    testing = ctx.parent.params.get("testing")
+    active = _resolve_profile(profile, testing)
+    args = ["aw-qt"]
+    if active:
+        args += ["--profile", active]
+    return subprocess.call(args)
+
+
+@main.command()
+def directories():
+    # Print all directories (respects AW_PROFILE set by the group callback)
+    from aw_core.dirs import get_data_dir, get_config_dir, get_cache_dir, get_log_dir
+
+    print("Directory paths used")
+    print(" - config: ", get_config_dir(None))
+    print(" - data:   ", get_data_dir(None))
+    print(" - logs:   ", get_log_dir(None))
+    print(" - cache:  ", get_cache_dir(None))
+
+
+@main.command()
+@click.pass_context
+@click.argument("module_name", type=str, required=False)
+@click.option(
+    "--since",
+    type=click.DateTime(formats=["%Y-%m-%d"]),
+    help="Only show logs since this date",
+)
+@click.option(
+    "--level",
+    type=click.Choice(LOGLEVELS),
+    help="Only show logs of this level, or higher.",
+)
+def logs(
+    ctx,
+    module_name: Optional[str] = None,
+    since: Optional[datetime] = None,
+    level: Optional[str] = None,
+):
+    from aw_core.dirs import get_log_dir
+
+    # AW_PROFILE was set by the group callback, so get_log_dir returns the profile-specific dir.
+    logdir: Path = Path(get_log_dir(None))
+
+    if module_name:
+        _print_newest_log(logdir / module_name, since, level)
+    else:
+        for subdir in sorted(logdir.iterdir()):
+            if subdir.is_dir():
+                _print_newest_log(subdir, since, level)
+
+
+def _print_newest_log(
+    path: Path, since: Optional[datetime], level: Optional[str]
+) -> None:
+    logfile = find_oldest_log(path)
+    if logfile:
+        print_log(logfile, since, level)
+    else:
+        print(f"No logfile found in {path}")
+
+
+if __name__ == "__main__":
+    main()