ezKit 1.12.0__py3-none-any.whl → 1.12.2__py3-none-any.whl

ezKit/xftp.py

@@ -1,4 +1,5 @@
-"""FTP"""
+"""FTP Library"""
+
 # ftplib: https://docs.python.org/3.12/library/ftplib.html
 import os
 from ftplib import FTP
@@ -17,7 +18,7 @@ class XFTP:
         username: str = "anonymous",
         password: str = "",
         encoding: str = "UTF-8",
-        debuglevel: int = 0
+        debuglevel: int = 0,
     ):
         """Initiation"""
         self.ftp = FTP()
@@ -61,7 +62,7 @@ class XFTP:
         logger.success("FTP connect closed")
         return True
 
-    def get_file_list(self, target='/') -> list[str] | None:
+    def get_file_list(self, target="/") -> list[str] | None:
         """Get file list"""
         try:
             self.chdir_to_remote(target)
@@ -84,7 +85,7 @@ class XFTP:
         try:
             dir_list = target.split("/")
             for i, _ in enumerate(dir_list):
-                dir_path = "/".join(dir_list[:i + 1])
+                dir_path = "/".join(dir_list[: i + 1])
                 try:
                     self.ftp.mkd(dir_path)
                 except Exception as e:
@@ -111,21 +112,23 @@ class XFTP:
         # 注意: exit() 并不会退出脚本, 配合 try 使用
         exit()
 
-    def x_exec(self, local_dir=".", local_file="", remote_dir="/", remote_file="", upload=False):
+    def x_exec(
+        self, local_dir=".", local_file="", remote_dir="/", remote_file="", upload=False
+    ):
         """Download or Upload"""
 
         bufsize = 1024
         local_path = f"{local_dir}/{local_file}"
         remote_path = f"{remote_dir}/{remote_file}"
 
-        info = 'Download'
+        info = "Download"
         if upload is True:
-            info = 'Upload'
+            info = "Upload"
 
         # 检查参数
         if upload is True:
             if local_file == "":
-                self.close('Argument Miss: local file')
+                self.close("Argument Miss: local file")
             # 如果没有设置 远程文件 名称, 则使用 本地文件 名称
             if remote_file == "":
                 remote_file = local_file
@@ -173,11 +176,15 @@ class XFTP:
                 stat = Path(local_file)
                 if stat.exists() and stat.is_file():
                     with open(local_file, "rb") as fid:
-                        self.ftp.storbinary(f'STOR {remote_file}', fid, bufsize)
-                    logger.success(f"{info} success: {local_path.replace('//', '/')} -> {remote_path.replace('//', '/')}")
+                        self.ftp.storbinary(f"STOR {remote_file}", fid, bufsize)
+                    logger.success(
+                        f"{info} success: {local_path.replace('//', '/')} -> {remote_path.replace('//', '/')}"
+                    )
                     return True
 
-                self.x_exit(f"{info} error: {local_path.replace('//', '/')} is not exist")
+                self.x_exit(
+                    f"{info} error: {local_path.replace('//', '/')} is not exist"
+                )
 
             else:
 
@@ -189,11 +196,15 @@ class XFTP:
                 # 下载文件
                 if remote_file in self.ftp.nlst():
                     with open(local_file, "wb") as fid:
-                        self.ftp.retrbinary(f'RETR {remote_file}', fid.write, bufsize)
-                    logger.success(f"{info} success: {remote_path.replace('//', '/')} -> {local_path.replace('//', '/')}")
+                        self.ftp.retrbinary(f"RETR {remote_file}", fid.write, bufsize)
+                    logger.success(
+                        f"{info} success: {remote_path.replace('//', '/')} -> {local_path.replace('//', '/')}"
+                    )
                     return True
 
-                self.x_exit(f"{info} error: {remote_path.replace('//', '/')} is not exist")
+                self.x_exit(
+                    f"{info} error: {remote_path.replace('//', '/')} is not exist"
+                )
 
         except Exception as e:
             # 第一层 try 使用 self.x_exit() 无效, 直接使用 self.close()
@@ -209,14 +220,25 @@ class XFTP:
             if stat.exists() is False:
                 self.close(f"Local directory error: {local_dir}")
             # 获取文件列表
-            local_files = [f for f in os.listdir(local_dir) if os.path.isfile(os.path.join(local_dir, f))]
+            local_files = [
+                f
+                for f in os.listdir(local_dir)
+                if os.path.isfile(os.path.join(local_dir, f))
+            ]
             for i in local_files:
-                self.x_exec(local_dir=local_dir, remote_dir=remote_dir, local_file=i, upload=True)
+                self.x_exec(
+                    local_dir=local_dir,
+                    remote_dir=remote_dir,
+                    local_file=i,
+                    upload=True,
+                )
         else:
             remote_files = self.get_file_list(remote_dir)
             if remote_files is not None:
                 for i in remote_files:
-                    self.x_exec(local_dir=local_dir, remote_dir=remote_dir, remote_file=i)
+                    self.x_exec(
+                        local_dir=local_dir, remote_dir=remote_dir, remote_file=i
+                    )
 
     def retrlines(self, remote_dir="/", cmd="LIST"):
         """Retrlines"""