Multi-Account & FinOps

import boto3
import json
from datetime import datetime, timedelta

orgs = boto3.client('organizations', region_name='us-east-1')
ce = boto3.client('ce', region_name='us-east-1')  # Cost Explorer는 us-east-1만 지원
budgets = boto3.client('budgets', region_name='us-east-1')
optimizer = boto3.client('compute-optimizer', region_name='ap-northeast-2')

# 1. 팀별(태그별) 월간 비용 분석
def get_cost_by_team(start_date: str, end_date: str) -> list[dict]:
    response = ce.get_cost_and_usage(
        TimePeriod={'Start': start_date, 'End': end_date},
        Granularity='MONTHLY',
        Filter={
            'Tags': {'Key': 'Team', 'Values': ['backend', 'ml', 'platform'], 'MatchOptions': ['EQUALS']}
        },
        GroupBy=[{'Type': 'TAG', 'Key': 'Team'}],
        Metrics=['UnblendedCost']
    )
    results = []
    for result in response['ResultsByTime']:
        for group in result['Groups']:
            team = group['Keys'][0].replace('Team$', '')
            cost = float(group['Metrics']['UnblendedCost']['Amount'])
            results.append({'team': team, 'cost_usd': cost, 'period': result['TimePeriod']['Start']})
    return results

# 2. 팀별 AWS Budget 생성
def create_team_budget(account_id: str, team: str, monthly_limit_usd: float) -> None:
    budgets.create_budget(
        AccountId=account_id,
        Budget={
            'BudgetName': f'{team}-monthly-budget',
            'BudgetLimit': {'Amount': str(monthly_limit_usd), 'Unit': 'USD'},
            'BudgetType': 'COST',
            'TimeUnit': 'MONTHLY',
            'CostFilters': {'TagKeyValue': [f'user:Team${team}']}
        },
        NotificationsWithSubscribers=[
            {
                'Notification': {
                    'NotificationType': 'ACTUAL',
                    'ComparisonOperator': 'GREATER_THAN',
                    'Threshold': 80.0,
                    'ThresholdType': 'PERCENTAGE'
                },
                'Subscribers': [
                    {'SubscriptionType': 'EMAIL', 'Address': f'{team}-lead@company.com'}
                ]
            },
            {
                'Notification': {
                    'NotificationType': 'FORECASTED',
                    'ComparisonOperator': 'GREATER_THAN',
                    'Threshold': 100.0,
                    'ThresholdType': 'PERCENTAGE'
                },
                'Subscribers': [
                    {'SubscriptionType': 'SNS', 'Address': 'arn:aws:sns:ap-northeast-2:123:cost-alerts'}
                ]
            }
        ]
    )
    print(f'Budget created for team {team}: ${monthly_limit_usd}/month')

# 3. EC2 적정 사이징 추천 조회 (Compute Optimizer)
def get_rightsizing_recommendations() -> list[dict]:
    response = optimizer.get_ec2_instance_recommendations()
    recommendations = []
    for rec in response.get('instanceRecommendations', []):
        if rec['findingReasonCodes'][0] == 'CPUUnderprovisioned' or \
           rec['finding'] == 'OVER_PROVISIONED':
            current = rec['currentInstanceType']
            suggested = rec['recommendationOptions'][0]['instanceType']
            saving = rec['recommendationOptions'][0].get('estimatedMonthlySavings', {}).get('value', 0)
            recommendations.append({
                'instance_id': rec['instanceArn'].split('/')[-1],
                'current': current, 'suggested': suggested,
                'monthly_savings_usd': float(saving)
            })
    return sorted(recommendations, key=lambda x: x['monthly_savings_usd'], reverse=True)

# 실행 예시
start = (datetime.now().replace(day=1) - timedelta(days=1)).replace(day=1).strftime('%Y-%m-%d')
end = datetime.now().replace(day=1).strftime('%Y-%m-%d')
costs = get_cost_by_team(start, end)
for c in costs:
    print(f"Team {c['team']}: ${c['cost_usd']:.2f} ({c['period']})")

create_team_budget('123456789012', 'ml', 5000.0)

recs = get_rightsizing_recommendations()
print(f'\nTop rightsizing opportunity: {recs[0]} → ${recs[0]["monthly_savings_usd"]:.0f}/month savings')