mirror of
https://github.com/fawney19/Aether.git
synced 2026-09-02 09:20:22 +08:00
1d9c77522a
- 删除全部 Python 源码 (src/) 及 Alembic 迁移脚本，归档至 _deprecated_py_src/ - 重构 Rust gateway ai_pipeline: 拆分 planner/finalize 模块，新增 contracts/adaptation 层 - 重组 handlers 模块为 admin/public/proxy/internal/shared 子模块结构 - 新增 executor 模块，引入 Rust 原生数据库迁移 (aether-data/migrations) - 简化 CI/Docker 构建流程，移除 base image 二级构建，统一为单一 app image - 移除 Python 相关基础设施文件 (entrypoint.sh, gunicorn_conf.py, Dockerfile.base)
53 lines
1.4 KiB
Python
53 lines
1.4 KiB
Python
"""OAuth Key 导出：provider-specific export builders.
|
||
|
||
每个 Provider adapter 在 ``register_all()`` 时注册自己的 ``build_export_data``，
|
||
导出端点通过 ``build_export_data()`` 分发。
|
||
|
||
未注册的 provider_type 使用默认实现（strip null + 临时字段）。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from typing import Any, Callable
|
||
|
||
# 所有 provider 共用的临时/无用字段
|
||
_DEFAULT_SKIP_KEYS = frozenset(
|
||
{
|
||
"access_token",
|
||
"expires_at",
|
||
"updated_at",
|
||
"token_type",
|
||
"scope",
|
||
}
|
||
)
|
||
|
||
ExportBuilder = Callable[[dict[str, Any], dict[str, Any] | None], dict[str, Any]]
|
||
|
||
_BUILDERS: dict[str, ExportBuilder] = {}
|
||
|
||
|
||
def register_export_builder(provider_type: str, builder: ExportBuilder) -> None:
|
||
_BUILDERS[provider_type] = builder
|
||
|
||
|
||
def _default_builder(
|
||
auth_config: dict[str, Any],
|
||
upstream_metadata: dict[str, Any] | None,
|
||
) -> dict[str, Any]:
|
||
"""默认导出：去掉 null、空字符串、临时字段。"""
|
||
return {
|
||
k: v
|
||
for k, v in auth_config.items()
|
||
if k not in _DEFAULT_SKIP_KEYS and v is not None and v != ""
|
||
}
|
||
|
||
|
||
def build_export_data(
|
||
provider_type: str,
|
||
auth_config: dict[str, Any],
|
||
upstream_metadata: dict[str, Any] | None,
|
||
) -> dict[str, Any]:
|
||
"""调用 provider-specific builder 构建导出数据。"""
|
||
builder = _BUILDERS.get(provider_type, _default_builder)
|
||
return builder(auth_config, upstream_metadata)
|