From 39fd601b8151d7d9342f207766e7c90401539f64 Mon Sep 17 00:00:00 2001
From: mjzhang95 <mjzhang95@163.com>
Date: Sun, 6 Apr 2025 19:09:48 +0800
Subject: [PATCH] Update views.py

---
 spug_api/apps/host/views.py | 21 +++++++++++++++++++--
 1 file changed, 19 insertions(+), 2 deletions(-)

diff --git a/spug_api/apps/host/views.py b/spug_api/apps/host/views.py
index db5036e..3257a6d 100644
--- a/spug_api/apps/host/views.py
+++ b/spug_api/apps/host/views.py
@@ -13,7 +13,8 @@ from apps.exec.models import ExecTemplate
 from apps.app.models import Deploy
 from apps.schedule.models import Task
 from apps.monitor.models import Detection
-from libs.ssh import SSH, AuthenticationException
+from libs.ssh import SSH
+from libs.telnet import Telnet, AuthenticationException
 from paramiko.ssh_exception import BadAuthenticationType
 from openpyxl import load_workbook
 from threading import Thread
@@ -193,7 +194,23 @@ def batch_valid(request):
 
 
 def _do_host_verify(form):
-    password = form.pop('password')
+    password = form.pop('password', None)
+    connect_type = form.pop('connect_type', 'ssh')
+    
+    if connect_type == 'telnet':
+        if not password:
+            return False
+        try:
+            with Telnet(form.hostname, form.port or 23, form.username, password) as tn:
+                return True
+        except AuthenticationException:
+            raise Exception('Telnet认证失败，请检查用户名和密码是否正确')
+        except socket.timeout:
+            raise Exception('连接主机超时，请检查网络')
+        except Exception as e:
+            raise Exception(f'Telnet连接失败: {str(e)}')
+    
+    # SSH验证逻辑
     if form.pkey:
         try:
             with SSH(form.hostname, form.port, form.username, form.pkey) as ssh: