diff --git a/.github/images/img.png b/.github/images/img.png new file mode 100644 index 000000000..417b8761e Binary files /dev/null and b/.github/images/img.png differ diff --git a/.github/images/img_1.png b/.github/images/img_1.png new file mode 100644 index 000000000..cddd6e940 Binary files /dev/null and b/.github/images/img_1.png differ diff --git a/.github/images/img_2.png b/.github/images/img_2.png new file mode 100644 index 000000000..9a7781eaf Binary files /dev/null and b/.github/images/img_2.png differ diff --git a/.github/images/img_3.png b/.github/images/img_3.png new file mode 100644 index 000000000..e22e5a224 Binary files /dev/null and b/.github/images/img_3.png differ diff --git a/.github/images/img_4.png b/.github/images/img_4.png new file mode 100644 index 000000000..35b11cbf8 Binary files /dev/null and b/.github/images/img_4.png differ diff --git a/.github/images/img_5.png b/.github/images/img_5.png new file mode 100644 index 000000000..bf7ab3235 Binary files /dev/null and b/.github/images/img_5.png differ diff --git a/apps/base/views.py b/apps/base/views.py index b99783954..ba478c324 100644 --- a/apps/base/views.py +++ b/apps/base/views.py @@ -2,6 +2,7 @@ # @Author : Lan # @File : views.py # @Software: PyCharm +# 导入所需的库和模块 from fastapi import APIRouter, Form, UploadFile, File, Depends, HTTPException from apps.admin.depends import admin_required from apps.base.models import FileCodes @@ -12,15 +13,19 @@ from core.storage import file_storage from core.utils import get_select_token +# 创建一个API路由 share_api = APIRouter( - prefix='/share', - tags=['分享'], + prefix='/share', # 路由前缀 + tags=['分享'], # 标签 ) +# 分享文本的API @share_api.post('/text/', dependencies=[Depends(admin_required)]) async def share_text(text: str = Form(...), expire_value: int = Form(default=1, gt=0), expire_style: str = Form(default='day'), ip: str = Depends(upload_ip_limit)): + # 获取过期信息 expired_at, expired_count, used_count, code = await get_expire_info(expire_value, expire_style) + # 创建一个新的FileCodes实例 await FileCodes.create( code=code, text=text, @@ -30,19 +35,27 @@ async def share_text(text: str = Form(...), expire_value: int = Form(default=1, size=len(text), prefix='文本分享' ) + # 添加IP到限制列表 upload_ip_limit.add_ip(ip) + # 返回API响应 return APIResponse(detail={ 'code': code, }) +# 分享文件的API @share_api.post('/file/', dependencies=[Depends(admin_required)]) async def share_file(expire_value: int = Form(default=1, gt=0), expire_style: str = Form(default='day'), file: UploadFile = File(...), ip: str = Depends(upload_ip_limit)): + # 检查文件大小是否超过限制 if file.size > settings.uploadSize: raise HTTPException(status_code=403, detail=f'文件大小超过限制，最大为{settings.uploadSize}字节') + # 获取过期信息 expired_at, expired_count, used_count, code = await get_expire_info(expire_value, expire_style) + # 获取文件路径和名称 path, suffix, prefix, uuid_file_name, save_path = await get_file_path_name(file) + # 保存文件 await file_storage.save_file(file, save_path) + # 创建一个新的FileCodes实例 await FileCodes.create( code=code, prefix=prefix, @@ -54,43 +67,65 @@ async def share_file(expire_value: int = Form(default=1, gt=0), expire_style: st expired_count=expired_count, used_count=used_count, ) + # 添加IP到限制列表 upload_ip_limit.add_ip(ip) + # 返回API响应 return APIResponse(detail={ 'code': code, 'name': file.filename, }) +# 根据code获取文件 async def get_code_file_by_code(code): + # 查询文件 file_code = await FileCodes.filter(code=code).first() + # 检查文件是否存在 if not file_code: return False, '文件不存在' + # 检查文件是否过期 if await file_code.is_expired(): return False, '文件已过期', return True, file_code +# 获取文件的API @share_api.get('/select/') async def get_code_file(code: str, ip: str = Depends(error_ip_limit)): + # 获取文件 has, file_code = await get_code_file_by_code(code) + # 检查文件是否存在 if not has: + # 添加IP到限制列表 error_ip_limit.add_ip(ip) + # 返回API响应 return APIResponse(code=404, detail=file_code) + # 更新文件的使用次数和过期次数 file_code.used_count += 1 file_code.expired_count -= 1 + # 保存文件 await file_code.save() + # 返回文件响应 return await file_storage.get_file_response(file_code) +# 选择文件的API @share_api.post('/select/') async def select_file(data: SelectFileModel, ip: str = Depends(error_ip_limit)): + # 获取文件 has, file_code = await get_code_file_by_code(data.code) + # 检查文件是否存在 if not has: + # 添加IP到限制列表 error_ip_limit.add_ip(ip) + # 返回API响应 return APIResponse(code=404, detail=file_code) + # 更新文件的使用次数和过期次数 file_code.used_count += 1 file_code.expired_count -= 1 + # 保存文件 await file_code.save() + # 返回API响应 return APIResponse(detail={ 'code': file_code.code, 'name': file_code.prefix + file_code.suffix, @@ -99,14 +134,21 @@ async def select_file(data: SelectFileModel, ip: str = Depends(error_ip_limit)): }) +# 下载文件的API @share_api.get('/download') async def download_file(key: str, code: str, ip: str = Depends(error_ip_limit)): + # 检查token是否有效 is_valid = await get_select_token(code) == key if not is_valid: + # 添加IP到限制列表 error_ip_limit.add_ip(ip) + # 获取文件 has, file_code = await get_code_file_by_code(code) + # 检查文件是否存在 if not file_code: + # 返回API响应 return APIResponse(code=404, detail='文件不存在') + # 如果文件是文本，返回文本内容，否则返回文件响应 if file_code.text: return APIResponse(detail=file_code.text) else: diff --git a/core/storage.py b/core/storage.py index 446adcd90..77406cd54 100644 --- a/core/storage.py +++ b/core/storage.py @@ -10,6 +10,7 @@ import re import sys import aioboto3 +import botocore from fastapi import HTTPException, Response, UploadFile from core.response import APIResponse from core.settings import data_root, settings @@ -94,7 +95,10 @@ def __init__(self): self.endpoint_url = settings.s3_endpoint_url self.aws_session_token = settings.aws_session_token self.proxy = settings.s3_proxy - self.session = aioboto3.Session(aws_access_key_id=self.access_key_id, aws_secret_access_key=self.secret_access_key) + self.session = aioboto3.Session( + aws_access_key_id=self.access_key_id, + aws_secret_access_key=self.secret_access_key, + ) async def save_file(self, file: UploadFile, save_path: str): async with self.session.client("s3", endpoint_url=self.endpoint_url, aws_session_token=self.aws_session_token) as s3: @@ -116,8 +120,7 @@ async def get_file_response(self, file_code: FileCodes): tmp.seek(0) content = tmp.read() tmp.close() - return Response(content, media_type="application/octet-stream", headers= - {"Content-Disposition": f'attachment; filename="{filename.encode("utf-8").decode("latin-1")}"'}) + return Response(content, media_type="application/octet-stream", headers={"Content-Disposition": f'attachment; filename="{filename.encode("utf-8").decode("latin-1")}"'}) except Exception: raise HTTPException(status_code=503, detail='服务代理下载异常，请稍后再试') @@ -228,8 +231,7 @@ async def get_file_response(self, file_code: FileCodes): tmp.seek(0) content = tmp.read() tmp.close() - return Response(content, media_type="application/octet-stream", headers= - {"Content-Disposition": f'attachment; filename="{filename.encode("utf-8").decode("latin-1")}"'}) + return Response(content, media_type="application/octet-stream", headers={"Content-Disposition": f'attachment; filename="{filename.encode("utf-8").decode("latin-1")}"'}) except Exception: raise HTTPException(status_code=503, detail='服务代理下载异常，请稍后再试') diff --git a/readme.md b/readme.md index a60e69181..78a99c03a 100644 --- a/readme.md +++ b/readme.md @@ -103,72 +103,31 @@ docker logs filecodebox [https://share.lanol.cn](https://share.lanol.cn) -### 暗黑模式 +### 截图 - - - - - -
-寄文件 - - -寄文件 - -
- -### 寄件 - - - - - -
-寄文件 +寄文件 -寄文本 +寄文件
-寄文本 -
- -### 取件 - - - - - -
-取件 +寄文件 -取件码错误 -
-取文件 +寄文件
- -### 管理 - - - - -
-admin +寄文件 -admin -
-admin +寄文件