1
# -*- coding: utf-8 -*-
2
from aliyunsdkcore.client import AcsClient
3
from aliyunsdkcore.request import CommonRequest
4
from aliyunsdkecs.request.v20140526 import StartInstancesRequest, StopInstancesRequest, DescribeInstancesRequest
5
import json
6
import sys
7
import logging
8
# pip install aliyun-python-sdk-core aliyun-python-sdk-ecs
9

10

11
# ================== 1. 配置日志 ==================
12
logging.basicConfig(
13
    level=logging.INFO,
14
    format="%(asctime)s - %(levelname)s - %(message)s",
15
    stream=sys.stdout
16
)
17
logger = logging.getLogger(__name__)
18

19
# ================== 2. 配置阿里云凭证和ECS实例信息 ==================
20
ACCESS_KEY_ID = 'LTAIxxxxxxx'      # 您的AccessKey ID
21
ACCESS_KEY_SECRET = 'xxxxxxxxxx'   # 您的AccessKey Secret
22
REGION_ID = 'cn-hongkong'          # 区域ID
23
ECS_INSTANCE_ID = 'i-xxxxxxxxx'    # 您要控制的ECS实例ID
24

25
# 流量阈值 (GB)
26
TRAFFIC_THRESHOLD_GB = 180
27

28
# ================== 3. 初始化客户端 ==================
29
try:
30
    client = AcsClient(ACCESS_KEY_ID, ACCESS_KEY_SECRET, REGION_ID)
31
    logger.info("AcsClient initialized successfully.")
32
except Exception as e:
33
    logger.error(f"Failed to initialize AcsClient: {e}")
34
    sys.exit(1)
35

36
# ================== 4. 查询当前总流量 ==================
37
def get_total_traffic_gb(client):
38
    request = CommonRequest()
39
    request.set_domain('cdt.aliyuncs.com')
40
    request.set_version('2021-08-13')
41
    request.set_action_name('ListCdtInternetTraffic')
42
    request.set_method('POST')
43

44
    try:
45
        response = client.do_action_with_exception(request)
46
        response_json = json.loads(response.decode('utf-8'))
47

48
        total_bytes = sum(d.get('Traffic', 0) for d in response_json.get('TrafficDetails', []))
49
        total_gb = total_bytes / (1024 ** 3)
50

51
        logger.info(f"当前总互联网流量: {total_gb:.2f} GB")
52
        return total_gb
53
    except Exception as e:
54
        logger.error(f"获取CDT流量失败: {e}")
55
        sys.exit(1)
56

57
# ================== 5. 查询ECS实例状态 ==================
58
def get_ecs_status(client, instance_id):
59
    try:
60
        request = DescribeInstancesRequest.DescribeInstancesRequest()
61
        request.set_InstanceIds([instance_id])
62
        response = client.do_action_with_exception(request)
63
        response_json = json.loads(response.decode('utf-8'))
64

65
        instances = response_json.get("Instances", {}).get("Instance", [])
66
        if not instances:
67
            logger.error("未找到该ECS实例信息。")
68
            return None
69

70
        status = instances[0].get("Status")
71
        logger.info(f"ECS实例 {instance_id} 当前状态: {status}")
72
        return status
73
    except Exception as e:
74
        logger.error(f"获取ECS实例状态失败: {e}")
75
        return None
76

77
# ================== 6. 启动ECS实例 ==================
78
def ecs_start(client, instance_id):
79
    status = get_ecs_status(client, instance_id)
80
    if status == "Running":
81
        logger.info(f"ECS实例 {instance_id} 已经是运行状态，无需启动。")
82
        return
83

84
    try:
85
        request = StartInstancesRequest.StartInstancesRequest()
86
        request.set_InstanceIds([instance_id])
87
        request.set_accept_format('json')
88

89
        response = client.do_action_with_exception(request)
90
        logger.info(f"ECS启动响应: {response.decode('utf-8')}")
91
    except Exception as e:
92
        logger.error(f"启动ECS实例失败: {e}")
93

94
# ================== 7. 停止ECS实例 ==================
95
def ecs_stop(client, instance_id):
96
    status = get_ecs_status(client, instance_id)
