Add cli

Author: lsbardel

blocks/sport.yaml

@@ -0,0 +1,94 @@
+html: false
+upstream: http://trading-api.prod.svc
+tags:
+  - trading
+  - api
+routes:
+  - name: acme
+    protocols:
+      - http
+      - https
+    paths:
+      - /.well-known/acme-challenge
+    tags:
+      - public
+      - acme
+      - status
+  - name: public
+    protocols:
+      - https
+    paths:
+      - /v1/
+      - /v1/countries
+      - /v1/docs
+      - /v1/spec
+      - /v1/status
+      - /v1/_tl
+    strip_path: false
+    plugins:
+      - name: cors
+        config:
+          origins:
+            - "*"
+          methods:
+            - GET
+    tags:
+      - public
+  - name: private
+    protocols:
+      - https
+    paths:
+      - /v1/accounts
+      - /v1/alerts
+      - /v1/betting
+      - /v1/entities
+      - /v1/events
+      - /v1/gateways
+      - /v1/instruments
+      - /v1/leagues
+      - /v1/names
+      - /v1/orders
+      - /v1/risk
+      - /v1/strategies
+      - /v1/teams
+      - /v1/tasks
+      - /v1/ohlc
+    strip_path: false
+    plugins:
+      - name: key-auth
+        config:
+          key_names:
+            - x-fluidily-api-key
+          anonymous: anonymous
+      - name: jwt
+        config:
+          claims_to_verify:
+            - exp
+          run_on_preflight: false
+          anonymous: anonymous
+      - name: acl
+        config:
+          allow:
+            - service
+            - service-fluidily
+            - users-fluidily
+      - name: request-termination
+        consumer:
+          username: anonymous
+        config:
+          status_code: 403
+          message: "requires authentication"
+      - name: cors
+        config:
+          origins:
+            - "*"
+          headers:
+            - Authorization
+            - Content-Type
+            - User-Agent
+            - x-fluidily-org-id
+          exposed_headers:
+            - Link
+            - X-Total-Count
+            - X-RateLimit-Limit-minute
+            - X-RateLimit-Remaining-minute

metablock/cli.py

@@ -0,0 +1,62 @@
+import asyncio
+import os
+from pathlib import Path
+from typing import cast
+
+import click
+import yaml
+from metablock import Metablock, Space
+
+METABLOCK_SPACE = os.environ.get("METABLOCK_SPACE", "")
+METABLOCK_API_TOKEN = os.environ.get("METABLOCK_API_TOKEN", "")
+
+
+def manifest(file_path: Path) -> dict:
+    return yaml.safe_load(file_path.read_text())
+
+
+@click.group()
+def cli() -> None:
+    pass
+
+
+@cli.command()
+@click.argument("path", type=click.Path(exists=True))
+@click.option("--space", "space_name", help="Space name", default=METABLOCK_SPACE)
+@click.option("--token", help="metablock API token", default=METABLOCK_API_TOKEN)
+def apply(path: str, space_name: str, token: str) -> None:
+    """Apply metablock manifest"""
+    asyncio.get_event_loop().run_until_complete(_apply(path, space_name, token))
+
+
+async def _apply(path: str, space_name: str, token: str) -> None:
+    if not token:
+        click.echo("metablock API token is required", err=True)
+        raise click.Abort()
+    if not space_name:
+        click.echo("metablock space is required", err=True)
+        raise click.Abort()
+    p = Path(path)
+    blocks = []
+    for file_path in p.glob("*.yaml"):
+        name = file_path.name.split(".")[0]
+        blocks.append((name, manifest(file_path)))
+    async with Metablock(auth_key=token) as mb:
+        space: Space = cast(Space, await mb.spaces.get(space_name))
+        svc = await space.blocks.get_list()
+        click.echo(f"space {space.name} has {len(svc)} blocks")
+        by_name = {s["name"]: s for s in svc}
+        for name, config in blocks:
+            block = by_name.get(name)
+            if block:
+                # update
+                click.echo(f"update block {name}")
+                await mb.services.update(block.id, **config)
+            else:
+                # create
+                click.echo(f"create new block {name}")
+                await space.services.create(name=name, **config)
+
+
+if __name__ == "__main__":
+    cli()

