Files
Forge-Engine/ai/scenario_client.py
T
williamandClaude Sonnet 5 c57420c8c9 Phase 3 : hardening qualite de code - typage strict, securite, dead code, a11y
Config strictement stricte partout (ruff, mypy --strict, bandit, vulture,
import-linter, eslint, stylelint), aucune regle desactivee "pour ne pas
casser le build" - l'existant a ete corrige pour la satisfaire plutot que
l'inverse. Hooks pre-commit locaux (language: system) bloquants.

- Typage mypy --strict propage a tout le moteur (db, screens, auth, core,
  ai, routes, puis publish/scripts/tests/app.py/build_css.py).
- Securite : fuite de handle fichier Windows corrigee dans l'export SCORM
  (routes/publish/export_scorm.py), CSRF/RNG non-crypto/xAPI documentes
  (# nosec, # NOSONAR justifies), nouveau db.json_for_script() (echappe
  "</script>" dans le JSON embarque en <script>, 25 sites).
- Architecture : imports circulaires/F811 nettoyes, contrats
  import-linter respectes, code mort retire (vulture).
- Accessibilite : 69 champs de formulaire sans label correctement
  associe corriges (for/id ou aria-label) sur 11 templates.
- ESLint/Stylelint : lot mecanique JS/CSS, regles ajustees puis
  appliquees (aucune desactivee sans verification individuelle).
- Tests : isolation du compte admin partage (nettoyage ponctuel +
  fixture de teardown automatique en filet de securite), suite complete
  verte (591 tests Python, 241 tests JS).
- SonarQube Community Build self-heberge (Docker + PostgreSQL) : rapport
  complet analyse point par point, faux positifs documentes.
- .gitattributes ajoute (LF force) : core.autocrlf=true sur cette machine
  faisait echouer ESLint (linebreak-style) via un bug connu de git
  (checkout "en place" qui ignore l'eol force sur un fichier deja
  present sur disque - contourne en supprimant puis recreant chaque
  fichier suivi).

djLint (H021, styles inline) volontairement saute pour ce commit -
backlog assume, deja documente, traite dans un lot separe.

Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
2026-09-15 16:06:15 +02:00

74 lines
3.1 KiB
Python

"""Client Scenario (voir plan Phase 2, §3) — génération d'images (fonds/
objets 2D uniquement, jamais de personnages/sprites — décision actée
lors de l'étude préalable). API confirmée (docs.scenario.com) :
authentification Basic (clé + secret), génération ASYNCHRONE par job —
POST déclenche un job, GET /jobs/{id} jusqu'à "success"/"failure", puis
GET /assets/{id} pour l'URL finale téléchargeable.
Ce module reste un simple client HTTP : il renvoie l'URL de l'image,
jamais ne la télécharge/l'enregistre lui-même — ça reste le travail de
ai/tools.py::_dispatch_add_generated_image (voir "Mes assets",
auth/user_assets_dir.py), pour que ce module soit testable seul."""
import os
import time
import requests
_BASE_URL = "https://api.cloud.scenario.com/v1"
class ScenarioNotConfiguredError(Exception):
"""SCENARIO_API_KEY/SCENARIO_API_SECRET/SCENARIO_MODEL_ID absents —
voir .env.example. SCENARIO_MODEL_ID est l'id du modèle entraîné sur
le style graphique unique de Forge Engine (pas encore créé tant que
l'entraînement Scenario n'a pas été fait)."""
class ScenarioGenerationError(Exception):
"""Le job Scenario a échoué ou n'a pas répondu à temps."""
def _credentials() -> tuple[tuple[str, str], str]:
api_key = os.environ.get("SCENARIO_API_KEY")
api_secret = os.environ.get("SCENARIO_API_SECRET")
model_id = os.environ.get("SCENARIO_MODEL_ID")
if not api_key or not api_secret or not model_id:
raise ScenarioNotConfiguredError(
"SCENARIO_API_KEY/SCENARIO_API_SECRET/SCENARIO_MODEL_ID absents — voir .env.example."
)
return (api_key, api_secret), model_id
def generate_image_url(
prompt: str, width: int = 768, height: int = 768, timeout: int = 120, poll_interval: int = 2
) -> str:
"""Lance une génération txt2img et attend le résultat — renvoie
l'URL finale de l'image (hébergée par Scenario, à télécharger par
l'appelant). Bloquant (poll_interval secondes entre chaque
vérification), borné par `timeout` secondes au total."""
auth, model_id = _credentials()
response = requests.post(
f"{_BASE_URL}/generate/txt2img",
auth=auth,
timeout=30,
json={"prompt": prompt, "modelId": model_id, "width": width, "height": height, "numSamples": 1},
)
response.raise_for_status()
job_id = response.json()["job"]["jobId"]
deadline = time.monotonic() + timeout
while time.monotonic() < deadline:
status_response = requests.get(f"{_BASE_URL}/jobs/{job_id}", auth=auth, timeout=30)
status_response.raise_for_status()
job = status_response.json()["job"]
if job["status"] == "success":
asset_id = job["metadata"]["assetIds"][0]
asset_response = requests.get(f"{_BASE_URL}/assets/{asset_id}", auth=auth, timeout=30)
asset_response.raise_for_status()
return str(asset_response.json()["asset"]["url"])
if job["status"] == "failure":
raise ScenarioGenerationError("Scenario a échoué à générer l'image.")
time.sleep(poll_interval)
raise ScenarioGenerationError("Scenario n'a pas répondu à temps.")