11

ftp.py

@@ -0,0 +1,76 @@
+import os
+import ftplib
+import socks
+import socket
+from urllib.parse import urlparse
+
+def set_proxy(proxy_url):
+    """设置代理"""
+    if proxy_url:
+        parsed_proxy = urlparse(proxy_url)
+        proxy_host = parsed_proxy.hostname
+        proxy_port = parsed_proxy.port
+        socks.set_default_proxy(socks.SOCKS5, proxy_host, proxy_port)  # 可以根据需要调整为 SOCKS5、SOCKS4 或 HTTP
+        socket.socket = socks.socksocket
+        print(f"已设置代理: {proxy_host}:{proxy_port}")
+
+def upload_file_to_ftp(server, username, password, file_path, remote_path, proxy_url=None):
+    """上传单个文件到FTP"""
+    try:
+        # 设置代理（如果有的话）
+        set_proxy(proxy_url)
+
+        with ftplib.FTP(server) as ftp:
+            ftp.login(user=username, passwd=password)
+            with open(file_path, 'rb') as file:
+                ftp.storbinary(f'STOR {remote_path}', file)
+            print(f"文件 {file_path} 成功上传到 {remote_path}！")
+
+    except ftplib.all_errors as e:
+        print(f"FTP 错误: {e}")
+    except Exception as e:
+        print(f"发生错误: {e}")
+
+def create_remote_directory(ftp, remote_directory):
+    """确保远程目录存在，如果不存在则创建"""
+    try:
+        ftp.cwd(remote_directory)  # 尝试进入远程目录
+    except ftplib.error_perm:
+        ftp.mkd(remote_directory)  # 如果远程目录不存在，则创建
+        ftp.cwd(remote_directory)  # 再次进入远程目录
+        print(f"远程目录 {remote_directory} 已创建！")
+
+def upload_directory_to_ftp(server, username, password, local_directory, remote_directory, proxy_url=None):
+    """上传整个目录及其子目录到FTP服务器"""
+    try:
+        # 设置代理（如果有的话）
+        set_proxy(proxy_url)
+
+        with ftplib.FTP(server) as ftp:
+            ftp.login(user=username, passwd=password)
+
+            # 确保远程目录存在
+            create_remote_directory(ftp, remote_directory)
+
+            # 遍历本地目录树
+            for root, dirs, files in os.walk(local_directory):
+                # 上传子文件夹
+                for dir_name in dirs:
+                    remote_dir_path = os.path.join(remote_directory, os.path.relpath(os.path.join(root, dir_name), local_directory)).replace("\\", "/")
+                    try:
+                        ftp.mkd(remote_dir_path)  # 在FTP服务器上创建子目录
+                        print(f"已创建远程目录: {remote_dir_path}")
+                    except ftplib.error_perm:
+                        pass  # 如果目录已经存在则忽略
+
+                # 上传文件
+                for filename in files:
+                    local_file_path = os.path.join(root, filename)  # 本地文件的完整路径
+                    relative_path = os.path.relpath(local_file_path, local_directory)  # 计算相对路径
+                    remote_file_path = os.path.join(remote_directory, relative_path).replace("\\", "/")  # 计算远程文件路径
+                    upload_file_to_ftp(server, username, password, local_file_path, remote_file_path, proxy_url)  # 上传文件
+
+    except ftplib.all_errors as e:
+        print(f"上传目录时发生错误: {e}")
+    except Exception as e:
+        print(f"发生错误: {e}")

main.py

@@ -12,8 +12,10 @@ import time
 import re
 import os
 import tkinter as tk
-import ftplib
-
+from ftp import set_proxy
+from ftp import upload_file_to_ftp
+from ftp import upload_directory_to_ftp
+from ftp import create_remote_directory
 #
 # def ftp_upload_with_proxy(server, username, password, local_file_path, remote_path, proxy):
 #     """使用 pycurl 上传单个文件到 FTP 服务器"""
@@ -69,50 +71,50 @@ import ftplib
 #
 #     except Exception as e:
 #         print(f"上传目录时发生错误: {e}")