poetry.lock

@@ -9,6 +9,8 @@ readme = "readme.md"
 [tool.poetry.dependencies]
 python = ">=3.10,<3.12"
 aiohttp = "^3.8.3"
+click = "^8.1.3"
+pyyaml = "^6.0"
 
 [tool.poetry.group.dev.dependencies]
 pytest-asyncio = "^0.21.0"
@@ -20,7 +22,11 @@ isort = "^5.11.3"
 flake8 = "^6.0.0"
 flake8-builtins = "^2.0.1"
 python-dotenv = "^1.0.0"
+types-pyyaml = "^6.0.12.10"
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]
 build-backend = "poetry.core.masonry.api"
+
+[tool.poetry.scripts]
+metablock = "metablock.cli:main"

pyproject.toml

@@ -0,0 +1,13 @@
+html: false
+upstream: http://metablock.prod.svc
+tags:
+  - test
+routes:
+  - name: main
+    protocols:
+      - http
+      - https
+    paths:
+      - /
+    tags:
+      - public

tests/blocks/test.yaml

@@ -1,37 +1,16 @@
-import pytest
-from metablock import Metablock, MetablockResponseError
+from click.testing import CliRunner
+from metablock.cli import cli
 
 
-def test_cli(cli: Metablock):
-    assert cli.url == "https://api.metablock.io/v1"
-    assert str(cli) == cli.url
+def test_cli_apply_path_error():
+    runner = CliRunner()
+    result = runner.invoke(cli, ["apply", "foo"])
+    assert result.exit_code == 2
+    assert "Invalid value for 'PATH': Path 'foo' does not exist" in result.output
 
 
-async def test_user(cli: Metablock):
-    user = await cli.get_user()
-    assert user.id
-    orgs = await user.orgs()
-    assert orgs
-
-
-async def test_user_403(cli: Metablock, invalid_headers: dict):
-    with pytest.raises(MetablockResponseError) as exc:
-        await cli.get_user(headers=invalid_headers)
-    assert exc.value.status == 403
-
-
-async def test_orgs_403(cli: Metablock, invalid_headers: dict):
-    user = await cli.get_user()
-    with pytest.raises(MetablockResponseError) as exc:
-        await user.orgs(headers=invalid_headers)
-    assert exc.value.status == 403
-
-
-async def test_space(cli: Metablock):
-    space = await cli.get_space()
-    assert space["name"] == "metablock"
-
-
-async def test_spec(cli: Metablock):
-    spec = await cli.spec()
-    assert spec
+def test_cli_apply_no_space():
+    runner = CliRunner()
+    result = runner.invoke(cli, ["apply", "tests/blocks"])
+    assert result.exit_code == 1
+    assert result.output.startswith("metablock space is required")

tests/test_cli.py

@@ -0,0 +1,36 @@
+import pytest
+from metablock import Metablock, MetablockResponseError
+
+
+def test_cli(cli: Metablock):
+    assert str(cli) == cli.url
+
+
+async def test_user(cli: Metablock):
+    user = await cli.get_user()
+    assert user.id
+    orgs = await user.orgs()
+    assert orgs
+
+
+async def test_user_403(cli: Metablock, invalid_headers: dict):
+    with pytest.raises(MetablockResponseError) as exc:
+        await cli.get_user(headers=invalid_headers)
+    assert exc.value.status == 403
+
+
+async def test_orgs_403(cli: Metablock, invalid_headers: dict):
+    user = await cli.get_user()
+    with pytest.raises(MetablockResponseError) as exc:
+        await user.orgs(headers=invalid_headers)
+    assert exc.value.status == 403
+
+
+async def test_space(cli: Metablock):
+    space = await cli.get_space()
+    assert space["name"] == "metablock"
+
+
+async def test_spec(cli: Metablock):
+    spec = await cli.spec()
+    assert spec