From b7fc6bab382edb03a1025b5259fe2d926f1fb7e2 Mon Sep 17 00:00:00 2001
From: yshtcn <k.yshtcn+github@gmail.com>
Date: Sun, 23 Jul 2023 14:21:52 +0800
Subject: [PATCH] Update heartbeat.py
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 增加自动复制范例的功能
- 修正无法正确读取UTF-8
---
 heartbeat.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/heartbeat.py b/heartbeat.py
index 0794cf6..cd06974 100644
--- a/heartbeat.py
+++ b/heartbeat.py
@@ -52,12 +52,19 @@ def quit_action(icon, item):
 image = create_image()
 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.read(os.path.join(current_dir, 'config.ini'))
+config.read(config_path, encoding='utf-8')  # 使用UTF-8编码来读取文件
 interval = config.getint('Settings', 'interval')
 heartbeat_url = config.get('Settings', 'heartbeat_url')
 
+
 # 创建一个新的Session对象，并根据需要配置代理
 session = requests.Session()
 if config.get('Settings', 'proxy_enabled', fallback='0') == '1':