yshtcn/heartbeat
1
0
Fork 0
mirror of https://github.com/yshtcn/heartbeat.git synced 2025-12-14 18:20:26 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update heartbeat.py

Browse Source 
增加socks5代理支持
This commit is contained in:
yshtcn 2023-07-23 13:20:28 +08:00 committed by GitHub
parent 4f9e58b33d
commit 6fc4e79126
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
heartbeat.py
View File

@ -16,24 +16,26 @@ log_file_path = os.path.join(current_dir, 'heartbeat.log')
logging.basicConfig(filename=log_file_path, level=logging.INFO) logging.basicConfig(filename=log_file_path, level=logging.INFO)
def create_image(): def create_image():
# Generate an image and draw a pattern
width, height = 64, 64 width, height = 64, 64
color1 = "black" color1 = "black"
color0 = "white" color0 = "white"
image = Image.new('RGB', (width, height), color1) image = Image.new('RGB', (width, height), color1)
d = ImageDraw.Draw(image) d = ImageDraw.Draw(image)
d.rectangle(
[(width // 2, 0), (width, height)], # 画图标
fill=color0) d.arc((32,0, 64,32), 0, 180, fill=color0)
d.rectangle((0, 32, 32, 64), fill=color0)
return image return image
# Create an Event object to signal the heartbeat thread to stop # Create an Event object to signal the heartbeat thread to stop
stop_heartbeat = threading.Event() stop_heartbeat = threading.Event()
def heartbeat(interval, heartbeat_url): def heartbeat(interval, heartbeat_url, session):
while not stop_heartbeat.is_set(): while not stop_heartbeat.is_set():
try: try:
response = requests.get(heartbeat_url) response = session.get(heartbeat_url)
logging.info(f"{datetime.now()} Response status code: {response.status_code}") logging.info(f"{datetime.now()} Response status code: {response.status_code}")
except requests.exceptions.RequestException as e: except requests.exceptions.RequestException as e:
logging.info(f"{datetime.now()} An error occurred: {e}") logging.info(f"{datetime.now()} An error occurred: {e}")
@ -50,16 +52,20 @@ def quit_action(icon, item):
image = create_image() image = create_image()
icon = pystray.Icon("系统心跳", image, "系统状态报送", menu=pystray.Menu(pystray.MenuItem('关闭主机状态报送', quit_action))) icon = pystray.Icon("系统心跳", image, "系统状态报送", menu=pystray.Menu(pystray.MenuItem('关闭主机状态报送', quit_action)))
# 从配置文件读取设置 # 从配置文件读取设置
config = configparser.ConfigParser() config = configparser.ConfigParser()
config.read(os.path.join(current_dir, 'config.ini')) config.read(os.path.join(current_dir, 'config.ini'))
interval = config.getint('Settings', 'interval') interval = config.getint('Settings', 'interval')
heartbeat_url = config.get('Settings', 'heartbeat_url') heartbeat_url = config.get('Settings', 'heartbeat_url')
# 创建一个新的Session对象，并根据需要配置代理
session = requests.Session()
if config.get('Settings', 'proxy_enabled', fallback='0') == '1':
session.proxies = {'http': config.get('Settings', 'proxy_url'),
'https': config.get('Settings', 'proxy_url')}
# Start the heartbeat function in a new thread # Start the heartbeat function in a new thread
t = threading.Thread(target=heartbeat, args=(interval, heartbeat_url)) t = threading.Thread(target=heartbeat, args=(interval, heartbeat_url, session))
t.start() t.start()
icon.run(setup) icon.run(setup)