From f1cdcf716c7d86b2fbabc6501fa7f9cd9acc7ed6 Mon Sep 17 00:00:00 2001 From: gouki Date: Mon, 3 Aug 2026 00:12:38 +0000 Subject: [PATCH] =?UTF-8?q?feat:=20=E6=95=B0=E6=8D=AE=E5=AF=BC=E5=87=BA/?= =?UTF-8?q?=E5=AF=BC=E5=85=A5=EF=BC=88=E8=B5=84=E4=BA=A7=E8=BF=81=E7=A7=BB?= =?UTF-8?q?/=E5=A4=87=E4=BB=BD=EF=BC=8C=E6=94=AF=E6=92=91=E9=83=A8?= =?UTF-8?q?=E7=BD=B2=E5=88=B0=E5=85=B6=E4=BB=96=E6=9C=8D=E5=8A=A1=E5=99=A8?= =?UTF-8?q?=EF=BC=89?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/routers/assets.py | 18 ++++++++++++++- app/services/asset_service.py | 43 ++++++++++++++++++++++++++++++++++- static/js/app.js | 41 ++++++++++++++++++++++++++++++++- 3 files changed, 99 insertions(+), 3 deletions(-) diff --git a/app/routers/assets.py b/app/routers/assets.py index 84e4023..d863d5d 100644 --- a/app/routers/assets.py +++ b/app/routers/assets.py @@ -5,7 +5,7 @@ from typing import List, Optional -from fastapi import APIRouter, Depends, Query +from fastapi import APIRouter, Body, Depends, Query from sqlmodel import Session from app.core.security import require_api_key @@ -32,6 +32,22 @@ def list_assets( ) +@router.get("/export", summary="导出所有资产（数据迁移/备份）") +def export_assets(session: Session = Depends(get_session)) -> dict: + return asset_service.export_assets(session) + + +@router.post( + "/import", + summary="导入资产（从导出数据，按 name+类型+服务商去重）", + dependencies=[Depends(require_api_key)], +) +def import_assets_endpoint( + data: dict = Body(...), session: Session = Depends(get_session) +) -> dict: + return asset_service.import_assets(session, data.get("assets", [])) + + @router.get("/{asset_id}", response_model=AssetRead, summary="资产详情") def get_asset(asset_id: int, session: Session = Depends(get_session)): return asset_service.get_asset(session, asset_id) diff --git a/app/services/asset_service.py b/app/services/asset_service.py index 0b544dc..a7b6a21 100644 --- a/app/services/asset_service.py +++ b/app/services/asset_service.py @@ -3,7 +3,7 @@ 统一处理 Asset 主表与其一对一详情表（VPSDetail/DomainDetail/AIAccount）的联动。 """ -from datetime import date +from datetime import date, datetime from typing import List, Optional from fastapi import HTTPException, status @@ -235,6 +235,47 @@ def delete_asset(session: Session, asset_id: int) -> None: session.commit() +def export_assets(session: Session) -> dict: + """导出所有资产（含 detail）为 JSON，用于数据迁移/备份""" + assets = session.exec(select(Asset)).all() + result = [] + for asset in assets: + read = _to_read(session, asset, _get_detail(session, asset)) + result.append(read.model_dump(mode="json")) + return { + "count": len(result), + "exported_at": datetime.utcnow().isoformat(), + "assets": result, + } + + +def import_assets(session: Session, assets_data: list) -> dict: + """从导出数据导入资产（按 name+asset_type+provider 去重：存在则更新，不存在则创建）""" + created = 0 + updated = 0 + errors = [] + skip_fields = {"id", "created_at", "updated_at", "days_to_expiry", "provider_name"} + for item in assets_data: + try: + payload = {k: v for k, v in item.items() if k not in skip_fields} + existing = session.exec( + select(Asset).where( + Asset.name == payload.get("name"), + Asset.asset_type == payload.get("asset_type"), + Asset.provider == payload.get("provider"), + ) + ).first() + if existing: + update_asset(session, existing.id, AssetUpdate(**payload)) + updated += 1 + else: + create_asset(session, AssetCreate(**payload)) + created += 1 + except Exception as e: # noqa: BLE001 + errors.append(f"{item.get('name', '?')}: {e}") + return {"created": created, "updated": updated, "errors": errors} + + def list_assets( session: Session, asset_type: Optional[AssetType] = None, diff --git a/static/js/app.js b/static/js/app.js index 9c883bc..76d470e 100644 --- a/static/js/app.js +++ b/static/js/app.js @@ -761,10 +761,20 @@ const SettingsView = { 深色模式 +
+

数据管理（迁移/备份）

+

导出所有资产为 JSON，可导入到新部署的实例，实现数据迁移。

+
+ + +
+
{{ dataResult }}
+
`, setup() { const notifyChannels = ref([]); const notifyResult = ref(''); + const dataResult = ref(''); onMounted(async () => { try { const r = await Api.get('/notify/channels'); notifyChannels.value = r.channels || []; } catch (e) { /* ignore */ } }); @@ -785,7 +795,36 @@ const SettingsView = { notifyResult.value = text; } catch (e) { notifyResult.value = '失败：' + e.message; } } - return { store, settings, save: saveSettings, applyDark, notifyChannels, notifyResult, testNotify, checkRenewals }; + async function exportData() { + dataResult.value = '导出中…'; + try { + const r = await Api.get('/assets/export'); + const blob = new Blob([JSON.stringify(r, null, 2)], { type: 'application/json' }); + const url = URL.createObjectURL(blob); + const a = document.createElement('a'); + a.href = url; + a.download = 'vps-manager-export-' + new Date().toISOString().slice(0, 10) + '.json'; + a.click(); + URL.revokeObjectURL(url); + dataResult.value = '导出成功：' + r.count + ' 个资产'; + } catch (e) { dataResult.value = '导出失败：' + e.message; } + } + function onImportFile(e) { + const file = e.target.files[0]; + if (!file) return; + const reader = new FileReader(); + reader.onload = async (ev) => { + dataResult.value = '导入中…'; + try { + const data = JSON.parse(ev.target.result); + const r = await Api.post('/assets/import', data); + dataResult.value = '导入完成：新增 ' + r.created + '，更新 ' + r.updated + (r.errors && r.errors.length ? '，错误 ' + r.errors.length : ''); + await loadAssets(); + } catch (err) { dataResult.value = '导入失败：' + err.message; } + }; + reader.readAsText(file); + } + return { store, settings, save: saveSettings, applyDark, notifyChannels, notifyResult, testNotify, checkRenewals, dataResult, exportData, onImportFile }; }, };