-
-def upload_file_to_ftp(server, username, password, file_path, remote_path):
-    try:
-        with ftplib.FTP(server) as ftp:
-            ftp.login(user=username, passwd=password)
-            with open(file_path, 'rb') as file:
-                ftp.storbinary(f'STOR {remote_path}', file)
-            print(f"文件 {file_path} 成功上传到 {remote_path}！")
-    except ftplib.all_errors as e:
-        print(f"FTP 错误: {e}")
-
-def upload_directory_to_ftp(server, username, password, local_directory, remote_directory):
-    """上传整个目录及其子目录到FTP服务器"""
-    try:
-        with ftplib.FTP(server) as ftp:
-            ftp.login(user=username, passwd=password)
-
-            # 确保远程目录存在
-            try:
-                ftp.cwd(remote_directory)  # 改变到远程目录
-            except ftplib.error_perm:
-                ftp.mkd(remote_directory)  # 如果远程目录不存在，则创建
-                ftp.cwd(remote_directory)  # 再次进入远程目录
-
-            # 遍历本地目录树
-            for root, dirs, files in os.walk(local_directory):
-                # 上传子文件夹
-                for dir_name in dirs:
-                    remote_dir_path = os.path.join(remote_directory, os.path.relpath(os.path.join(root, dir_name), local_directory)).replace("\\", "/")
-                    try:
-                        ftp.mkd(remote_dir_path)  # 在FTP服务器上创建子目录
-                        print(f"已创建远程目录: {remote_dir_path}")
-                    except ftplib.error_perm:
-                        pass  # 如果目录已经存在则忽略
-
-                # 上传文件
-                for filename in files:
-                    local_file_path = os.path.join(root, filename)  # 本地文件的完整路径
-                    relative_path = os.path.relpath(local_file_path, local_directory)  # 计算相对路径
-                    remote_file_path = os.path.join(remote_directory, relative_path).replace("\\", "/")  # 计算远程文件路径
-                    ftp_upload_with_proxy(server, username, password, local_file_path, remote_file_path, proxy_url)  # 上传文件
-
-    except ftplib.all_errors as e:
-        print(f"上传目录时发生错误: {e}")
+#
+# def upload_file_to_ftp(server, username, password, file_path, remote_path):
+#     try:
+#         with ftplib.FTP(server) as ftp:
+#             ftp.login(user=username, passwd=password)
+#             with open(file_path, 'rb') as file:
+#                 ftp.storbinary(f'STOR {remote_path}', file)
+#             print(f"文件 {file_path} 成功上传到 {remote_path}！")
+#     except ftplib.all_errors as e:
+#         print(f"FTP 错误: {e}")
+#
+# def upload_directory_to_ftp(server, username, password, local_directory, remote_directory):
+#     """上传整个目录及其子目录到FTP服务器"""
+#     try:
+#         with ftplib.FTP(server) as ftp:
+#             ftp.login(user=username, passwd=password)
+#
+#             # 确保远程目录存在
+#             try:
+#                 ftp.cwd(remote_directory)  # 改变到远程目录
+#             except ftplib.error_perm:
+#                 ftp.mkd(remote_directory)  # 如果远程目录不存在，则创建
+#                 ftp.cwd(remote_directory)  # 再次进入远程目录
+#
+#             # 遍历本地目录树
+#             for root, dirs, files in os.walk(local_directory):
+#                 # 上传子文件夹
+#                 for dir_name in dirs:
+#                     remote_dir_path = os.path.join(remote_directory, os.path.relpath(os.path.join(root, dir_name), local_directory)).replace("\\", "/")
+#                     try:
+#                         ftp.mkd(remote_dir_path)  # 在FTP服务器上创建子目录
+#                         print(f"已创建远程目录: {remote_dir_path}")
+#                     except ftplib.error_perm:
+#                         pass  # 如果目录已经存在则忽略
+#
+#                 # 上传文件
+#                 for filename in files:
+#                     local_file_path = os.path.join(root, filename)  # 本地文件的完整路径
+#                     relative_path = os.path.relpath(local_file_path, local_directory)  # 计算相对路径
+#                     remote_file_path = os.path.join(remote_directory, relative_path).replace("\\", "/")  # 计算远程文件路径
+#                     upload_file_to_ftp(server, username, password, local_file_path, remote_file_path, )  # 上传文件
+#
+#     except ftplib.all_errors as e:
+#         print(f"上传目录时发生错误: {e}")
 
 def launch_proxy():
     print("开启代理...")
@@ -225,7 +227,7 @@ def shot(chromedriver_path ,chrome_path):
         ftp_username = matches[2]
         ftp_password = matches[3]
         # 调用上传函数
-        upload_directory_to_ftp(ftp_server, ftp_username, ftp_password, local_file_path, remote_file_path, proxy_url)
+        upload_directory_to_ftp(ftp_server, ftp_username, ftp_password, local_file_path, remote_file_path)
 
         # url = f"http://{matches[0]}:{matches[1]}/exam_setup.php"
         # response = requests.get(url, cookies=session_cookies)