fix: 修复房间号获取格式

setup.py

@@ -5,7 +5,7 @@
 
 setup(
  name='tsugu-api-python',
- version='1.1.0',
+ version='1.1.1',
  author='WindowsSov8',
  author_email='[email protected]',
  description='Tsugu BanGDream Bot 的功能 API 统合包',

tsugu_api/_bandoristation.py

@@ -1,7 +1,7 @@
 from tsugu_api import settings
 from tsugu_api._network import Api
 from tsugu_api._typing import _StationRoom
-from tsugu_api.exception import RoomSubmitFailure
+from tsugu_api.exception import RoomQueryFailure, RoomSubmitFailure
 
 BANDORI_STATION_URL = 'https://api.bandoristation.com/index.php'
 
@@ -47,7 +47,10 @@ def query_room_number() -> list[_StationRoom]:
  }
 
  # 发送请求
- return Api(
+ response = Api(
  BANDORI_STATION_URL,
  proxy=settings.backend_proxy
  ).get(params).json()
+ if response['status'] == 'failure':
+ raise RoomQueryFailure(response['response'])
+ return response['response']

tsugu_api/exception.py

@@ -1,5 +1,12 @@
 class RoomSubmitFailure(BaseException):
- '''车牌号上传失败'''
+ '''房间号上传失败'''
+ def __init__(self, response: str) -> None:
+ '''初始化'''
+ self.message = response
+ return
+
+class RoomQueryFailure(BaseException):
+ '''房间号获取失败'''
  def __init__(self, response: str) -> None:
  '''初始化'''
  self.message = response

tsugu_api_async/_bandoristation.py

@@ -3,7 +3,7 @@
 from tsugu_api_async import settings
 from tsugu_api_async._network import Api
 from tsugu_api_async._typing import _StationRoom
-from tsugu_api_async.exception import RoomSubmitFailure
+from tsugu_api_async.exception import RoomQueryFailure, RoomSubmitFailure
 
 BANDORI_STATION_URL = 'https://api.bandoristation.com/index.php'
 
@@ -55,5 +55,8 @@ async def query_room_number() -> list[_StationRoom]:
  BANDORI_STATION_URL,
  proxy=settings.backend_proxy
  ).get(params)
- if isinstance(response, Response): return response.json()
- else: return await response.json()
+ if isinstance(response, Response): response = response.json()
+ else: response = await response.json()
+ if response['status'] == 'failure':
+ raise RoomQueryFailure(response['response'])
+ return response['response']

tsugu_api_async/exception.py

@@ -1,5 +1,12 @@
 class RoomSubmitFailure(BaseException):
- '''车牌号上传失败'''
+ '''房间号上传失败'''
+ def __init__(self, response: str) -> None:
+ '''初始化'''
+ self.message = response
+ return
+
+class RoomQueryFailure(BaseException):
+ '''房间号获取失败'''
  def __init__(self, response: str) -> None:
  '''初始化'''
  self.message = response