freedakgmail
104 lines
3.8 KiB
Python
104 lines
3.8 KiB
Python
"""R11 适配器：渠道佣金与代理商套利。
|
|
|
|
源明细：SrcTerminalBinding
|
|
映射到：Entity(IMEI, MSISDN) + 关系(BOUND_DEVICE) + MetricEvent
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import uuid
|
|
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.datahub.graph_repo import add_relationship, upsert_entity
|
|
from app.datahub.models import MetricEvent
|
|
from app.datahub.ontology import EntityType, RelationshipType
|
|
from app.datahub.staging import SrcTerminalBinding
|
|
from app.ingest.base import BaseAdapter, IngestResult
|
|
from app.ingest.registry import register_adapter
|
|
|
|
|
|
@register_adapter
|
|
class TerminalBindingAdapter(BaseAdapter):
|
|
"""SrcTerminalBinding → Entity(IMEI, MSISDN) + BOUND_DEVICE + MetricEvent。"""
|
|
|
|
source_system = "BSS"
|
|
staging_table = "src_terminal_binding"
|
|
|
|
def ingest(
|
|
self,
|
|
session: Session,
|
|
data_version_id: uuid.UUID | None = None,
|
|
batch_size: int = 1000,
|
|
) -> IngestResult:
|
|
result = IngestResult()
|
|
query = session.query(SrcTerminalBinding)
|
|
if data_version_id:
|
|
query = query.filter(SrcTerminalBinding.data_version_id == data_version_id)
|
|
rows = query.limit(batch_size).all()
|
|
|
|
for row in rows:
|
|
try:
|
|
# IMEI 实体
|
|
imei_entity = upsert_entity(
|
|
session,
|
|
entity_type=EntityType.IMEI,
|
|
business_key=row.imei,
|
|
display_name=row.brand_model or row.imei,
|
|
attributes={
|
|
"brand_model": row.brand_model,
|
|
"region": row.region,
|
|
},
|
|
data_version_id=data_version_id,
|
|
)
|
|
result.entities.append(imei_entity)
|
|
|
|
# MSISDN 实体
|
|
msisdn_entity = upsert_entity(
|
|
session,
|
|
entity_type=EntityType.MSISDN,
|
|
business_key=row.msisdn,
|
|
display_name=row.msisdn,
|
|
attributes={"region": row.region},
|
|
data_version_id=data_version_id,
|
|
)
|
|
result.entities.append(msisdn_entity)
|
|
|
|
# 绑定关系
|
|
rel = add_relationship(
|
|
session, RelationshipType.BOUND_DEVICE, msisdn_entity, imei_entity,
|
|
attributes={
|
|
"activate_time": str(row.activate_time) if row.activate_time else None,
|
|
"subsidy_amount": row.subsidy_amount,
|
|
},
|
|
data_version_id=data_version_id,
|
|
)
|
|
result.relationships.append(rel)
|
|
|
|
# 终端激活/补贴事件
|
|
if row.activate_time:
|
|
event = MetricEvent(
|
|
event_time=row.activate_time,
|
|
subject_type="imei",
|
|
subject_key=row.imei,
|
|
metric_name="terminal_activate",
|
|
metric_value=row.subsidy_amount + row.commission_amount,
|
|
attributes={
|
|
"msisdn": row.msisdn,
|
|
"subsidy_amount": row.subsidy_amount,
|
|
"commission_amount": row.commission_amount,
|
|
"online_days": row.online_days,
|
|
"post_activate_traffic_mb": row.post_activate_traffic_mb,
|
|
"cross_province_flag": row.cross_province_flag,
|
|
},
|
|
data_version_id=data_version_id,
|
|
)
|
|
session.add(event)
|
|
result.metric_events.append(event)
|
|
|
|
result.row_count += 1
|
|
except Exception:
|
|
result.error_count += 1
|
|
|
|
return result
|