фикс отвала при успешной авторизации

backend/app/services/discovery/ssh_client.py

@@ -84,21 +84,35 @@ async def ssh_run_multi(
         _log_attempt(i, len(options), host, opt)
         try:
             connect_kwargs = _build_connect_kwargs(host, port, opt)
-            async with asyncssh.connect(**connect_kwargs) as conn:
+            conn = await asyncssh.connect(**connect_kwargs)
-                result = await asyncio.wait_for(
-                    conn.run(command, check=True), timeout=timeout
-                )
-                logger.info(
-                    "[discovery] SSH auth succeeded for %s with %s(%s)",
-                    host, opt.type, opt.username,
-                )
-                return result.stdout
         except Exception as exc:
             last_exc = exc
             logger.info(
                 "[discovery] SSH auth failed for %s with %s(%s): %s",
                 host, opt.type, opt.username, exc,
             )
+            continue
+
+        logger.info(
+            "[discovery] SSH auth succeeded for %s with %s(%s)",
+            host, opt.type, opt.username,
+        )
+        try:
+            async with conn:
+                result = await asyncio.wait_for(
+                    conn.run(command, check=True), timeout=timeout
+                )
+                return result.stdout
+        except asyncio.TimeoutError as exc:
+            err = TimeoutError(f"SSH command timed out after {timeout}s on {host}: {command}")
+            logger.warning("[discovery] %s", err)
+            raise err from exc
+        except Exception as exc:
+            logger.warning(
+                "[discovery] SSH command failed on %s after auth with %s(%s): %s",
+                host, opt.type, opt.username, exc,
+            )
+            raise
     raise last_exc
 
 

backend/app/services/ssh.py

@@ -251,21 +251,10 @@ async def _try_connect(
 
     try:
         conn = await asyncio.wait_for(asyncssh.connect(**connect_kwargs), timeout=15)
-        async with conn:
-            result = await asyncio.wait_for(conn.run(command, check=False), timeout=timeout)
-        duration = int((time.monotonic() - start) * 1000)
-        return SSHResult(
-            success=result.exit_status == 0,
-            stdout=(result.stdout or "").strip(),
-            stderr=(result.stderr or "").strip(),
-            exit_code=result.exit_status or 0,
-            duration_ms=duration,
-            auth_ok=True,  # Connection was established — auth succeeded
-        )
     except asyncio.TimeoutError:
         return SSHResult(
             success=False,
-            error="Connection timed out",
+            error="SSH connection timed out",
             duration_ms=int((time.monotonic() - start) * 1000),
         )
     except (asyncssh.DisconnectError, asyncssh.PermissionDenied, asyncssh.ConnectionLost) as e:
@@ -281,6 +270,32 @@ async def _try_connect(
             duration_ms=int((time.monotonic() - start) * 1000),
         )
 
+    try:
+        async with conn:
+            result = await asyncio.wait_for(conn.run(command, check=False), timeout=timeout)
+        duration = int((time.monotonic() - start) * 1000)
+        return SSHResult(
+            success=result.exit_status == 0,
+            stdout=(result.stdout or "").strip(),
+            stderr=(result.stderr or "").strip(),
+            exit_code=result.exit_status or 0,
+            duration_ms=duration,
+            auth_ok=True,
+        )
+    except asyncio.TimeoutError:
+        return SSHResult(
+            success=False,
+            error=f"SSH command timed out after {timeout}s",
+            duration_ms=int((time.monotonic() - start) * 1000),
+            auth_ok=True,
+        )
+    except Exception as e:
+        return SSHResult(
+            success=False,
+            error=str(e),
+            duration_ms=int((time.monotonic() - start) * 1000),
+            auth_ok=True,
+        )
 
 async def _save_last_auth(device: Device, db: AsyncSession, opt: _AuthOption) -> None:
     """Persist the successful auth method to device.ssh_last_auth."""