1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
|
#!/usr/bin/python
import sys
import psutil
import serial
import platform
from time import sleep, time
from datetime import datetime
import os
if platform.system() == "Windows":
import win32com.client
# Constants
FILE_DESCRIPTOR = 'COM3' if platform.system() == 'Windows' else '/dev/ttyUSB0'
BAUD_RATE = 9600
SERIAL_TIMEOUT = 1 # Timeout for serial operations in seconds
CONNECTION_WAIT = 6 # Time to wait for the serial connection to establish
SEND_INTERVAL = 2 # Interval between sending data in seconds
# CPU temperature path for Linux
#CPU_TEMP_PATH = '/sys/devices/pci0000:00/0000:00:18.3/hwmon/hwmon4/temp1_input'
CPU_TEMP_PATH = os.environ.get('CPU_TEMP_PATH', '/sys/devices/pci0000:00/0000:00:18.3/hwmon/hwmon3/temp1_input')
# Open serial connection
def open_serial_connection():
try:
return serial.Serial(port=FILE_DESCRIPTOR, baudrate=BAUD_RATE, timeout=SERIAL_TIMEOUT)
except serial.SerialException as e:
sys.exit(f"Failed to connect to the serial port: {e}")
# Send command to Arduino
def send_command(arduino, command):
try:
arduino.write(command.encode())
sleep(0.1)
except serial.SerialException as e:
print(f"Failed to send command: {e}")
# Get the current time
def get_time():
return datetime.now().strftime('%H:%M:%S')
# Get RAM information
def get_ram_info():
ram = psutil.virtual_memory()
return int(ram.total / (1024 * 1024)), int(ram.available / (1024 * 1024))
# Get OS name
def get_os_name():
if platform.system() == 'Windows':
return platform.system()
else:
try:
with open('/etc/os-release') as f:
for line in f:
if line.startswith('NAME='):
return line.split('=')[1].strip().strip('"')
except FileNotFoundError:
return "Unknown"
# Get kernel version
def get_kernel_version():
if platform.system() == 'Windows':
return platform.version()
else:
try:
with open('/proc/version') as f:
return f.readline().split()[2].split('-')[0]
except FileNotFoundError:
return "Unknown"
# Get system uptime
def get_uptime():
uptime_seconds = psutil.boot_time()
uptime = datetime.now() - datetime.fromtimestamp(uptime_seconds)
hours, remainder = divmod(int(uptime.total_seconds()), 3600)
minutes = remainder // 60
return f"{hours:02d}:{minutes:02d}"
# Get CPU temperature
def get_cpu_temp():
if platform.system() == 'Windows':
# For Windows, use WMI to get CPU temperature
wmi = win32com.client.Dispatch("WbemScripting.SWbemLocator")
service = wmi.ConnectServer(".", "root\cimv2")
temperature_info = service.ExecQuery(
"SELECT * FROM Win32_TemperatureProbe")
for sensor in temperature_info:
if sensor.CurrentReading is not None:
# Convert from tenths of Kelvin to Celsius
return sensor.CurrentReading / 10.0 - 273.15
return 0 # Return 0 if there's no temperature sensor data
else:
try:
with open(CPU_TEMP_PATH) as f:
return int(f.read().strip()) / 1000
except (FileNotFoundError, ValueError):
return 0 # Return 0 if there's an error reading CPU temperature
# Main data gathering and sending function
def gather_and_send_data(arduino, ram_max, os_name, kernel_version):
while True:
start_time = time()
try:
cpu_temp = int(get_cpu_temp())
cpu_usage = int(psutil.cpu_percent(interval=None))
_, ram_free = get_ram_info()
current_time = get_time()
uptime = get_uptime()
request = (f"\nCpuTemp={cpu_temp},CpuUsage={cpu_usage},RamMax={ram_max},"
f"RamFree={ram_free},Time={current_time},OS={os_name},"
f"Kernel={kernel_version},Uptime={uptime}")
send_command(arduino, request)
except Exception as e:
print(f"An error occurred: {e}")
elapsed_time = time() - start_time
if elapsed_time < SEND_INTERVAL:
sleep(SEND_INTERVAL - elapsed_time)
# Main function to handle program start
def main():
arduino = open_serial_connection()
sleep(CONNECTION_WAIT) # Wait for the connection to establish
ram_max, _ = get_ram_info()
os_name = get_os_name()
kernel_version = get_kernel_version()
try:
gather_and_send_data(arduino, ram_max, os_name, kernel_version)
except KeyboardInterrupt:
print("Interrupted by user")
finally:
if arduino.is_open:
arduino.close()
if __name__ == "__main__":
main()
|