def rapid_upload_info(self,
remotepath: str,
check: bool = True) -> Optional[PcsRapidUploadInfo]:
pcs_file = self.meta(remotepath)[0]
content_length = pcs_file.size
if content_length < 256 * constant.OneK:
return None
fs = self.file_stream(remotepath, pcs=False)
if not fs:
return None
data = fs.read(256 * constant.OneK)
assert data and len(data) == 256 * constant.OneK
slice_md5 = calu_md5(data)
assert (content_length
and content_length == fs._auto_decrypt_request.content_length)
content_md5 = fs._auto_decrypt_request.content_md5
content_crc32 = fs._auto_decrypt_request.content_crc32 or 0
if not content_md5:
return None
block_list = pcs_file.block_list
if block_list and len(
block_list) == 1 and block_list[0] == pcs_file.md5:
return PcsRapidUploadInfo(
slice_md5=slice_md5,
content_md5=content_md5,
content_crc32=content_crc32,
content_length=content_length,
remotepath=pcs_file.path,
)
if check:
try:
# Try rapid_upload_file
self.rapid_upload_file(
slice_md5,
content_md5,
content_crc32,
content_length,
pcs_file.path,
local_ctime=pcs_file.local_ctime,
local_mtime=pcs_file.local_mtime,
ondup="overwrite",
)
except BaiduPCSError as err:
# 31079: "未找到文件MD5"
if err.error_code != 31079:
raise err
return None
return PcsRapidUploadInfo(
slice_md5=slice_md5,
content_md5=content_md5,
content_crc32=content_crc32,
content_length=content_length,
remotepath=pcs_file.path,
)
def rapid_upload(
api: BaiduPCSApi,
remotedir: str,
link: str = "",
slice_md5: str = "",
content_md5: str = "",
content_crc32: int = 0,
content_length: int = 0,
filename: str = "",
no_ignore_existing: bool = False,
rapiduploadinfo_file: Optional[str] = None,
user_id: Optional[int] = None,
user_name: Optional[str] = None,
):
"""Rapid upload with params
If given `link` and `filename`, then filename of link will be replace by `filename`
"""
if link:
slice_md5, content_md5, content_crc32, content_length, _filename = _parse_link(
link)
content_crc32 = content_crc32 or 0
filename = filename or _filename
remotepath = join_path(remotedir, filename)
assert all([slice_md5, content_md5, content_length
]), f"`rapid_upload`: parsing rapid upload link fails: {link}"
if not no_ignore_existing:
if api.exists(remotepath):
return
try:
pcs_file = api.rapid_upload_file(
slice_md5,
content_md5,
content_crc32,
content_length,
remotepath,
ondup="overwrite",
)
if rapiduploadinfo_file:
save_rapid_upload_info(
rapiduploadinfo_file,
slice_md5,
content_md5,
content_crc32,
content_length,
remotepath=remotepath,
user_id=user_id,
user_name=user_name,
)
print(f"[i blue]Save to[/i blue] {pcs_file.path}")
except Exception as err:
link = PcsRapidUploadInfo(
slice_md5=slice_md5,
content_md5=content_md5,
content_crc32=content_crc32,
content_length=content_length,
remotepath=remotepath,
).cs3l()
print(
f"[i yellow]Rapid Upload fails[/i yellow]: error: {err} link: {link}",
)