97
    if status == "Stopped":
98
        logger.info(f"ECS实例 {instance_id} 已经是停止状态，无需再次停止。")
99
        return
100

101
    try:
102
        request = StopInstancesRequest.StopInstancesRequest()
103
        request.set_InstanceIds([instance_id])
104
        request.set_ForceStop(False)
105
        request.set_accept_format('json')
106

107
        response = client.do_action_with_exception(request)
108
        logger.info(f"ECS停止响应: {response.decode('utf-8')}")
109
    except Exception as e:
110
        logger.error(f"停止ECS实例失败: {e}")
111

112
# ================== 8. 主流程 ==================
113
def main():
114
    total_gb = get_total_traffic_gb(client)
115

116
    if total_gb < TRAFFIC_THRESHOLD_GB:
117
        logger.info(f"流量 {total_gb:.2f} GB < 阈值 {TRAFFIC_THRESHOLD_GB} GB，尝试启动 ECS")
118
        ecs_start(client, ECS_INSTANCE_ID)
119
    else:
120
        logger.info(f"流量 {total_gb:.2f} GB ≥ 阈值 {TRAFFIC_THRESHOLD_GB} GB，尝试停止 ECS")
121
        ecs_stop(client, ECS_INSTANCE_ID)
122

123
    logger.info("脚本执行完毕。")
124

125
if __name__ == "__main__":
126
    main()

1
nano /root/cdt_daily_report.sh

1
#!/bin/bash Bash 解释器
2

3
# ============================================
4
# 使用说明
5
# ============================================
6
# 1. Telegram Bot 配置
7
#    - TG_BOT_TOKEN：你的 Telegram Bot Token
8
#    - TG_CHAT_ID：你的 Telegram Chat ID
9
#
10
# 2. 阿里云 ECS / CDT 配置
11
#    - ACCESS_KEY_ID / ACCESS_KEY_SECRET：AK/SK
12
#    - REGION_ID：实例所在地域 ID（如 cn-hongkong）
13
#    - ECS_INSTANCE_ID：实例的 ECS ID
14
#    - ECS_NAME：实例名称（自定义）
15
# ============================================
16

17
# ================================
18
# 1. 配置信息（请填写）
19
# ================================
20
TG_BOT_TOKEN="REDACTED_TG_BOT_TOKEN"
21
TG_CHAT_ID="REDACTED_CHAT_ID"
22

23
ACCESS_KEY_ID='REDACTED_AK'
24
ACCESS_KEY_SECRET='REDACTED_SK'
25
REGION_ID='cn-hongkong'
26
ECS_INSTANCE_ID='REDACTED_ECS_ID'
27
ECS_NAME='AliCloud-CDT'
28

29
# 流量阈值设置（GB），CDT 免费额度通常为 200GB
30
LIMIT=200
31

32
# ================================
33
# 发送 Telegram 消息函数
34
# ================================
35
send_tg() {
36
    curl -s -o /dev/null -X POST "https://api.telegram.org/bot${TG_BOT_TOKEN}/sendMessage" \
37
        -H "Content-Type: application/json" \
38
        -d "{\"chat_id\": \"${TG_CHAT_ID}\", \"text\": \"$1\", \"parse_mode\": \"HTML\"}"
39
}
40

41
# ================================
42
# Python 查询 CDT 流量
43
# ================================
44
query_cdt() {
45
python3 <<EOF
46
from aliyunsdkcore.client import AcsClient
47
from aliyunsdkcore.request import CommonRequest
48
import json
49

50
client = AcsClient("$ACCESS_KEY_ID", "$ACCESS_KEY_SECRET", "$REGION_ID")
51

52
request = CommonRequest()
53
request.set_domain('cdt.aliyuncs.com')
54
request.set_version('2021-08-13')
55
request.set_action_name('ListCdtInternetTraffic')
56
request.set_method('POST')
57

58
try:
59
    response = client.do_action_with_exception(request)
60
    data = json.loads(response.decode('utf-8'))
61
    total_bytes = sum(d.get('Traffic', 0) for d in data.get('TrafficDetails', []))
62
    print(f"{total_bytes / (1024**3):.2f}")
63
except Exception as e:
64
    print("0.00")
65
EOF
66
}
67

