56 lines
2.3 KiB
Python
56 lines
2.3 KiB
Python
from io import BytesIO
|
||
|
||
from ufile import filemanager, config, httprequest
|
||
from const import PUBLIC_KEY, PRIVATE_KEY, UPLOAD_SUFFIX, DOWNLOAD_SUFFIX
|
||
|
||
|
||
def _download_file_save_to_bio(url, header):
|
||
try:
|
||
response = httprequest.requests.get(url, headers=header, stream=True)
|
||
except httprequest.RequestException as e:
|
||
return None, httprequest.ResponseInfo(None, e)
|
||
if response.status_code in [200, 206]:
|
||
bio = BytesIO()
|
||
for block in response.iter_content(config.BLOCKSIZE):
|
||
bio.write(block)
|
||
else:
|
||
return
|
||
return bio
|
||
|
||
|
||
class BioFileManager(filemanager.FileManager):
|
||
def download_bytes_file(self, bucket, key, isprivate=True, expires=None, content_range=None, header=None):
|
||
"""
|
||
:param bucket: string类型, UFile空间名称
|
||
:param key: string类型， 下载文件在空间中的名称
|
||
:param isprivate: boolean类型，如果为私有空间则为True
|
||
:param expires: integer类型，私有文件链接有效时间
|
||
:param content_range: tuple类型，元素为两个整型
|
||
:param header: dict类型，http 请求header，键值对类型分别为string，比如{'User-Agent': 'Google Chrome'}
|
||
:return: ret: 如果http状态码为[200, 204, 206]之一则返回None，否则如果服务器返回json信息则返回dict类型，键值对类型分别为string, unicode string类型，否则返回空的dict
|
||
:return: ResponseInfo: 响应的具体信息，UCloud UFile 服务器返回信息或者网络链接异常
|
||
"""
|
||
if header is None:
|
||
header = dict()
|
||
else:
|
||
filemanager._check_dict(header)
|
||
|
||
if expires is None:
|
||
expires = config.get_default('expires')
|
||
|
||
if 'User-Agent' not in header:
|
||
header['User-Agent'] = config.get_default('user_agent')
|
||
|
||
if isinstance(content_range, tuple) and len(content_range) == 2:
|
||
header['Range'] = 'bytes=' + '-'.join(map(lambda x: str(x), content_range))
|
||
|
||
if not isprivate:
|
||
url = self.public_download_url(bucket, key)
|
||
else:
|
||
url = self.private_download_url(bucket, key, expires, header, True)
|
||
|
||
return _download_file_save_to_bio(url, header)
|
||
|
||
|
||
client = BioFileManager(PUBLIC_KEY, PRIVATE_KEY, UPLOAD_SUFFIX, DOWNLOAD_SUFFIX)
|