From b93e3013d06d47b18aedf12f2a200d689e3872cc Mon Sep 17 00:00:00 2001
From: qiangyanwen <508737091@qq.com>
Date: Sat, 7 Jan 2023 18:39:27 +0800
Subject: [PATCH] add test

---
 utils/ssh_client.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/utils/ssh_client.py b/utils/ssh_client.py
index a99091a..7ca0fdf 100644
--- a/utils/ssh_client.py
+++ b/utils/ssh_client.py
@@ -4,6 +4,7 @@
 # @Author   :qiangyanwen
 # @File     :ssh_client.py
 import paramiko
+from config.log import logger
 
 
 class SSHConnection(object):
@@ -17,7 +18,7 @@ class SSHConnection(object):
         self._client = None
 
     def __enter__(self):
-        print("客户端开始创建连接.....")
+        logger.info("ssh开始创建链接")
         transport = paramiko.Transport((self._host, self._port))
         transport.connect(username=self._username, password=self._password)
         self._transport = transport
@@ -52,7 +53,7 @@ class SSHConnection(object):
             self._transport.close()
         if self._client:
             self._client.close()
-        print("客户端关闭已成功.....")
+        logger.info("ssh 关闭连接")
 
 
 with SSHConnection("47.96.135.132", 22, "root", "Qyw1994@520") as ssh: