mirror of
https://github.com/RhenCloud/Cloud-Index.git
synced 2025-12-06 15:26:10 +08:00
141 lines
3.7 KiB
Python
141 lines
3.7 KiB
Python
import os
|
||
import tomllib
|
||
from pathlib import Path
|
||
|
||
import dotenv
|
||
from flask import Flask
|
||
|
||
from handlers.routes import main_route
|
||
from storages.factory import StorageFactory
|
||
|
||
dotenv.load_dotenv()
|
||
|
||
|
||
# 从 pyproject.toml 读取版本号
|
||
def get_version() -> str:
|
||
"""从 pyproject.toml 读取项目版本号"""
|
||
try:
|
||
pyproject_path = Path(__file__).parent / "pyproject.toml"
|
||
with open(pyproject_path, "rb") as f:
|
||
pyproject_data = tomllib.load(f)
|
||
return pyproject_data.get("project", {}).get("version", "0.0.0")
|
||
except Exception:
|
||
return "0.0.0"
|
||
|
||
|
||
__version__ = get_version()
|
||
|
||
app = Flask(__name__)
|
||
|
||
|
||
# 注册蓝图
|
||
app.register_blueprint(main_route)
|
||
|
||
# 初始化存储（使用工厂模式）
|
||
storage = StorageFactory.get_storage()
|
||
|
||
# 缩略图默认 TTL（秒），可通过环境变量覆盖
|
||
THUMB_TTL = int(os.environ.get("THUMB_TTL_SECONDS", "3600"))
|
||
|
||
|
||
# 注册一个安全的 filesizeformat 过滤器，处理 None 和非数字值
|
||
@app.template_filter("filesizeformat")
|
||
def filesizeformat_filter(value):
|
||
try:
|
||
if value is None:
|
||
return "-"
|
||
num = float(value) # 使用 float 而不是 int 以保持精度
|
||
except Exception:
|
||
return "-"
|
||
|
||
for unit in ["B", "KB", "MB", "GB", "TB"]:
|
||
if num < 1024:
|
||
# 对于字节，显示整数
|
||
if unit == "B":
|
||
return f"{int(num)}{unit}"
|
||
# 其他单位保留两位小数
|
||
return f"{num:.2f}{unit}"
|
||
num = num / 1024.0
|
||
return f"{num:.2f}PB"
|
||
|
||
|
||
# 注册一个文件图标过滤器
|
||
@app.template_filter("fileicon")
|
||
def fileicon_filter(filename):
|
||
if not filename:
|
||
return "fas fa-file"
|
||
|
||
ext = filename.lower().split(".")[-1] if "." in filename else ""
|
||
|
||
# 图片文件
|
||
if ext in ["jpg", "jpeg", "png", "gif", "bmp", "webp", "svg"]:
|
||
return "fas fa-image"
|
||
|
||
# 音频文件
|
||
if ext in ["mp3", "wav", "ogg", "flac", "m4a", "aac"]:
|
||
return "fas fa-music"
|
||
|
||
# 视频文件
|
||
if ext in ["mp4", "webm", "avi", "mov", "wmv", "flv", "mkv"]:
|
||
return "fas fa-video"
|
||
|
||
# 文档文件
|
||
if ext in ["pdf", "doc", "docx", "txt", "md", "rtf"]:
|
||
return "fas fa-file-alt"
|
||
|
||
# 压缩文件
|
||
if ext in ["zip", "rar", "7z", "tar", "gz"]:
|
||
return "fas fa-file-archive"
|
||
|
||
# 代码文件
|
||
if ext in ["py", "js", "html", "css", "java", "cpp", "c", "php"]:
|
||
return "fas fa-file-code"
|
||
|
||
# 表格文件
|
||
if ext in ["xls", "xlsx", "csv"]:
|
||
return "fas fa-file-excel"
|
||
|
||
# 演示文件
|
||
if ext in ["ppt", "pptx"]:
|
||
return "fas fa-file-powerpoint"
|
||
|
||
# 默认文件图标
|
||
return "fas fa-file"
|
||
|
||
|
||
def get_public_url(key: str) -> str:
|
||
"""
|
||
生成对象的公共访问 URL
|
||
"""
|
||
return storage.get_public_url(key)
|
||
|
||
|
||
def format_timestamp(timestamp) -> str:
|
||
"""
|
||
格式化时间戳为人类可读的格式
|
||
"""
|
||
return storage.format_timestamp(timestamp)
|
||
|
||
|
||
def generate_presigned_url(s3_client, bucket_name: str, key: str, expires: int = None) -> str:
|
||
"""为指定对象生成 presigned URL（GET）。"""
|
||
return storage.generate_presigned_url(key, expires)
|
||
|
||
|
||
def get_file_url(key: str) -> str:
|
||
"""生成通过服务器访问文件的 URL"""
|
||
return f"/file/{key}"
|
||
|
||
|
||
# 注册全局模板变量
|
||
@app.context_processor
|
||
def inject_version():
|
||
"""向所有模板注入版本号"""
|
||
return {"app_version": __version__}
|
||
|
||
|
||
if __name__ == "__main__":
|
||
port = int(os.environ.get("PORT", 5000))
|
||
host = os.environ.get("HOST", "0.0.0.0")
|
||
app.run(host=host, port=port, debug=True)
|