A R R R Associates
61 lines
3.0 KiB
Python
61 lines
3.0 KiB
Python
from __future__ import annotations
|
|
|
|
import io
|
|
from pathlib import Path
|
|
import zipfile
|
|
|
|
ERP_LOCAL_AGENT_VERSION = "1.0.0"
|
|
ERP_LOCAL_AGENT_NAME = "ERP Local Agent"
|
|
RUNTIME_ROOT = Path(__file__).resolve().parent / "local_agent_runtime"
|
|
|
|
|
|
def build_agent_env(*, erp_base_url: str, node_code: str, node_secret: str, storage_root: str, tenant_id=None, branch_id=None, sync_interval_seconds: int = 30, request_timeout_seconds: int = 60, tunnel_enabled: bool = True, tunnel_reconnect_seconds: int = 10) -> str:
|
|
erp_base_url = (erp_base_url or "").strip().rstrip("/")
|
|
if erp_base_url.startswith("http://"):
|
|
host = erp_base_url[7:].split("/", 1)[0].split(":", 1)[0].lower()
|
|
if host not in {"localhost", "127.0.0.1", "::1"}:
|
|
erp_base_url = "https://" + erp_base_url[7:]
|
|
storage_root = (storage_root or r"D:\AuditFirmStorage").strip()
|
|
return (
|
|
f"ERP_BASE_URL={erp_base_url}\n"
|
|
f"NODE_CODE={(node_code or '').strip()}\n"
|
|
f"NODE_SECRET={(node_secret or '').strip()}\n"
|
|
f"STORAGE_ROOT={storage_root}\n"
|
|
f"TENANT_ID={'' if tenant_id is None else tenant_id}\n"
|
|
f"AUDIT_FIRM_ID={'' if tenant_id is None else tenant_id}\n"
|
|
f"BRANCH_ID={'' if branch_id is None else branch_id}\n"
|
|
f"SYNC_INTERVAL_SECONDS={int(sync_interval_seconds or 30)}\n"
|
|
f"POLL_INTERVAL_SECONDS={int(sync_interval_seconds or 30)}\n"
|
|
f"REQUEST_TIMEOUT_SECONDS={int(request_timeout_seconds or 60)}\n"
|
|
f"TUNNEL_ENABLED={str(bool(tunnel_enabled)).lower()}\n"
|
|
f"TUNNEL_RECONNECT_SECONDS={int(tunnel_reconnect_seconds or 10)}\n"
|
|
"AUTO_UPDATE=true\n"
|
|
"UPDATE_CHECK_INTERVAL_SECONDS=300\n"
|
|
)
|
|
|
|
|
|
def _build_zip(*, env_text: str | None, include_env: bool, include_admin_readme: bool) -> bytes:
|
|
if not RUNTIME_ROOT.exists():
|
|
raise RuntimeError(f"ERP Local Agent runtime is missing: {RUNTIME_ROOT}")
|
|
buffer = io.BytesIO()
|
|
with zipfile.ZipFile(buffer, "w", compression=zipfile.ZIP_DEFLATED) as dst:
|
|
for path in sorted(RUNTIME_ROOT.rglob("*")):
|
|
if not path.is_file() or "__pycache__" in path.parts:
|
|
continue
|
|
name = path.relative_to(RUNTIME_ROOT).as_posix()
|
|
dst.writestr(name, path.read_bytes())
|
|
if include_env and env_text is not None:
|
|
dst.writestr(".env", env_text)
|
|
if include_admin_readme:
|
|
dst.writestr("README_ERP_LOCAL_AGENT.txt", f"ERP Local Agent {ERP_LOCAL_AGENT_VERSION}\nStorage behaviour and D:\\AuditFirmStorage are preserved. Future compatible releases self-update securely from ERP.\n")
|
|
return buffer.getvalue()
|
|
|
|
|
|
def build_preconfigured_agent_zip(*, env_text: str, include_admin_readme: bool = False) -> bytes:
|
|
return _build_zip(env_text=env_text, include_env=True, include_admin_readme=include_admin_readme)
|
|
|
|
|
|
def build_agent_update_zip() -> bytes:
|
|
# Update package deliberately excludes .env, .venv, data, logs and client files.
|
|
return _build_zip(env_text=None, include_env=False, include_admin_readme=False)
|