mirror of
https://github.com/xfgryujk/blivechat.git
synced 2024-12-27 13:20:29 +08:00
113 lines
3.0 KiB
Python
113 lines
3.0 KiB
Python
# -*- coding: utf-8 -*-
|
||
import argparse
|
||
import logging
|
||
import logging.handlers
|
||
import os
|
||
import webbrowser
|
||
|
||
import tornado.ioloop
|
||
import tornado.web
|
||
|
||
import api.chat
|
||
import api.main
|
||
import config
|
||
import models.database
|
||
import services.avatar
|
||
import services.chat
|
||
import services.translate
|
||
import update
|
||
import utils.request
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
routes = [
|
||
(r'/api/server_info', api.main.ServerInfoHandler),
|
||
(r'/api/emoticon', api.main.UploadEmoticonHandler),
|
||
|
||
(r'/api/chat', api.chat.ChatHandler),
|
||
(r'/api/room_info', api.chat.RoomInfoHandler),
|
||
(r'/api/avatar_url', api.chat.AvatarHandler),
|
||
|
||
(rf'{api.main.EMOTICON_BASE_URL}/(.*)', tornado.web.StaticFileHandler, {'path': api.main.EMOTICON_UPLOAD_PATH}),
|
||
(r'/(.*)', api.main.MainHandler, {'path': config.WEB_ROOT})
|
||
]
|
||
|
||
|
||
def main():
|
||
args = parse_args()
|
||
|
||
init_logging(args.debug)
|
||
config.init()
|
||
|
||
utils.request.init()
|
||
models.database.init(args.debug)
|
||
|
||
services.avatar.init()
|
||
services.translate.init()
|
||
services.chat.init()
|
||
|
||
update.check_update()
|
||
|
||
run_server(args.host, args.port, args.debug)
|
||
|
||
|
||
def parse_args():
|
||
parser = argparse.ArgumentParser(description='用于OBS的仿YouTube风格的bilibili直播评论栏')
|
||
parser.add_argument('--host', help='服务器host，默认和配置中的一样', default=None)
|
||
parser.add_argument('--port', help='服务器端口，默认和配置中的一样', type=int, default=None)
|
||
parser.add_argument('--debug', help='调试模式', action='store_true')
|
||
return parser.parse_args()
|
||
|
||
|
||
def init_logging(debug):
|
||
filename = os.path.join(config.BASE_PATH, 'log', 'blivechat.log')
|
||
stream_handler = logging.StreamHandler()
|
||
file_handler = logging.handlers.TimedRotatingFileHandler(
|
||
filename, encoding='utf-8', when='midnight', backupCount=7, delay=True
|
||
)
|
||
logging.basicConfig(
|
||
format='{asctime} {levelname} [{name}]: {message}',
|
||
style='{',
|
||
level=logging.INFO if not debug else logging.DEBUG,
|
||
handlers=[stream_handler, file_handler]
|
||
)
|
||
|
||
# 屏蔽访问日志
|
||
logging.getLogger('tornado.access').setLevel(logging.WARNING)
|
||
|
||
|
||
def run_server(host, port, debug):
|
||
cfg = config.get_config()
|
||
if host is None:
|
||
host = cfg.host
|
||
if port is None:
|
||
port = cfg.port
|
||
|
||
app = tornado.web.Application(
|
||
routes,
|
||
websocket_ping_interval=10,
|
||
debug=debug,
|
||
autoreload=False
|
||
)
|
||
try:
|
||
app.listen(
|
||
port,
|
||
host,
|
||
xheaders=cfg.tornado_xheaders,
|
||
max_body_size=1024 * 1024,
|
||
max_buffer_size=1024 * 1024
|
||
)
|
||
except OSError:
|
||
logger.warning('Address is used %s:%d', host, port)
|
||
return
|
||
finally:
|
||
if cfg.open_browser_at_startup:
|
||
url = 'http://localhost/' if port == 80 else f'http://localhost:{port}/'
|
||
webbrowser.open(url)
|
||
logger.info('Server started: %s:%d', host, port)
|
||
tornado.ioloop.IOLoop.current().start()
|
||
|
||
|
||
if __name__ == '__main__':
|
||
main()
|