mirror of
https://github.com/zhayujie/chatgpt-on-wechat.git
synced 2026-02-04 09:17:41 +08:00
49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
# encoding:utf-8
|
|
|
|
import os
|
|
from config import conf, load_config
|
|
from channel import channel_factory
|
|
from common.log import logger
|
|
from plugins import *
|
|
import signal
|
|
import sys
|
|
|
|
def sigterm_handler_wrap(_signo):
|
|
old_handler = signal.getsignal(_signo)
|
|
def func(_signo, _stack_frame):
|
|
logger.info("signal {} received, exiting...".format(_signo))
|
|
conf().save_user_datas()
|
|
return old_handler(_signo, _stack_frame)
|
|
signal.signal(_signo, func)
|
|
|
|
def run():
|
|
try:
|
|
# load config
|
|
load_config()
|
|
# ctrl + c
|
|
sigterm_handler_wrap(signal.SIGINT)
|
|
# kill signal
|
|
sigterm_handler_wrap(signal.SIGTERM)
|
|
|
|
# create channel
|
|
channel_name=conf().get('channel_type', 'wx')
|
|
|
|
if "--cmd" in sys.argv:
|
|
channel_name = 'terminal'
|
|
|
|
if channel_name == 'wxy':
|
|
os.environ['WECHATY_LOG']="warn"
|
|
# os.environ['WECHATY_PUPPET_SERVICE_ENDPOINT'] = '127.0.0.1:9001'
|
|
|
|
channel = channel_factory.create_channel(channel_name)
|
|
if channel_name in ['wx','wxy','terminal','wechatmp','wechatmp_service']:
|
|
PluginManager().load_plugins()
|
|
|
|
# startup channel
|
|
channel.startup()
|
|
except Exception as e:
|
|
logger.error("App startup failed!")
|
|
logger.exception(e)
|
|
|
|
if __name__ == '__main__':
|
|
run() |