68
# ================================
69
# Python 获取 ECS 公网 IP
70
# ================================
71
get_public_ip() {
72
python3 <<EOF
73
from aliyunsdkcore.client import AcsClient
74
from aliyunsdkcore.request import CommonRequest
75
import json
76

77
client = AcsClient("$ACCESS_KEY_ID", "$ACCESS_KEY_SECRET", "$REGION_ID")
78

79
request = CommonRequest()
80
request.set_domain('ecs.aliyuncs.com')
81
request.set_version('2014-05-26')
82
request.set_action_name('DescribeInstances')
83
request.set_method('POST')
84
request.add_query_param('InstanceIds', f'["$ECS_INSTANCE_ID"]')
85

86
try:
87
    response = client.do_action_with_exception(request)
88
    data = json.loads(response.decode('utf-8'))
89
    instance = data['Instances']['Instance'][0]
90
    eip = instance.get('EipAddress', {}).get('IpAddress')
91
    pub = instance.get('PublicIpAddress', {}).get('IpAddress', [])
92
    if eip:
93
        print(eip)
94
    elif pub:
95
        print(pub[0])
96
    else:
97
        print("无公网IP")
98
except:
99
    print("获取失败")
100
EOF
101
}
102

103
# ================================
104
# 执行查询与计算
105
# ================================
106
TRAFFIC=$(query_cdt)
107
PUBLIC_IP=$(get_public_ip)
108

109
# 计算剩余与百分比
110
REMAIN=$(awk "BEGIN {r=$LIMIT-$TRAFFIC; if(r<0) r=0; printf \"%.2f\", r}")
111
USAGE_PERCENT=$(awk "BEGIN {printf \"%.2f\", ($TRAFFIC/$LIMIT)*100}")
112

113
# 颜色等级判定
114
COLOR=$(awk 'BEGIN {p='$USAGE_PERCENT'; if(p<60) print "🟢 正常"; else if(p<90) print "🟡 注意"; else print "🔴 危险";}')
115

116
# ================================
117
# 生成消息内容（已修改：强制北京时间）
118
# ================================
119
# 将 date 改为 TZ=Asia/Shanghai date
120
MESSAGE="<b>📊 CDT 每日流量报告</b>
121

122
<b>🖥 ${ECS_NAME}</b>
123
┃ 📦 实例：${ECS_INSTANCE_ID}
124
┃ 🌐 公网 IP：${PUBLIC_IP}
125
┃ 📍 地域：${REGION_ID}
126
┃ 📈 已用：<b>${TRAFFIC} GB</b>
127
┃ 💾 剩余：<b>${REMAIN} GB</b>
128
┃ 🔥 使用率：<b>${USAGE_PERCENT}%</b>  ${COLOR}
129

130
⏰ <i>更新时间：$(TZ=Asia/Shanghai date '+%Y-%m-%d %H:%M')</i>"
131

132
# 发送通知
133
send_tg "$MESSAGE"
134

135
# ================================
136
# 写入日志（已修改：强制北京时间）
137
# ================================
138
LOG_DIR="/var/log/cdt_daily_report"
139
mkdir -p $LOG_DIR
140

141
{
142
    echo "-------------------------------------------"
143
    # 这里也改成了北京时间
144
    echo "更新时间：$(TZ=Asia/Shanghai date '+%Y-%m-%d %H:%M')"
145
    echo "${ECS_NAME}"
146
    echo "┃ 实例：${ECS_INSTANCE_ID}"
147
    echo "┃ 公网 IP：${PUBLIC_IP}"
148
    echo "┃ 地域：${REGION_ID}"
149
    echo "┃ 已用：${TRAFFIC} GB"
150
    echo "┃ 剩余：${REMAIN} GB"
151
    echo "┃ 使用率：${USAGE_PERCENT}%  $(echo ${COLOR} | sed 's/.* //')"
152
    echo "-------------------------------------------"
153
# 这里的文件名也改成了北京时间，防止跨月的那天日志建错文件
154
} >> "${LOG_DIR}/cdt_report_$(TZ=Asia/Shanghai date '+%Y%m').log"