htlee
2ee8a0e7ff
- prediction: G-01/G-04/G-05/G-06 위반 분류 + 쌍끌이 공조 탐지 추가 - backend: 모선 확정/제외 API + signal-batch 항적 프록시 + ParentResolution 점수 근거 필드 확장 - frontend: 어구 탐지 그리드 다중필터/지도 flyTo, 후보 검토 패널(점수 근거+확정/제외), 24h convex hull 리플레이 + TripsLayer 애니메이션 - gitignore: 루트 .venv/ 추가
105 lines
3.7 KiB
Python
105 lines
3.7 KiB
Python
"""
|
|
위반 유형 라벨링 — 분석 결과에 violation_categories[] 태깅.
|
|
|
|
vessel_analysis_results의 각 행에 대해 5개 위반 카테고리를 판정하고
|
|
violation_categories TEXT[] 컬럼을 업데이트합니다.
|
|
"""
|
|
import logging
|
|
from psycopg2.extras import execute_batch
|
|
|
|
from config import qualified_table
|
|
from db.kcgdb import get_conn
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
VAR_TABLE = qualified_table('vessel_analysis_results')
|
|
|
|
|
|
def classify_violations(result: dict) -> list[str]:
|
|
"""단일 분석 결과에 대해 위반 유형 리스트 반환.
|
|
|
|
판정 기준:
|
|
- EEZ_VIOLATION: 중국선박(412*) + EEZ/NLL/특별금어구역 + 비허가
|
|
- DARK_VESSEL: is_dark + 30분 이상 갭
|
|
- MMSI_TAMPERING: spoofing_score > 0.6
|
|
- ILLEGAL_TRANSSHIP: transship_suspect
|
|
- RISK_BEHAVIOR: 위반 없이 risk_score >= 70
|
|
"""
|
|
violations = []
|
|
|
|
zone = result.get('zone_code', '') or ''
|
|
risk_score = result.get('risk_score', 0) or 0
|
|
is_dark = result.get('is_dark', False)
|
|
spoofing = result.get('spoofing_score', 0) or 0
|
|
transship = result.get('transship_suspect', False)
|
|
gap_min = result.get('gap_duration_min', 0) or 0
|
|
mmsi = str(result.get('mmsi', '') or '')
|
|
# permit_status는 선택적 — 없으면 중국 선박인지로 판단 (412* prefix)
|
|
permit = result.get('permit_status') or ''
|
|
is_chinese = mmsi.startswith('412') or mmsi.startswith('413')
|
|
|
|
# EEZ 침범: 중국선박이 한국 해역에 진입 (중국선박은 기본적으로 비허가 상정)
|
|
if zone in ('NLL', 'SPECIAL_FISHING_1', 'SPECIAL_FISHING_2',
|
|
'SPECIAL_FISHING_3', 'SPECIAL_FISHING_4', 'EEZ_KR'):
|
|
if is_chinese and permit not in ('VALID', 'PERMITTED'):
|
|
violations.append('EEZ_VIOLATION')
|
|
elif permit in ('NONE', 'EXPIRED', 'REVOKED'):
|
|
violations.append('EEZ_VIOLATION')
|
|
|
|
# 다크베셀
|
|
if is_dark and gap_min > 30:
|
|
violations.append('DARK_VESSEL')
|
|
|
|
# MMSI 변조
|
|
if spoofing > 0.6:
|
|
violations.append('MMSI_TAMPERING')
|
|
|
|
# 불법환적
|
|
if transship:
|
|
violations.append('ILLEGAL_TRANSSHIP')
|
|
|
|
# 어구 불법 (gear_judgment이 있는 경우만 — 현재는 scheduler에서 채우지 않음)
|
|
gear_judgment = result.get('gear_judgment', '') or ''
|
|
if gear_judgment in ('NO_PERMIT', 'GEAR_MISMATCH', 'ZONE_VIOLATION', 'SEASON_VIOLATION', 'PAIR_TRAWL'):
|
|
violations.append('ILLEGAL_GEAR')
|
|
|
|
# 위험 행동 (다른 위반 없이 고위험)
|
|
if not violations and risk_score >= 70:
|
|
violations.append('RISK_BEHAVIOR')
|
|
|
|
return violations
|
|
|
|
|
|
def run_violation_classifier(analysis_results: list[dict]) -> dict:
|
|
"""
|
|
분석 결과 리스트에 위반 카테고리를 라벨링하고 DB 업데이트.
|
|
|
|
AnalysisResult에는 DB id가 없으므로 (mmsi, analyzed_at)으로 UPDATE.
|
|
|
|
Returns:
|
|
{ 'classified': int, 'violations_found': int }
|
|
"""
|
|
updates = []
|
|
violations_found = 0
|
|
|
|
for result in analysis_results:
|
|
violations = classify_violations(result)
|
|
mmsi = result.get('mmsi')
|
|
analyzed_at = result.get('analyzed_at')
|
|
if mmsi and analyzed_at and violations:
|
|
updates.append((violations, str(mmsi), analyzed_at))
|
|
violations_found += len(violations)
|
|
|
|
if updates:
|
|
with get_conn() as conn:
|
|
execute_batch(
|
|
conn.cursor(),
|
|
f"UPDATE {VAR_TABLE} SET violation_categories = %s "
|
|
f"WHERE mmsi = %s AND analyzed_at = %s",
|
|
updates,
|
|
)
|
|
conn.commit()
|
|
|
|
logger.info(f'violation_classifier: classified={len(updates)}, violations={violations_found}')
|
|
return {'classified': len(updates), 'violations_found': violations_found}
|