73 lines
2.1 KiB
Python
73 lines
2.1 KiB
Python
import os
|
||
from datetime import datetime
|
||
from typing import AnyStr, BinaryIO, Callable, Union, IO
|
||
from flask import send_file, Response
|
||
|
||
from pywxdump.file.AttachmentAbstract import Attachment
|
||
from pywxdump.file.LocalAttachment import LocalAttachment
|
||
from pywxdump.file.S3Attachment import S3Attachment
|
||
|
||
|
||
def determine_strategy(file_path: str) -> Attachment:
|
||
if file_path.startswith("s3://"):
|
||
return S3Attachment()
|
||
else:
|
||
return LocalAttachment()
|
||
|
||
|
||
def exists(path: str) -> bool:
|
||
return determine_strategy(path).exists(path)
|
||
|
||
|
||
def open_file(path: str, mode: str) -> IO:
|
||
return determine_strategy(path).open(path, mode)
|
||
|
||
|
||
def makedirs(path: str) -> bool:
|
||
return determine_strategy(path).makedirs(path)
|
||
|
||
|
||
def join(__a: str, *paths: str) -> str:
|
||
return determine_strategy(__a).join(__a, *paths)
|
||
|
||
|
||
def dirname(path: str) -> str:
|
||
return determine_strategy(path).dirname(path)
|
||
|
||
|
||
def basename(path: str) -> str:
|
||
return determine_strategy(path).basename(path)
|
||
|
||
|
||
def send_attachment(
|
||
path_or_file: Union[os.PathLike[AnyStr], str],
|
||
mimetype: Union[str, None] = None,
|
||
as_attachment: bool = False,
|
||
download_name: Union[str, None] = None,
|
||
conditional: bool = True,
|
||
etag: Union[bool, str] = True,
|
||
last_modified: Union[datetime, int, float, None] = None,
|
||
max_age: Union[None, int, Callable[[Union[str, None]], Union[int, None]]] = None,
|
||
) -> Response:
|
||
file_io = open_file(path_or_file, "rb")
|
||
|
||
# 如果没有提供 download_name 或 mimetype，则从 path_or_file 中获取文件名和 MIME 类型
|
||
if download_name is None:
|
||
download_name = basename(path_or_file)
|
||
if mimetype is None:
|
||
mimetype = 'application/octet-stream'
|
||
|
||
return send_file(file_io, mimetype, as_attachment, download_name, conditional, etag, last_modified, max_age)
|
||
|
||
|
||
def download_file(db_path, local_path):
|
||
with open(local_path, 'wb') as f:
|
||
with open_file(db_path, 'rb') as r:
|
||
f.write(r.read())
|
||
return local_path
|
||
|
||
|
||
def isLocalPath(path: str) -> bool:
|
||
return isinstance(determine_strategy(path), LocalAttachment)
|
||
|