feat: 完成基本功能

This commit is contained in:
Rock Chin
2022-12-08 12:06:04 +08:00
parent 9ca641a2e5
commit 77ec1c7ff0
5 changed files with 53 additions and 16 deletions

17
main.py
View File

@@ -1,10 +1,13 @@
import os
import shutil
import sys
import threading
import time
import pkg.openai.manager
import pkg.database.manager
import pkg.openai.session
import pkg.qqbot.manager
def init_db():
@@ -32,9 +35,23 @@ def main():
# 加载所有未超时的session
pkg.openai.session.load_sessions()
# 初始化qq机器人
qqbot = pkg.qqbot.manager.QQBotManager(mirai_http_api_config=config.mirai_http_api_config,
timeout=config.process_message_timeout, retry=config.retry_times)
qq_bot_thread = threading.Thread(target=qqbot.bot.run, args=(), daemon=True)
qq_bot_thread.start()
if __name__ == '__main__':
if len(sys.argv) > 1 and sys.argv[1] == 'init_db':
init_db()
sys.exit(0)
main()
while True:
try:
time.sleep(86400)
except KeyboardInterrupt:
print("程序退出")
break