refactor: unify data preparations and research input snapshots
Deploy production / deploy (push) Successful in 53s
Deploy production / deploy (push) Successful in 53s
This commit is contained in:
@@ -106,8 +106,33 @@ class Backtests:
|
||||
"limits": "并发和批大小为本地配置,并非平台可用额度;账户外提交不在预算内",
|
||||
}
|
||||
|
||||
async def bind_preparations(self, body):
|
||||
from ..preparations.service import Preparations
|
||||
from ..research.expressions import analyze
|
||||
if not body.preparation_refs and not body.input_ids:
|
||||
return
|
||||
await Preparations(self.db).bind(body)
|
||||
snapshots = [await Preparations(self.db).snapshot(i) for i in body.input_ids]
|
||||
for candidate in body.candidates:
|
||||
scope = dict(instrument_type=candidate.settings.instrumentType, region=candidate.settings.region,
|
||||
universe=candidate.settings.universe, delay=candidate.settings.delay)
|
||||
if any(s["scope"] != scope for s in snapshots):
|
||||
raise HTTPException(422, "数据准备集合与回测范围不一致")
|
||||
fields = {}
|
||||
for snapshot in snapshots:
|
||||
for field, kind in snapshot["field_types"].items():
|
||||
if field in fields and fields[field] != kind:
|
||||
raise HTTPException(422, "输入字段类型冲突")
|
||||
fields[field] = kind
|
||||
validation = analyze(candidate.expression, fields)
|
||||
if validation["syntax"] or validation["types"]:
|
||||
raise HTTPException(422, ";".join(validation["syntax"] + validation["types"]))
|
||||
body.source.input_snapshot_ids = body.input_ids
|
||||
body.source.input_snapshot_id = body.input_ids[0] if len(body.input_ids) == 1 else None
|
||||
|
||||
async def save_draft(self, body, draft_id=None):
|
||||
data = body.model_dump(mode="json", exclude={"version"})
|
||||
await self.bind_preparations(body)
|
||||
data = body.model_dump(mode="json", exclude={"version", "preparation_refs", "input_ids"})
|
||||
if draft_id:
|
||||
changed = await self.db.execute(
|
||||
update(BacktestDraft)
|
||||
@@ -168,6 +193,7 @@ class Backtests:
|
||||
producer. ai_context separately identifies whoever starts the execution.
|
||||
"""
|
||||
if body.inline:
|
||||
await self.bind_preparations(body.inline)
|
||||
data = body.inline.model_dump(mode="json")
|
||||
if self.ai_context and not preserve_source:
|
||||
data["source"] = {
|
||||
|
||||
Reference in New Issue
Block a user