优化sf获取cnc程序和ftp

sf_mrs_connect/models/ftp_operate.py

@@ -15,7 +15,7 @@ class FtpController():
 
     ftp = FTP()
 
-    def __init__(self, host="192.168.50.202", port=21, username="ftpuser", password="123456"):
+    def __init__(self, host, port, username, password):
         try:
             self.ftp.connect(host, port)
             self.ftp.login(username, password)
@@ -27,17 +27,18 @@ class FtpController():
     def download_file_tree(self, target_dir, serverdir):
         if not os.path.exists(serverdir):
             os.makedirs(serverdir)
+        try:
             self.ftp.cwd(target_dir)  # 切换工作路径
             remotenames = self.ftp.nlst()
             for file in remotenames:
                 server = os.path.join(serverdir, file)
                 if file.find(".") != -1:
                     self.download_file(server, file)
-        else:
-            return
+        except:
+            return False
 
-    # 下载指定目录下的指定文件
-    def download_file(self, serverfile, remotefile):
-        file_handler = open(serverfile, 'wb')
-        self.ftp.retrbinary('RETR ' + remotefile, file_handler.write)
-        file_handler.close()
+        # 下载指定目录下的指定文件
+        def download_file(self, serverfile, remotefile):
+            file_handler = open(serverfile, 'wb')
+            self.ftp.retrbinary('RETR ' + remotefile, file_handler.write)
+            file_handler.close()