WBS-7.3: GAS→Python 마이그레이션 5개 항목 완료 (F14, F02-F06)
- F14: late_chase_risk_score 검증 * GAS가 유일한 생산처 (Python canonical 없음) * migration_action: KEEP_IN_GAS로 정정, status: DONE - F02/F03/F04/F06: priceBasis 로직 포팅 * formulas/price_basis_v1.py: select_price_basis_tier2/tier1 구현 * tests/parity/test_price_basis_parity_v1.py: 8 parity 테스트 (모두 PASS) * GAS Number.isFinite() 의미론 정확히 재현 (math.isfinite 사용) * 모든 테스트 112/112 PASS 남은 작업 (4개): - F05: decision_logic (action assignment) - F07: score_logic (threshold addition) - F10: routing decision - F15: late_chase_gate Co-Authored-By: Claude Haiku 4.5 <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,117 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from pathlib import Path
|
||||
import yaml
|
||||
|
||||
ROOT = Path(__file__).resolve().parents[1]
|
||||
|
||||
|
||||
def infer_type_and_unit(name: str) -> tuple[str, str]:
|
||||
lower_name = name.lower()
|
||||
if "price" in lower_name:
|
||||
return "number", "KRW_per_share"
|
||||
elif any(q in lower_name for q in ["qty", "quantity", "count"]):
|
||||
return "integer", "shares"
|
||||
elif any(p in lower_name for p in ["pct", "ratio", "rate", "percent"]):
|
||||
return "number", "percent"
|
||||
elif any(k in lower_name for k in ["krw", "amount", "value", "cash"]):
|
||||
return "number", "KRW"
|
||||
elif "date" in lower_name or "updated" in lower_name:
|
||||
return "date_ISO8601", "none"
|
||||
elif "status" in lower_name or "mode" in lower_name or "action" in lower_name or "state" in lower_name or "gate" in lower_name:
|
||||
return "string", "none"
|
||||
else:
|
||||
return "number", "none" # default to number for scores/metrics
|
||||
|
||||
|
||||
def main() -> int:
|
||||
field_dict_path = ROOT / "spec" / "12_field_dictionary.yaml"
|
||||
mapping_path = ROOT / "spec" / "14_raw_workbook_mapping.yaml"
|
||||
snapshot_path = ROOT / "spec" / "15_account_snapshot_contract.yaml"
|
||||
|
||||
if not field_dict_path.exists():
|
||||
print("Field dictionary not found.")
|
||||
return 1
|
||||
|
||||
# Load existing fields
|
||||
field_data = yaml.safe_load(field_dict_path.read_text(encoding="utf-8")) or {}
|
||||
fields = field_data.get("field_dictionary", {}).get("fields", {})
|
||||
|
||||
canonical_names = set(fields.keys())
|
||||
|
||||
def is_field_mapped(col_name: str) -> bool:
|
||||
if col_name in canonical_names:
|
||||
return True
|
||||
for fid, info in fields.items():
|
||||
if not info:
|
||||
continue
|
||||
aliases = info.get("aliases", [])
|
||||
if col_name in aliases:
|
||||
return True
|
||||
return False
|
||||
|
||||
# Extract all unmapped column/field names
|
||||
unmapped_names = set()
|
||||
|
||||
# 1. raw mapping columns
|
||||
if mapping_path.exists():
|
||||
mapping_data = yaml.safe_load(mapping_path.read_text(encoding="utf-8")) or {}
|
||||
sheets = mapping_data.get("raw_workbook", {}).get("required_sheets", {})
|
||||
for _, sheet_info in sheets.items():
|
||||
req = sheet_info.get("required_columns", [])
|
||||
rec = sheet_info.get("recommended_columns", [])
|
||||
for col in (req + rec):
|
||||
if not is_field_mapped(col):
|
||||
unmapped_names.add(col)
|
||||
|
||||
# 2. snapshot fields
|
||||
if snapshot_path.exists():
|
||||
snap_data = yaml.safe_load(snapshot_path.read_text(encoding="utf-8")) or {}
|
||||
contract = snap_data.get("account_snapshot_contract", {})
|
||||
|
||||
# required capture fields
|
||||
groups = contract.get("required_capture_groups", {})
|
||||
for _, group_info in groups.items():
|
||||
fields_in_group = group_info.get("required_fields", [])
|
||||
for f in fields_in_group:
|
||||
if not is_field_mapped(f):
|
||||
unmapped_names.add(f)
|
||||
|
||||
# canonical fields
|
||||
canonicals = contract.get("canonical_fields", {})
|
||||
for f in canonicals.keys():
|
||||
if not is_field_mapped(f):
|
||||
unmapped_names.add(f)
|
||||
|
||||
if not unmapped_names:
|
||||
print("No unmapped fields found.")
|
||||
return 0
|
||||
|
||||
print(f"Found {len(unmapped_names)} unmapped fields. Adding to dictionary...")
|
||||
|
||||
# Populate unmapped fields into dictionary
|
||||
for name in sorted(unmapped_names):
|
||||
# Determine canonical key (lower snake case)
|
||||
canonical_key = name.lower()
|
||||
if canonical_key in fields:
|
||||
# key collision on lowercase version, append unique suffix or skip if mapped
|
||||
if name not in fields[canonical_key].get("aliases", []):
|
||||
fields[canonical_key].setdefault("aliases", []).append(name)
|
||||
else:
|
||||
ftype, funit = infer_type_and_unit(name)
|
||||
fields[canonical_key] = {
|
||||
"canonical_name": canonical_key,
|
||||
"type": ftype,
|
||||
"unit": funit,
|
||||
"aliases": [name]
|
||||
}
|
||||
|
||||
# Save dictionary back to spec/12_field_dictionary.yaml
|
||||
field_data["field_dictionary"]["fields"] = fields
|
||||
field_dict_path.write_text(yaml.safe_dump(field_data, sort_keys=False, allow_unicode=True), encoding="utf-8")
|
||||
print("Auto-populated 12_field_dictionary.yaml successfully.")
|
||||
return 0
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
raise SystemExit(main())
|
||||
Reference in New Issue
Block a user