dclient aliases command, closes #2

dclient/cli.py

@@ -1,6 +1,12 @@
 import click
 import httpx
 import json
+import appdirs
+import pathlib
+
+
+def get_config_dir():
+    return pathlib.Path(appdirs.user_config_dir("dclient"))
 
 
 @click.group()
@@ -18,6 +24,10 @@ def query(url, sql):
 
     Returns a JSON array of objects
     """
+    aliases_file = get_config_dir() / "aliases.json"
+    aliases = _load_aliases(aliases_file)
+    if url in aliases:
+        url = aliases[url]
     if not url.endswith(".json"):
         url += ".json"
     response = httpx.get(url, params={"sql": sql, "_shape": "objects"})
@@ -31,3 +41,38 @@ def query(url, sql):
         raise click.ClickException(": ".join(bits))
     else:
         click.echo(json.dumps(response.json()["rows"], indent=2))
+
+
+@cli.group()
+def alias():
+    "Manage aliases for different instances"
+
+
+@alias.command(name="list")
+def list_():
+    "List aliases"
+    aliases_file = get_config_dir() / "aliases.json"
+    aliases = _load_aliases(aliases_file)
+    for alias, url in aliases.items():
+        click.echo(f"{alias} = {url}")
+
+
+@alias.command()
+@click.argument("name")
+@click.argument("url")
+def add(name, url):
+    "Add an alias"
+    config_dir = get_config_dir()
+    config_dir.mkdir(parents=True, exist_ok=True)
+    aliases_file = config_dir / "aliases.json"
+    aliases = _load_aliases(aliases_file)
+    aliases[name] = url
+    aliases_file.write_text(json.dumps(aliases, indent=4))
+
+
+def _load_aliases(aliases_file):
+    if aliases_file.exists():
+        aliases = json.loads(aliases_file.read_text())
+    else:
+        aliases = {}
+    return aliases