丁云鹏 4 meses atrás
pai
commit
39b9812699

+ 9 - 1
recommend-model-produce/src/main/python/tools/utils/compress.py

@@ -22,7 +22,15 @@ def compress_file_tar(file_path, output_filename):
     # 创建一个 tarfile 对象,使用 'w:gz' 模式表示写入 gzip 压缩的 tar 包
     with tarfile.open(output_filename, "w:gz", format=tarfile.GNU_FORMAT) as tar:
         # 将文件添加到 tar 包中,arcname 指定在 tar 包中的相对路径
-        tar.add(file_path, arcname=os.path.basename(file_path))
+        #tar.add(file_path, arcname=os.path.basename(file_path))
+        file_name = os.path.basename(file_path)
+        # 打开文件
+        with open(file_path, 'rb') as file_to_add:
+            # 创建一个TarInfo对象,设置其名称为文件名
+            tar_info = tarfile.TarInfo(name=file_name)
+            # 从文件中读取内容并写入TarInfo对象
+            tar_info.size = os.path.getsize(file_path)
+            tar.addfile(tar_info, fileobj=file_to_add)
 
 def compress_tar(folder_path, output_filename):
     # 确保输出文件名以 .tar.gz 结尾