import psutil
import threading
import time
import os
from datetime import datetime, timedelta
from collections import deque
from dotenv import load_dotenv

load_dotenv()
log_file_path = os.getenv("NET_SPEED_PATH", '/data/ww/py_sys_monitor/info/net_speed')

def ensure_log_file_exists():
    if not os.path.exists(log_file_path):
        os.makedirs(os.path.dirname(log_file_path), exist_ok=True)
        with open(log_file_path, 'w') as f:
            now = datetime.now()
            for i in range(12):
                past_time = (now - timedelta(hours=i)).replace(minute=0, second=0, microsecond=0).strftime('%Y-%m-%d %H:%M:%S')
                f.write(f"{past_time},0,0\n")

ensure_log_file_exists()

def convert_to_mbps(bytes_per_second):
    # Convert to Megabits per second (Mbps) 保留3位小数
    # 做演示，将数据乘以10
    return round(bytes_per_second * 10 * 8 / (1024 * 1024), 3)  

def log_network_speed():
    while True:
        net_io_start = psutil.net_io_counters()
        time.sleep(3600)  # Sleep for 1 hour
        net_io_end = psutil.net_io_counters()
        
        bytes_sent_per_second = (net_io_end.bytes_sent - net_io_start.bytes_sent) / 3600
        bytes_recv_per_second = (net_io_end.bytes_recv - net_io_start.bytes_recv) / 3600
        
        upload_speed_mbps = convert_to_mbps(bytes_sent_per_second)
        download_speed_mbps = convert_to_mbps(bytes_recv_per_second)
        
        # Get the current time and round down to the nearest hour
        now = datetime.now()
        current_time = now.replace(minute=0, second=0, microsecond=0).strftime('%Y-%m-%d %H:%M:%S')
        
        with open(log_file_path, 'a') as f:
            f.write(f"{current_time},{upload_speed_mbps},{download_speed_mbps}\n")

def get_last_12_hours_speed():
    speeds = []
    now = datetime.now()
    
    # Generate the last 12 hours' timestamps
    timestamps = [(now - timedelta(hours=i)).replace(minute=0, second=0, microsecond=0).strftime('%Y-%m-%d %H:%M:%S') for i in range(12)]
    timestamp_speed_map = {timestamp: {'upload_speed_mbps': 0, 'download_speed_mbps': 0} for timestamp in timestamps}

    with open(log_file_path, 'r') as f:
        lines = deque(f, maxlen=12)  # Read only the last 12 lines

    for line in lines:
        timestamp, upload_speed, download_speed = line.strip().split(',')
        if timestamp in timestamp_speed_map:
            timestamp_speed_map[timestamp] = {
                'upload_speed_mbps': float(upload_speed),
                'download_speed_mbps': float(download_speed)
            }

    for timestamp in timestamps:
        speeds.append({
            'timestamp': timestamp,
            'upload_speed_mbps': timestamp_speed_map[timestamp]['upload_speed_mbps'],
            'download_speed_mbps': timestamp_speed_map[timestamp]['download_speed_mbps']
        })

    return speeds

# Start the thread to log network speed
thread = threading.Thread(target=log_network_speed)
thread.daemon = True
thread.start()