mirror of
https://github.com/Sitoi/dailycheckin.git
synced 2024-11-17 13:48:03 +08:00
🐛 修复(main.py):使用.get()方法获取access_token,以防止KeyError
✨ 功能(main.py):添加对access_token为空的检查,并在token过期时返回错误信息
This commit is contained in:
parent
474c01aa52
commit
e8f6efdda9
@ -19,7 +19,7 @@ class AliYun(CheckIn):
|
||||
url = "https://auth.aliyundrive.com/v2/account/token"
|
||||
data = {"grant_type": "refresh_token", "refresh_token": refresh_token}
|
||||
response = requests.post(url=url, json=data).json()
|
||||
access_token = response["access_token"]
|
||||
access_token = response.get("access_token")
|
||||
return access_token
|
||||
|
||||
def sign(self, access_token):
|
||||
@ -62,6 +62,8 @@ class AliYun(CheckIn):
|
||||
def main(self):
|
||||
refresh_token = self.check_item.get("refresh_token")
|
||||
access_token = self.update_token(refresh_token)
|
||||
if not access_token:
|
||||
return [{"name": "阿里云盘", "value": "token 过期"}]
|
||||
msg = self.sign(access_token)
|
||||
msg = "\n".join([f"{one.get('name')}: {one.get('value')}" for one in msg])
|
||||
return msg
|
||||
|
Loading…
Reference in New Issue
Block a user