Files
system-backu/backend/app/utils/crypto.py
T
publ 4e3a4b4602 feat: 数据备份管理系统 v0.1.0
带 Web 操作界面的备份系统,支持:
- 数据源:MySQL 数据库(mysqldump)、服务器目录(tar.gz)
- 存储目标:本地目录、S3 兼容对象存储(MinIO/Ceph/AWS S3)
- 触发方式:手动 + Cron 定时调度(APScheduler)
- 备份还原:MySQL 库、目录
- JWT 登录认证 + Fernet 字段加密
- 保留策略:按数量 + 按天数双重清理

技术栈:FastAPI + SQLAlchemy 2 + APScheduler + aioboto3;
前端 Vite + React 18 + TypeScript + Ant Design 5 + Zustand。
Docker Compose 一键起。

Co-Authored-By: Claude <noreply@anthropic.com>
2026-06-22 18:27:41 +08:00

47 lines
1.3 KiB
Python

"""Fernet 对称加密工具:用于加密存储敏感字段(MySQL 密码、S3 SecretKey)。"""
from cryptography.fernet import Fernet, InvalidToken
from app.config import get_settings
_fernet: Fernet | None = None
def _get_fernet() -> Fernet:
global _fernet
if _fernet is None:
_fernet = Fernet(get_settings().FERNET_KEY.encode())
return _fernet
def encrypt_str(plain: str) -> str:
"""加密字符串,返回 base64-urlsafe 字符串(可安全存数据库 Text 字段)。"""
if plain is None:
return ""
return _get_fernet().encrypt(plain.encode()).decode()
def decrypt_str(token: str) -> str:
"""解密字符串。空字符串返回空字符串。"""
if not token:
return ""
try:
return _get_fernet().decrypt(token.encode()).decode()
except InvalidToken as e:
raise ValueError(f"Failed to decrypt token (FERNET_KEY 可能已被更改): {e}") from e
def encrypt_dict(data: dict) -> str:
"""加密字典:序列化为 JSON 再加密。"""
import json
return encrypt_str(json.dumps(data, ensure_ascii=False))
def decrypt_dict(token: str) -> dict:
"""解密字典。"""
import json
if not token:
return {}
return json.loads(decrypt_str(token))