From d066e86f9e16fbaa5517ff07b59b9a73208f1da4 Mon Sep 17 00:00:00 2001 From: LWR Date: Mon, 24 Jul 2023 22:28:53 +0800 Subject: [PATCH] fix: Catch LiveException when load new user who room id is 0 in MySQLDataSource --- starbot/core/datasource.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/starbot/core/datasource.py b/starbot/core/datasource.py index a50e9a9..8a7fda9 100644 --- a/starbot/core/datasource.py +++ b/starbot/core/datasource.py @@ -11,7 +11,7 @@ from pydantic import ValidationError from .model import LiveOn, LiveOff, LiveReport, DynamicUpdate, PushTarget, PushType from .room import Up from .sender import Bot -from ..exception.DataSourceException import DataSourceException +from ..exception import LiveException, DataSourceException from ..utils import config @@ -480,4 +480,7 @@ class MySQLDataSource(DataSource): super().format_data() logger.success(f"已成功载入 UID: {uid} 的推送配置") - await up.connect() + try: + await up.connect() + except LiveException as ex: + logger.error(ex.msg)