awcs-scripts 1.0.2__tar.gz → 1.1.0__tar.gz

{awcs_scripts-1.0.2 → awcs_scripts-1.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: awcs-scripts
-Version: 1.0.2
+Version: 1.1.0
 Summary: Command line scripts for WCS maps and the ARMS API
 Project-URL: Repository, https://gitlab.sprydev.ru/automacon/agv/wcs/awcs_scripts
 Author: AWCS
@@ -43,7 +43,7 @@ uv tool install .
 
 ```
 uvx --from awcs-scripts awcs help
-uvx --from awcs-scripts awcs mrmap-to-raster path/to/map.mrmap --resolution 0.05 --out-dir path/to/out
+uvx --from awcs-scripts awcs mrmap-to-raster path/to/map.mrmap [--resolution 0.05] [--out-dir path/to/out]
 uvx --from awcs-scripts awcs users me
 ```
 
@@ -59,8 +59,8 @@ awcs version
 #### Карты
 
 ```
-awcs mrmap-to-raster path/to/map.mrmap --out-dir path/to/out
-awcs mrmap-to-bg path/to/map.mrmap path/to/layout.lif --layout_id layoutId --resolution 0.05 --output path/to/layout.out.lif
+awcs mrmap-to-raster path/to/map.mrmap [--out-dir path/to/out]
+awcs mrmap-to-bg path/to/map.mrmap path/to/layout.lif --layout_id layoutId [--resolution 0.05] [--output path/to/layout.out.lif]
 ```
 
 `awcs mrmap-to-raster` пишет PNG, GeoTIFF и JSON. `--resolution` задаёт метры на пиксель, по умолчанию берётся `resolution_2d` из карты. `awcs mrmap-to-bg` подставляет этот растр в `backgroundImage` одного layout. Без `--output` исходный `.lif` перезаписывается.
@@ -74,16 +74,19 @@ awcs users me
 awcs mission_templates list
 awcs robots list
 awcs missions list
-awcs missions create --template_code PickPallet --params "{\"zone\":\"A\"}"
-awcs mission_templates execute PickPallet --params "{\"zone\":\"A\"}"
+awcs missions create --template_code PickPallet [--params "{\"zone\":\"A\"}"]
+awcs mission_templates execute PickPallet [--params "{\"zone\":\"A\"}"]
 awcs projects list
-awcs projects create --lif_json path/to/layout.lif -m MAIRUI -v 1.0.0 --desc "line 1" --sync
+awcs projects create --lif_json path/to/layout.lif -m MAIRUI -v 1.0.0 [--desc "line 1"] [--sync]
+awcs projects upload --lif_json path/to/layout.lif -m MAIRUI -v 1.0.0 [--desc "line 1"] [--sync]
+awcs projects download
+awcs projects download 12 -o path/to/layout.lif
 awcs projects activate 12
 ```
 
 `awcs login` проверяет ключ через `GET /users/me` и только после ответа 200 сохраняет origin и ключ. Профиль лежит вне репозитория: путь из `AWCS_CONFIG`, иначе `%APPDATA%\awcs\config.ini` на Windows и `~/.config/awcs/config.ini` в остальных системах. Ключ в вывод не печатается. Клиент сам дописывает `/api/v1` и шлёт заголовок `X-Api-Key`.
 
-У списков есть `--page`, `--per_page`, `--filter`, `--sort_by`. `awcs missions create` принимает `--template_id` или `--template_code` и необязательный `--params`. `awcs mission_templates execute` отправляет `--params` плоским JSON-объектом. `awcs projects create` загружает LIF как `multipart/form-data`: `-m` это `--manufacturer`, `-v` и `--version` это `--project_version`, `--desc` это `--description`, `--sync` это `--synchronous`.
+У списков есть `--page`, `--per_page`, `--filter`, `--sort_by`. `awcs missions create` принимает `--template_id` или `--template_code` и необязательный `--params`. `awcs mission_templates execute` отправляет `--params` плоским JSON-объектом. `awcs projects create` загружает LIF как `multipart/form-data`: `-m` это `--manufacturer`, `-v` и `--version` это `--project_version`, `--desc` это `--description`, `--sync` это `--synchronous`. `awcs projects upload` делает то же самое. `awcs projects download` без id забирает активный проект (`GET /projects/active`), с id — этот проект, и сохраняет исходный LIF. `-o` задаёт файл или существующую папку; без него файл пишется в текущий каталог под именем вложения. Путь печатается в stdout.
 
 ### Удаление
 

{awcs_scripts-1.0.2 → awcs_scripts-1.1.0}/README.md

@@ -28,7 +28,7 @@ uv tool install .
 
 ```
 uvx --from awcs-scripts awcs help
-uvx --from awcs-scripts awcs mrmap-to-raster path/to/map.mrmap --resolution 0.05 --out-dir path/to/out
+uvx --from awcs-scripts awcs mrmap-to-raster path/to/map.mrmap [--resolution 0.05] [--out-dir path/to/out]
 uvx --from awcs-scripts awcs users me
 ```
 
@@ -44,8 +44,8 @@ awcs version
 #### Карты
 
 ```
-awcs mrmap-to-raster path/to/map.mrmap --out-dir path/to/out
-awcs mrmap-to-bg path/to/map.mrmap path/to/layout.lif --layout_id layoutId --resolution 0.05 --output path/to/layout.out.lif
+awcs mrmap-to-raster path/to/map.mrmap [--out-dir path/to/out]
+awcs mrmap-to-bg path/to/map.mrmap path/to/layout.lif --layout_id layoutId [--resolution 0.05] [--output path/to/layout.out.lif]
 ```
 
 `awcs mrmap-to-raster` пишет PNG, GeoTIFF и JSON. `--resolution` задаёт метры на пиксель, по умолчанию берётся `resolution_2d` из карты. `awcs mrmap-to-bg` подставляет этот растр в `backgroundImage` одного layout. Без `--output` исходный `.lif` перезаписывается.
@@ -59,16 +59,19 @@ awcs users me
 awcs mission_templates list
 awcs robots list
 awcs missions list
-awcs missions create --template_code PickPallet --params "{\"zone\":\"A\"}"
-awcs mission_templates execute PickPallet --params "{\"zone\":\"A\"}"
+awcs missions create --template_code PickPallet [--params "{\"zone\":\"A\"}"]
+awcs mission_templates execute PickPallet [--params "{\"zone\":\"A\"}"]
 awcs projects list
-awcs projects create --lif_json path/to/layout.lif -m MAIRUI -v 1.0.0 --desc "line 1" --sync
+awcs projects create --lif_json path/to/layout.lif -m MAIRUI -v 1.0.0 [--desc "line 1"] [--sync]
+awcs projects upload --lif_json path/to/layout.lif -m MAIRUI -v 1.0.0 [--desc "line 1"] [--sync]
+awcs projects download
+awcs projects download 12 -o path/to/layout.lif
 awcs projects activate 12
 ```
 
 `awcs login` проверяет ключ через `GET /users/me` и только после ответа 200 сохраняет origin и ключ. Профиль лежит вне репозитория: путь из `AWCS_CONFIG`, иначе `%APPDATA%\awcs\config.ini` на Windows и `~/.config/awcs/config.ini` в остальных системах. Ключ в вывод не печатается. Клиент сам дописывает `/api/v1` и шлёт заголовок `X-Api-Key`.
 
-У списков есть `--page`, `--per_page`, `--filter`, `--sort_by`. `awcs missions create` принимает `--template_id` или `--template_code` и необязательный `--params`. `awcs mission_templates execute` отправляет `--params` плоским JSON-объектом. `awcs projects create` загружает LIF как `multipart/form-data`: `-m` это `--manufacturer`, `-v` и `--version` это `--project_version`, `--desc` это `--description`, `--sync` это `--synchronous`.
+У списков есть `--page`, `--per_page`, `--filter`, `--sort_by`. `awcs missions create` принимает `--template_id` или `--template_code` и необязательный `--params`. `awcs mission_templates execute` отправляет `--params` плоским JSON-объектом. `awcs projects create` загружает LIF как `multipart/form-data`: `-m` это `--manufacturer`, `-v` и `--version` это `--project_version`, `--desc` это `--description`, `--sync` это `--synchronous`. `awcs projects upload` делает то же самое. `awcs projects download` без id забирает активный проект (`GET /projects/active`), с id — этот проект, и сохраняет исходный LIF. `-o` задаёт файл или существующую папку; без него файл пишется в текущий каталог под именем вложения. Путь печатается в stdout.
 
 ### Удаление
 

{awcs_scripts-1.0.2 → awcs_scripts-1.1.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "awcs-scripts"
-version = "1.0.2"
+version = "1.1.0"
 description = "Command line scripts for WCS maps and the ARMS API"
 readme = "README.md"
 requires-python = ">=3.10"

{awcs_scripts-1.0.2 → awcs_scripts-1.1.0}/src/awcs_scripts/__init__.py

@@ -1,3 +1,3 @@
 """Scripts for WCS maps and the ARMS API."""
 
-__version__ = "1.0.2"
+__version__ = "1.1.0"

{awcs_scripts-1.0.2 → awcs_scripts-1.1.0}/src/awcs_scripts/api.py

@@ -24,6 +24,13 @@ def origin_of(url: str) -> str:
     return value
 
 
+def _strip_api_prefix(path: str) -> str:
+    """Drop a leading /api/v1 so the client base URL does not double it."""
+    if path.startswith(_API_PREFIX + "/"):
+        return path[len(_API_PREFIX) :]
+    return path
+
+
 class Client:
     def __init__(
         self,
@@ -64,3 +71,30 @@ class Client:
         if response.status_code >= 400:
             raise ApiError(response.status_code, response.text)
         return response
+
+    def get(self, url_or_path: str) -> httpx.Response:
+        """GET an API path or an absolute URL.
+
+        A leading /api/v1 is removed so the client base URL does not double it.
+        An absolute URL on this origin is sent with the API key. An absolute URL
+        on another host is fetched without the key.
+        """
+        if url_or_path.startswith(("http://", "https://")):
+            return self._get_absolute(url_or_path)
+        return self.request("GET", _strip_api_prefix(url_or_path))
+
+    def _get_absolute(self, url: str) -> httpx.Response:
+        target = httpx.URL(url)
+        origin = httpx.URL(self.origin)
+        same_origin = (
+            target.scheme == origin.scheme
+            and target.host == origin.host
+            and target.port == origin.port
+        )
+        if not same_origin:
+            response = httpx.get(url, timeout=self._client.timeout, follow_redirects=True)
+            if response.status_code >= 400:
+                raise ApiError(response.status_code, response.text)
+            return response
+        params = dict(target.params) or None
+        return self.request("GET", _strip_api_prefix(target.path), params=params)

{awcs_scripts-1.0.2 → awcs_scripts-1.1.0}/src/awcs_scripts/cli.py

@@ -158,6 +158,45 @@ def _cmd_projects_activate(args: argparse.Namespace) -> None:
         _emit(client.request("POST", f"/projects/{args.id}/activate"))
 
 
+def _download_name(source: dict[str, Any], project_id: object) -> str:
+    raw = str(source.get("filename") or "").replace("\\", "/")
+    name = Path(raw).name
+    if name and name not in {".", ".."}:
+        return name
+    return f"project-{project_id}.lif"
+
+
+def _download_destination(output: Path | None, filename: str) -> Path:
+    if output is None:
+        return Path(filename)
+    output = output.expanduser()
+    if output.is_dir():
+        return output / filename
+    return output
+
+
+def _cmd_projects_download(args: argparse.Namespace) -> None:
+    with _open_client() as client:
+        if args.id is None:
+            project = client.request("GET", "/projects/active").json()
+        else:
+            project = client.request("GET", f"/projects/{args.id}").json()
+        source = project.get("source") if isinstance(project, dict) else None
+        if not isinstance(source, dict):
+            source = {}
+        download_url = source.get("download_url")
+        if not download_url:
+            attachment_id = source.get("id")
+            if not attachment_id:
+                _fail("project has no map file")
+            download_url = f"/attachments/{attachment_id}/download"
+        response = client.get(str(download_url))
+    project_id = project.get("id", args.id) if isinstance(project, dict) else args.id
+    dest = _download_destination(args.output, _download_name(source, project_id))
+    dest.write_bytes(response.content)
+    print(dest)
+
+
 _TOOL_COMMANDS = {
     "mrmap-to-raster": "mrmap-to-raster.py",
     "mrmap-to-bg": "mrmap-to-bg.py",
@@ -229,13 +268,24 @@ def build_parser() -> argparse.ArgumentParser:
     projects_list = projects_sub.add_parser("list", help="GET /projects")
     _add_list_flags(projects_list)
     projects_list.set_defaults(func=_cmd_projects_list)
-    projects_create = projects_sub.add_parser("create", help="POST /projects with a LIF file")
+    projects_create = projects_sub.add_parser(
+        "create",
+        aliases=["upload"],
+        help="POST /projects with a LIF file (alias: upload)",
+    )
     projects_create.add_argument("--lif_json", type=Path, required=True, help="Path to a .lif file")
     projects_create.add_argument("--manufacturer", "-m", required=True)
     projects_create.add_argument("--project_version", "--version", "-v", required=True, dest="project_version")
     projects_create.add_argument("--description", "--desc", dest="description")
     projects_create.add_argument("--synchronous", "--sync", action="store_true", dest="synchronous")
     projects_create.set_defaults(func=_cmd_projects_create)
+    projects_download = projects_sub.add_parser(
+        "download",
+        help="Save a project LIF file; the active project when id is omitted",
+    )
+    projects_download.add_argument("id", type=int, nargs="?", help="Project id; omit to download the active project")
+    projects_download.add_argument("--output", "-o", type=Path, help="Output file, or an existing directory")
+    projects_download.set_defaults(func=_cmd_projects_download)
     projects_activate = projects_sub.add_parser("activate", help="POST /projects/{id}/activate")
     projects_activate.add_argument("id", type=int)
     projects_activate.set_defaults(func=_cmd_projects_activate)

{awcs_scripts-1.0.2 → awcs_scripts-1.1.0}/tests/test_api.py

@@ -7,6 +7,8 @@ from awcs_scripts.api import Client, origin_of
 from awcs_scripts.cli import main
 from awcs_scripts.config import load_profile
 
+_LIF = b'{"layouts":[]}'
+
 
 def test_origin_strips_api_prefix():
     assert origin_of("https://dev.wcs.automacon.ru/") == "https://dev.wcs.automacon.ru"
@@ -100,7 +102,8 @@ def test_template_execute_body(tmp_path, monkeypatch):
     assert seen["body"] == {"zone": "A"}
 
 
-def test_projects_create_and_activate(tmp_path, monkeypatch):
+@pytest.mark.parametrize("command", ["create", "upload"])
+def test_projects_create_and_activate(tmp_path, monkeypatch, command):
     _profile(tmp_path, monkeypatch)
     lif = tmp_path / "layout.lif"
     lif.write_text("{}", encoding="utf-8")
@@ -113,7 +116,7 @@ def test_projects_create_and_activate(tmp_path, monkeypatch):
     _install(monkeypatch, handler)
     main([
         "projects",
-        "create",
+        command,
         "--lif_json",
         str(lif),
         "-m",
@@ -143,6 +146,91 @@ def test_projects_create_and_activate(tmp_path, monkeypatch):
     assert seen[1].url.path == "/api/v1/projects/12/activate"
 
 
+def test_projects_download_active_uses_source_filename(tmp_path, monkeypatch, capsys):
+    _profile(tmp_path, monkeypatch)
+    monkeypatch.chdir(tmp_path)
+    seen = []
+
+    def handler(request: httpx.Request) -> httpx.Response:
+        seen.append(request.url.path)
+        if request.url.path == "/api/v1/projects/active":
+            return httpx.Response(200, json={
+                "id": 4,
+                "source": {
+                    "id": 9,
+                    "filename": "nested/line.lif",
+                    "download_url": "/api/v1/attachments/9/download",
+                },
+            })
+        assert request.headers["x-api-key"] == "secret-key"
+        return httpx.Response(200, content=_LIF)
+
+    _install(monkeypatch, handler)
+    main(["projects", "download"])
+    assert seen == ["/api/v1/projects/active", "/api/v1/attachments/9/download"]
+    assert (tmp_path / "line.lif").read_bytes() == _LIF
+    assert capsys.readouterr().out.strip() == "line.lif"
+
+
+def test_projects_download_id_to_output(tmp_path, monkeypatch, capsys):
+    _profile(tmp_path, monkeypatch)
+    dest = tmp_path / "saved.lif"
+    folder = tmp_path / "maps"
+    folder.mkdir()
+
+    def handler(request: httpx.Request) -> httpx.Response:
+        if request.url.path == "/api/v1/projects/12":
+            return httpx.Response(200, json={
+                "id": 12,
+                "source": {"id": 3, "filename": "layout.lif"},
+            })
+        assert request.url.path == "/api/v1/attachments/3/download"
+        return httpx.Response(200, content=_LIF)
+
+    _install(monkeypatch, handler)
+    main(["projects", "download", "12", "-o", str(dest)])
+    assert dest.read_bytes() == _LIF
+    assert capsys.readouterr().out.strip() == str(dest)
+
+    main(["projects", "download", "12", "-o", str(folder)])
+    assert (folder / "layout.lif").read_bytes() == _LIF
+
+
+def test_projects_download_without_source_fails(tmp_path, monkeypatch):
+    _profile(tmp_path, monkeypatch)
+    _install(monkeypatch, lambda request: httpx.Response(200, json={"id": 1, "source": None}))
+    with pytest.raises(SystemExit) as caught:
+        main(["projects", "download", "1"])
+    assert caught.value.code == 1
+
+
+def test_client_get_same_origin_keeps_query():
+    def handler(request: httpx.Request) -> httpx.Response:
+        assert request.url.path == "/api/v1/attachments/9/download"
+        assert request.url.params["x"] == "1"
+        assert request.headers["x-api-key"] == "secret-key"
+        return httpx.Response(200, content=_LIF)
+
+    with Client("https://dev.wcs.automacon.ru", "secret-key", transport=httpx.MockTransport(handler)) as client:
+        response = client.get("https://dev.wcs.automacon.ru/api/v1/attachments/9/download?x=1")
+    assert response.content == _LIF
+
+
+def test_client_get_external_omits_api_key(monkeypatch):
+    seen = {}
+
+    def fake_get(url, **kwargs):
+        seen["url"] = url
+        seen["headers"] = kwargs.get("headers")
+        return httpx.Response(200, content=_LIF)
+
+    monkeypatch.setattr("awcs_scripts.api.httpx.get", fake_get)
+    with Client("https://dev.wcs.automacon.ru", "secret-key") as client:
+        response = client.get("https://files.example/map.lif")
+    assert response.content == _LIF
+    assert seen == {"url": "https://files.example/map.lif", "headers": None}
+
+
 def _profile(tmp_path, monkeypatch):
     config = tmp_path / "config.ini"
     config.write_text("[default]\nurl = https://dev.wcs.automacon.ru\napikey = secret-key\n", encoding="utf-8")

{awcs_scripts-1.0.2 → awcs_scripts-1.1.0}/uv.lock

@@ -23,7 +23,7 @@ wheels = [
 
 [[package]]
 name = "awcs-scripts"
-version = "1.0.2"
+version = "1.1.0"
 source = { editable = "." }
 dependencies = [
     { name = "httpx" },