Синхронизация с server/: нормализация видео, овал-подсказка, хосты с путём, DNR-стенд
- app.py: probe_video(), HDR→bt709, fps=30, -t 20, логирование progress/face_ok, split_host_path() для хостов с путём (domain.com/v1/giga), FPG_WS_PORT - templates/index.html: овал-подсказка, canvas 1080×1080, запас 10% - templates/result.html: стилевые правки (цвета) - gigadoc_ws.py: поддержка пути в --sdk-host - test_fpg.py: переведён на DNR-стенд (bearer subprotocol, Origin) - README: хосты с путём, FPG_WS_PORT, нормализация - .gitignore: server/, wordpress/, wordpress-plugin/ (отдельные репозитории)
This commit is contained in:
1 parent
421183b7e3
commit
7375618b1d
9 files changed
+423
-108
No files matched your search
+44
-12
@@ -1,9 +1,12 @@
|
||||
"""
|
||||
Тестовый скрипт для SDK ФПГ (GigaDoc).
|
||||
Тестовый скрипт для SDK ФПГ (GigaDoc, персональный стенд DNR).
|
||||
Покрывает: авторизацию → создание сессии → WebSocket (изображения) → удаление сессии.
|
||||
|
||||
Использование:
|
||||
python3 test_fpg.py [--ws-host HOST] [--ws-port PORT] [--image-dir DIR]
|
||||
|
||||
Хосты можно переопределить через env: FPG_REST_HOST, FPG_WS_HOST, FPG_WS_PORT.
|
||||
Хост может содержать путь (domain.com/v1/giga) — он учитывается при сборке URL.
|
||||
"""
|
||||
|
||||
import argparse
|
||||
@@ -17,17 +20,37 @@ import time
|
||||
import requests
|
||||
|
||||
# ─── Конфигурация ─────────────────────────────────────────────
|
||||
OAUTH_URL = "https://oauth.sbermed.ai/token"
|
||||
CLIENT_ID = "client_fpg_front"
|
||||
OAUTH_URL = os.environ.get("FPG_OAUTH_URL", "https://oauth.sbermed.ai/token")
|
||||
CLIENT_ID = os.environ.get("FPG_CLIENT_ID", "client_fpg_front_dnr")
|
||||
CLIENT_SECRET = os.environ.get("FPG_CLIENT_SECRET", "")
|
||||
|
||||
REST_HOST = "gigadoc-sdk.sbermed.ai"
|
||||
REST_BASE = f"https://{REST_HOST}"
|
||||
REST_HOST = os.environ.get("FPG_REST_HOST", "gigadoc-dnr-sdk.sbermed.ai")
|
||||
WS_HOST = os.environ.get("FPG_WS_HOST", "gigadoc-dnr-sdk.sbermed.ai")
|
||||
WS_PORT = int(os.environ.get("FPG_WS_PORT", "8443"))
|
||||
WS_ORIGIN = os.environ.get("FPG_WS_ORIGIN", "https://gigadoc-dnr-sdk.sbermed.ai")
|
||||
|
||||
DEVICE_ID = "web_prod"
|
||||
FORM_FACTOR = "web_prod"
|
||||
|
||||
|
||||
def split_host_path(host: str) -> tuple:
|
||||
"""Разделить 'host/path' на (scheme, hostname, base_path)."""
|
||||
host = host.strip().rstrip("/")
|
||||
scheme = ""
|
||||
if "://" in host:
|
||||
scheme, host = host.split("://", 1)
|
||||
parts = host.split("/", 1)
|
||||
hostname = parts[0]
|
||||
base_path = "/" + parts[1] if len(parts) > 1 else ""
|
||||
return scheme, hostname, base_path
|
||||
|
||||
|
||||
_rest_scheme, _rest_hostname, _rest_base_path = split_host_path(REST_HOST)
|
||||
REST_BASE = f"{_rest_scheme or 'https'}://{_rest_hostname}{_rest_base_path}"
|
||||
|
||||
_ws_scheme, WS_HOSTNAME, WS_BASE_PATH = split_host_path(WS_HOST)
|
||||
|
||||
|
||||
def get_token() -> str:
|
||||
"""Получить access token через client_credentials."""
|
||||
resp = requests.post(
|
||||
@@ -76,13 +99,20 @@ async def ws_scan(token: str, uid: str, ws_host: str, ws_port: int, image_dir: s
|
||||
"""Подключиться по WebSocket, отправить изображения, получить результат."""
|
||||
import websockets
|
||||
|
||||
uri = f"wss://{ws_host}:{ws_port}/api/v2/ws/predict/{uid}/image"
|
||||
# ws_host может быть с путём (domain.com/v1/giga) — разделяем.
|
||||
_scheme, _hostname, _base_path = split_host_path(ws_host)
|
||||
_base_path = _base_path or WS_BASE_PATH
|
||||
|
||||
uri = f"wss://{_hostname}:{ws_port}{_base_path}/api/v2/ws/predict/{uid}/image"
|
||||
print(f"[WS] Подключение: {uri}")
|
||||
|
||||
# Авторизация через subprotocol bearer.<token> (как фронт),
|
||||
# Origin — обязательный заголовок.
|
||||
async with websockets.connect(
|
||||
uri,
|
||||
open_timeout=10,
|
||||
additional_headers={"Authorization": f"Bearer {token}"},
|
||||
subprotocols=[f"bearer.{token}"],
|
||||
extra_headers={"Origin": WS_ORIGIN},
|
||||
) as ws:
|
||||
print("[WS] Соединение установлено")
|
||||
|
||||
@@ -145,16 +175,18 @@ def create_test_image():
|
||||
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser(description="Тест SDK ФПГ")
|
||||
parser.add_argument("--ws-host", default="sdk.sbermed.ai", help="WebSocket host")
|
||||
parser.add_argument("--ws-port", type=int, default=8443, help="WebSocket port")
|
||||
parser = argparse.ArgumentParser(description="Тест SDK ФПГ (DNR)")
|
||||
parser.add_argument("--ws-host", default=WS_HOSTNAME, help="WebSocket host (может быть с путём)")
|
||||
parser.add_argument("--ws-port", type=int, default=WS_PORT, help="WebSocket port")
|
||||
parser.add_argument("--image-dir", default=".", help="Директория с JPEG-файлами")
|
||||
parser.add_argument("--rest-only", action="store_true", help="Только REST API (без WS)")
|
||||
args = parser.parse_args()
|
||||
|
||||
print("=" * 60)
|
||||
print("🏥 Тестирование SDK ФПГ (GigaDoc)")
|
||||
print("🏥 Тестирование SDK ФПГ (GigaDoc DNR)")
|
||||
print("=" * 60)
|
||||
print(f"[CONFIG] REST: {REST_BASE}")
|
||||
print(f"[CONFIG] WS: wss://{args.ws_host}:{args.ws_port}")
|
||||
|
||||
# 0. Тестовое изображение
|
||||
create_test_image()
|
||||
@@ -192,4 +224,4 @@ def main():
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
main()
|
||||
Reference in new issue
Block a user