tencentblueking / bkpaas-python-sdk Goto Github PK

paht = "/app/xxxx"
经过_save() 处理之后 会变为 app/xxxx

def get_available_name(self, name, max_length=None):
    """
    Return a filename that's free on the target storage system and
    available for new content to be written to.
    """
    dir_name, file_name = os.path.split(name)
    file_root, file_ext = os.path.splitext(file_name)
    # If the filename already exists, add an underscore and a random 7
    # character alphanumeric string (before the file extension, if one
    # exists) to the filename until the generated filename doesn't exist.
    # Truncate original name if required, so the new filename does not
    # exceed the max_length.
    while self.exists(name) or (max_length and len(name) > max_length):
        # file_ext includes the dot.
        name = os.path.join(dir_name, "%s_%s%s" % (file_root, get_random_string(7), file_ext))
        if max_length is None:
            continue
        # Truncate file_root if max_length exceeded.
        truncation = len(name) - max_length
        if truncation > 0:
            file_root = file_root[:-truncation]
            # Entire file_root was truncated in attempt to find an available filename.
            if not file_root:
                raise SuspiciousFileOperation(
                    'Storage can not find an available filename for "%s". '
                    'Please make sure that the corresponding file field '
                    'allows sufficient "max_length".' % name
                )
            name = os.path.join(dir_name, "%s_%s%s" % (file_root, get_random_string(7), file_ext))
    return name

class StorageFileOverwriteMixin:

  file_overwrite = settings.FILE_OVERWRITE

  def get_available_name(self, name, max_length=None):
      """重写获取文件有效名称函数，支持在 file_overwrite=True 时不随机生成文件名"""

      dir_name, file_name = os.path.split(name)
      file_root, file_ext = os.path.splitext(file_name)

      def _gen_random_name(_file_root) -> str:
          # 在文件名的起始位置添加随机串，源码规则为 "%s_%s%s" % (_file_root, get_random_string(7), file_ext)
          # 上述规则对 .tar.gz 不友好，会在类型后缀中间加随机串，所以改为随机串作为前缀
          return os.path.join(dir_name, "%s_%s%s" % (get_random_string(7), _file_root, file_ext))

      # not self.file_overwrite and self.exists(name) 利用 and 短路特点，如果 file_overwrite=True 就无需校验文件是否存在
      while (not self.file_overwrite and self.exists(name)) or (max_length and len(name) > max_length):
          # file_ext includes the dot.
          name = name if self.file_overwrite else _gen_random_name(file_root)

          if max_length is None:
              continue
          # Truncate file_root if max_length exceeded.
          truncation = len(name) - max_length
          if truncation > 0:
              file_root = file_root[:-truncation]
              # Entire file_root was truncated in attempt to find an available filename.
              if not file_root:
                  raise SuspiciousFileOperation(
                      'Storage can not find an available filename for "%s". '
                      "Please make sure that the corresponding file field "
                      'allows sufficient "max_length".' % name
                  )
              name = name if self.file_overwrite else _gen_random_name(file_root)
      return name