From 45295b52db228981639fd78e69c0b35c0698b475 Mon Sep 17 00:00:00 2001
From: Rock Chin <junyan_qin@outlook.com>
Date: Fri, 9 Dec 2022 15:44:49 +0800
Subject: [PATCH] =?UTF-8?q?perf:=20=E8=B0=83=E6=95=B4main.py=E4=B8=AD?=
 =?UTF-8?q?=E7=9A=84=E5=BC=95=E5=85=A5=E9=A1=BA=E5=BA=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 README.md |  2 +-
 main.py   | 40 ++++++++++++++++++++--------------------
 2 files changed, 21 insertions(+), 21 deletions(-)

diff --git a/README.md b/README.md
index 56dde91c..2ebdfe51 100644
--- a/README.md
+++ b/README.md
@@ -47,7 +47,7 @@ git clone https://github.com/RockChinQ/QChatGPT
 2. 安装依赖
 
 ```bash
-pip install pymysql yiri-mirai openai colorlog
+pip install pymysql yiri-mirai openai colorlog func_timeout
 ```
 
 3. 运行一次主程序，生成配置文件
diff --git a/main.py b/main.py
index 520fa9ff..f13921c8 100644
--- a/main.py
+++ b/main.py
@@ -7,7 +7,6 @@ import time
 import logging
 import colorlog
 
-
 log_colors_config = {
     'DEBUG': 'green',  # cyan white
     'INFO': 'white',
@@ -16,6 +15,7 @@ log_colors_config = {
     'CRITICAL': 'bold_red',
 }
 
+
 def init_db():
     import config
     import pkg.database.manager
@@ -46,6 +46,11 @@ def main():
     ))
     logging.getLogger().addHandler(sh)
 
+    import pkg.openai.manager
+    import pkg.database.manager
+    import pkg.openai.session
+    import pkg.qqbot.manager
+
     # 主启动流程
     openai_interact = pkg.openai.manager.OpenAIInteract(config.openai_config['api_key'], config.completion_api_params)
 
@@ -61,25 +66,6 @@ def main():
     qq_bot_thread = threading.Thread(target=qqbot.bot.run, args=(), daemon=True)
     qq_bot_thread.start()
 
-
-if __name__ == '__main__':
-    # 检查是否有config.py,如果没有就把config-template.py复制一份,并退出程序
-    if not os.path.exists('config.py'):
-        shutil.copy('config-template.py', 'config.py')
-        print('请先在config.py中填写配置')
-        sys.exit(0)
-
-    if len(sys.argv) > 1 and sys.argv[1] == 'init_db':
-        init_db()
-        sys.exit(0)
-
-    import pkg.openai.manager
-    import pkg.database.manager
-    import pkg.openai.session
-    import pkg.qqbot.manager
-
-    main()
-
     logging.info('程序启动完成')
 
     while True:
@@ -95,3 +81,17 @@ if __name__ == '__main__':
                     raise e
             print("程序退出")
             break
+
+
+if __name__ == '__main__':
+    # 检查是否有config.py,如果没有就把config-template.py复制一份,并退出程序
+    if not os.path.exists('config.py'):
+        shutil.copy('config-template.py', 'config.py')
+        print('请先在config.py中填写配置')
+        sys.exit(0)
+
+    if len(sys.argv) > 1 and sys.argv[1] == 'init_db':
+        init_db()
+        sys.exit(0)
+
+    main()