电子技术论坛_中国专业的电子工程师学习交流社区-中电网技术论坛

标题: python文件操作工具类（2） [打印本页]

作者: look_w 时间: 2019-3-4 19:19 标题: python文件操作工具类（2）

@staticmethod
def upload_base64_file(base64_obj, dst_file_path=settings.UPLOAD_DEFAULT_FOLDER, file_ext=".jpg"):
      """
      上传base64文件流到服务器指定目录
      :param dst_file_path: 文件存放路径 "tmp/"
      :param base64_obj: base64流对象
      :param file_ext: 新文件扩展名：默认.jpg
      :return: 文件上传存入的文件目录
      """
      ret = {'status': False, 'msg': ''}
      try:
         # 将base64转成文件流对象
         file_str = base64_obj.split(',')[1].encode('utf-8')
         file_data = base64.b64encode(file_str)
         # 当前时间戳：年月日时分秒
         cur_time = int(round(time.time() * 1000))
         # 拼接新的文件名
         file_name = "%s%s" % (cur_time, file_ext)
         # 拼接文件存放目录
         file_path = "%s/%s" % (settings.MEDIA_ROOT, dst_file_path)
         save_path = "%s%s%s" % (dst_file_path, file_name, file_ext)
         # 判断文件目录
         FileOperateUtil.validate_folder_exists(file_path, is_create=True)
         try:
            # 拼接文件最终存放全路径
            real_path = "%s%s" % (file_path, file_name)
            # 上传文件操作
            destination = open(real_path, 'wb+')
            # 写入文件
            destination.write(file_data)
            ret = {'status': 200, 'msg': '上传成功', 'file_path': save_path}
         except Exception as e:
            log.debug("上传base64文件，写文件发生异常，%s" % e)
            raise Exception("上传base64文件，写文件发生异常")
         finally:
            destination.close()
      except Exception as e:
         log.debug("上传base64文件发生异常,%s" % e)
         ret = {'status': 500, 'msg': e}
      return ret

@staticmethod
def download_batch_file(src_file_folder, zip_folder=settings.ZIP_DEFAULT_FOLDER, down_file_name=None, is_delete=True):
      """
      将文件夹压缩打包下载
      :param src_file_folder: 要下载的文件目录
      :param zip_folder: 打包存放目录
      :param down_file_name: 下载时指定文件名
      :param is_delete: 下载完成后是否删除打包的文件，默认：删除
      :return: 带有文件流的response响应
      """
      # 判断待打包目录是否存在，不存在则抛出异常
      FileOperateUtil.validate_folder_exists(src_file_folder)
      # 判断打包存放目录是佛存在，不存在则新建
      FileOperateUtil.validate_folder_exists(zip_folder, is_create=True)
      # 设置响应头
      if os.path.exists(src_file_folder):
         cur_time = int(round(time.time() * 1000))
         zip_file_name = "%s" % cur_time
         zip_file_path = "%s/%s" % (settings.MEDIA_ROOT, zip_folder, zip_file_name)
         # 打包该目录
         FileOperateUtil.zip_dir(src_file_folder, zip_file_path)
         # 判断打包生成的文件是否是个文件
         if os.path.isfile(zip_file_path):
            filename = os.path.basename(zip_file_path)
            # 若未指定下载文件名，则已文件本身名称为准
            if down_file_name is None:
                  down_file_name = filename
            # 读取文件流，并构造请求响应
            response = HttpResponse(FileOperateUtil.file_iterator(zip_file_path))
            response['Content_type'] = "application/octet-stream"
            response['Content-Disposition'] = 'attachment; filename={0}'.format(down_file_name.encode('utf-8'))
         else:
            response = HttpResponse("下载失败,不是一个文件!", content_type="text/plain;charset=utf-8")
      else:
         response = HttpResponse("文件不存在,下载失败!", content_type="text/plain;charset=utf-8")
      return response

欢迎光临电子技术论坛_中国专业的电子工程师学习交流社区-中电网技术论坛 (http://bbs.eccn.com/)