mirror of
https://github.com/RockChinQ/QChatGPT.git
synced 2024-11-16 11:42:44 +08:00
64 lines
1.8 KiB
Python
64 lines
1.8 KiB
Python
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():
|
|
import config
|
|
database = pkg.database.manager.DatabaseManager(**config.mysql_config)
|
|
|
|
database.initialize_database()
|
|
|
|
|
|
def main():
|
|
# 检查是否有config.py,如果没有就把config-template.py复制一份,并退出程序
|
|
if not os.path.exists('config.py'):
|
|
shutil.copy('config-template.py', 'config.py')
|
|
print('请先在config.py中填写配置')
|
|
return
|
|
# 导入config.py
|
|
assert os.path.exists('config.py')
|
|
import config
|
|
|
|
# 主启动流程
|
|
openai_interact = pkg.openai.manager.OpenAIInteract(config.openai_config['api_key'], config.completion_api_params)
|
|
|
|
database = pkg.database.manager.DatabaseManager(**config.mysql_config)
|
|
|
|
# 加载所有未超时的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:
|
|
try:
|
|
for session in pkg.openai.session.sessions:
|
|
pkg.openai.session.sessions[session].persistence()
|
|
except Exception as e:
|
|
if not isinstance(e, KeyboardInterrupt):
|
|
raise e
|
|
print("程序退出")
|
|
break
|