zioinfo-mail/itsm/core/push_notify.py
DESKTOP-TKLFCPR\ython e228faabf5 feat(itsm): G-1~G-12 확장 기능 + 하네스/봇/설치스크립트 구현
G-1: 메신저 Webhook Relay + _send_to_room 실제 httpx 호출 구현
G-2: POST /api/tasks/bulk SR 대량작업 엔드포인트 (최대 100건)
G-3: 라이선스 만료 알림 스케줄러 (매일 09:00 KST)
G-4: 체험판 upgrade_banner 필드 + license.py 배너 로직
G-5: core/auto_rca.py + incidents/problem auto-rca 엔드포인트
G-6: core/deploy_impact.py + vibe impact-analysis 엔드포인트
G-7: core/ticket_classifier.py + SR 생성 시 AI 분류 + ai-suggestion API
G-8: VulnPatchRecord 모델 + vuln_scan 패치추적 4개 엔드포인트
G-9: core/jira_sync.py + gateway Jira/Confluence 연동 엔드포인트
G-10: core/push_notify.py + routers/push.py + PushSubscription 모델
G-11: approvals 다중승인 (위임/서명/기한초과/마감연장)
G-12: alembic.ini + migrations/ + cicd/migrate_to_postgres.sh

하네스: guardia-orchestrator 확장기능 Phase 반영
봇명령어: /sr /status /license /bulk 슬래시 명령어 추가
설치스크립트: setup/ (Ubuntu, CentOS, RHEL, Windows) --test 옵션 포함

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-05-29 18:18:52 +09:00

94 lines
2.6 KiB
Python

"""PWA Web Push 알림 — pywebpush 라이브러리 활용.
환경변수:
VAPID_PRIVATE_KEY: VAPID 개인키 (base64url 또는 PEM)
VAPID_PUBLIC_KEY: VAPID 공개키 (브라우저 subscribe 시 사용)
VAPID_CLAIMS_EMAIL: 관리자 이메일 (mailto: claim)
"""
from __future__ import annotations
import json
import logging
import os
from typing import Optional
logger = logging.getLogger(__name__)
VAPID_PRIVATE_KEY = os.getenv("VAPID_PRIVATE_KEY", "")
VAPID_PUBLIC_KEY = os.getenv("VAPID_PUBLIC_KEY", "")
VAPID_EMAIL = os.getenv("VAPID_CLAIMS_EMAIL", "admin@guardia.local")
async def send_push(
subscription_info: dict,
title: str,
body: str,
url: str = "/",
icon: str = "/static/icons/icon-192.png",
) -> bool:
"""단일 푸시 구독에 알림 전송. 성공 시 True."""
if not VAPID_PRIVATE_KEY:
logger.debug("VAPID_PRIVATE_KEY 미설정 — 푸시 알림 스킵")
return False
try:
from pywebpush import webpush, WebPushException
webpush(
subscription_info=subscription_info,
data=json.dumps({
"title": title,
"body": body,
"url": url,
"icon": icon,
}),
vapid_private_key=VAPID_PRIVATE_KEY,
vapid_claims={"sub": f"mailto:{VAPID_EMAIL}"},
)
return True
except Exception as e:
logger.warning("푸시 알림 전송 실패: %s", str(e)[:100])
return False
async def broadcast_push(
title: str,
body: str,
url: str = "/",
user_ids: Optional[list] = None,
) -> dict:
"""
여러 사용자에게 Push 브로드캐스트.
user_ids=None 이면 전체 구독자에게 발송.
Returns: {"sent": int, "failed": int}
"""
from database import SessionLocal
from models import PushSubscription
from sqlalchemy import select
sent = 0
failed = 0
try:
async with SessionLocal() as db:
q = select(PushSubscription)
if user_ids:
q = q.where(PushSubscription.user_id.in_(user_ids))
subs = (await db.execute(q)).scalars().all()
for sub in subs:
info = {
"endpoint": sub.endpoint,
"keys": {
"p256dh": sub.p256dh,
"auth": sub.auth,
},
}
ok = await send_push(info, title, body, url)
if ok:
sent += 1
else:
failed += 1
except Exception as e:
logger.error("브로드캐스트 푸시 오류: %s", e)
return {"sent": sent, "failed": failed}