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 
- 增加自动复制范例的功能
- 修正无法正确读取UTF-8
This commit is contained in:
yshtcn 2023-07-23 14:21:52 +08:00 committed by GitHub
parent c01ed6e00c
commit b7fc6bab38
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
heartbeat.py
View File

@ -52,12 +52,19 @@ 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.ini是否存在，如果不存在，从config.Exsample.ini复制一份
config_path = os.path.join(current_dir, 'config.ini')
if not os.path.exists(config_path):
exsample_config_path = os.path.join(current_dir, 'config.Exsample.ini')
shutil.copyfile(exsample_config_path, config_path)
# 从配置文件读取设置 # 从配置文件读取设置
config = configparser.ConfigParser() config = configparser.ConfigParser()
config.read(os.path.join(current_dir, 'config.ini')) config.read(config_path, encoding='utf-8') # 使用UTF-8编码来读取文件
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对象，并根据需要配置代理
session = requests.Session() session = requests.Session()
if config.get('Settings', 'proxy_enabled', fallback='0') == '1': if config.get('Settings', 'proxy_enabled', fallback='0') == '1':