mirror of
https://github.com/RhenCloud/Cloud-Index.git
synced 2025-12-06 07:06:41 +08:00
85 lines
2.0 KiB
Python
85 lines
2.0 KiB
Python
import tomllib
|
||
from pathlib import Path
|
||
|
||
from flask import Flask
|
||
|
||
import utils
|
||
from config import Config
|
||
from handlers.routes import main_route
|
||
from storages.factory import StorageFactory
|
||
|
||
# 验证配置
|
||
Config.validate()
|
||
|
||
|
||
# 从 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()
|
||
|
||
|
||
# 注册模板过滤器
|
||
@app.template_filter("filesizeformat")
|
||
def filesizeformat_filter(value):
|
||
"""格式化文件大小"""
|
||
return utils.format_file_size(value)
|
||
|
||
|
||
@app.template_filter("fileicon")
|
||
def fileicon_filter(filename):
|
||
"""获取文件图标"""
|
||
return utils.get_file_icon(filename)
|
||
|
||
|
||
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__":
|
||
app.run(host=Config.HOST, port=Config.PORT, debug=Config.DEBUG)
|