add:支持使用阿里云OSS进行对象存储服务

fix:删除utils中未使用的第三方库chardet
This commit is contained in:
ZHYCarge
2023-01-15 22:06:01 +08:00
parent a142242bca
commit f87571e693
7 changed files with 93 additions and 10 deletions
+52 -3
View File
@@ -3,10 +3,58 @@ import asyncio
from datetime import datetime
from pathlib import Path
from typing import BinaryIO
from fastapi import UploadFile
import settings
import oss2
class AliyunFileStore:
auth = oss2.Auth(settings.KeyId, settings.KeySecret)
bucket = oss2.Bucket(auth, settings.OSS_ENDPOINT, settings.BUCKET_NAME)
def upload_file(self, upload_filepath, remote_filepath):
self.bucket.put_object_from_file(remote_filepath, upload_filepath)
async def get_text(self, file: UploadFile, key: str):
ext = file.filename.split('.')[-1]
now = datetime.now()
path = f"FileCodeBox/upload/{now.year}/{now.month}/{now.day}"
text = f"{path}/{f'{key}.{ext}'}"
return text
async def get_filepath(self, text: str):
return text
@staticmethod
async def get_size(file: UploadFile):
f = file.file
f.seek(0, os.SEEK_END)
size = f.tell()
f.seek(0, os.SEEK_SET)
return size
@staticmethod
def _save(filepath, file: BinaryIO):
with open(filepath, 'wb') as f:
chunk_size = 256 * 1024
chunk = file.read(chunk_size)
while chunk:
f.write(chunk)
chunk = file.read(chunk_size)
async def save_file(self, file: UploadFile, remote_filepath: str):
now = int(datetime.now().timestamp())
upload_filepath = settings.DATA_ROOT+str(now)
await asyncio.to_thread(self._save, upload_filepath, file.file)
self.upload_file(upload_filepath,remote_filepath)
await asyncio.to_thread(os.remove,upload_filepath)
async def delete_files(self, texts):
tasks = [self.delete_file(text) for text in texts]
await asyncio.gather(*tasks)
async def delete_file(self, text: str):
self.bucket.delete_object(text)
class FileSystemStorage:
@@ -68,5 +116,6 @@ class FileSystemStorage:
STORAGE_ENGINE = {
"filesystem": FileSystemStorage
"filesystem": FileSystemStorage,
"aliyunsystem": AliyunFileStore
}