tele_check_block_china.py

import os
import subprocess
import time
import requests

# Các IP cần kiểm tra
CHECK_IPS = ['china.phamanh.id.vn', '111.180.200.1', '117.50.76.1', '202.189.9.1']
# File để lưu thời gian gửi thông báo lần cuối
LAST_NOTIFY_FILE = "/root/last_notify_time.txt"
# File lưu số AWS
AWS_NUMBER_FILE = "/root/aws_number.txt"

# Thông tin của bot Telegram
TELEGRAM_TOKEN = "7440782622:AAFF-Yh8uqueYN5OkLw8VCw1krl2EbAl8vk"  # Token bot của bạn
CHAT_IDS = [5179418894, 5869289981]  # Danh sách CHAT_ID của bạn

# Hàm lấy IP của VPS từ ipinfo.io
def get_vps_ip():
    try:
        result = subprocess.run(['curl', '-s', 'ipinfo.io/ip'], stdout=subprocess.PIPE, check=True)
        vps_ip = result.stdout.decode('utf-8').strip()
        return vps_ip
    except subprocess.CalledProcessError as e:
        print(f"Lỗi khi lấy IP của VPS: {e}")
        return None

# Hàm ping IP
def ping_ip(ip):
    response = os.system(f"ping -c 4 {ip} > /dev/null 2>&1")  # Thực hiện 4 lần ping và ẩn thông báo
    return response == 0  # Trả về True nếu ping thành công, False nếu không thành công

# Hàm gửi thông báo qua Telegram
def send_telegram_message(text):
    url = f"https://api.telegram.org/bot{TELEGRAM_TOKEN}/sendMessage"
    for chat_id in CHAT_IDS:  # Lặp qua từng chat ID trong danh sách
        payload = {
            'chat_id': chat_id,
            'text': text
        }
        
        try:
            response = requests.post(url, json=payload)
            if response.status_code == 200:
                print(f"Thông báo đã được gửi đến Telegram chat ID {chat_id}.")
            else:
                print(f"Lỗi khi gửi thông báo đến chat ID {chat_id}: {response.text}")
        except Exception as e:
            print(f"Lỗi khi gửi thông báo đến chat ID {chat_id}: {e}")

# Hàm lấy thời gian gửi thông báo lần cuối từ file
def get_last_notify_time():
    try:
        with open(LAST_NOTIFY_FILE, 'r') as file:
            last_time = float(file.read())
            return last_time
    except FileNotFoundError:
        return 0  # Nếu file không tồn tại, coi như chưa từng gửi thông báo

# Hàm lưu thời gian gửi thông báo hiện tại
def save_last_notify_time():
    with open(LAST_NOTIFY_FILE, 'w') as file:
        file.write(str(time.time()))

# Hàm lấy số AWS từ file
def get_aws_number():
    try:
        with open(AWS_NUMBER_FILE, 'r') as file:
            aws_number = file.read().strip()
            return aws_number
    except FileNotFoundError:
        print("Không tìm thấy file lưu số AWS.")
        return None

# Lấy IP của VPS
VPS_IP = get_vps_ip()

if VPS_IP:
    # Kiểm tra các IP và chỉ gửi thông báo nếu tất cả đều không ping được
    all_failed = True
    for ip in CHECK_IPS:
        if ping_ip(ip):
            all_failed = False  # Nếu ping được ít nhất một IP, không cần gửi thông báo
            print(f"Ping thành công đến IP: {ip}")
            break
        else:
            print(f"Không thể ping đến IP: {ip}")

    if all_failed:
        # Nếu không ping được bất kỳ IP nào, kiểm tra thời gian gửi thông báo lần cuối
        last_notify_time = get_last_notify_time()
        current_time = time.time()
        if current_time - last_notify_time > 1800:  # 30 phút = 1800 giây
            aws_number = get_aws_number()  # Lấy số AWS từ file
            # Tạo thông báo
            text = f"IP máy chủ {VPS_IP} đã bị chặn tại Trung Quốc. Vui lòng đổi IP."
            if aws_number:  # Nếu số AWS có, thêm vào thông báo
                text += f"\nSố AWS: {aws_number}"
            send_telegram_message(text)  # Gửi thông báo tới Telegram
            save_last_notify_time()  # Lưu thời gian gửi thông báo hiện tại
        else:
            print("Đã gửi thông báo cảnh báo trong vòng 30 phút trước, không gửi lại.")
else:
    print("Không thể lấy IP của VPS.")