zero/dailycheckin
1
0
Fork 0
mirror of https://github.com/Sitoi/dailycheckin.git synced 2024-11-17 13:48:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

🐛 修复(main.py)：使用.get()方法获取access_token，以防止KeyError

Browse Source 
✨ 功能(main.py)：添加对access_token为空的检查，并在token过期时返回错误信息
This commit is contained in:
shitao 2024-02-22 09:27:26 +08:00
parent 474c01aa52
commit e8f6efdda9
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dailycheckin/aliyun/main.py
View File

@ -19,7 +19,7 @@ class AliYun(CheckIn):
url = "https://auth.aliyundrive.com/v2/account/token" url = "https://auth.aliyundrive.com/v2/account/token"
data = {"grant_type": "refresh_token", "refresh_token": refresh_token} data = {"grant_type": "refresh_token", "refresh_token": refresh_token}
response = requests.post(url=url, json=data).json() response = requests.post(url=url, json=data).json()
access_token = response["access_token"] access_token = response.get("access_token")
return access_token return access_token
def sign(self, access_token): def sign(self, access_token):
@ -62,6 +62,8 @@ class AliYun(CheckIn):
def main(self): def main(self):
refresh_token = self.check_item.get("refresh_token") refresh_token = self.check_item.get("refresh_token")
access_token = self.update_token(refresh_token) access_token = self.update_token(refresh_token)
if not access_token:
return [{"name": "阿里云盘", "value": "token 过期"}]
msg = self.sign(access_token) msg = self.sign(access_token)
msg = "\n".join([f"{one.get('name')}: {one.get('value')}" for one in msg]) msg = "\n".join([f"{one.get('name')}: {one.get('value')}" for one in msg])
return msg return msg