|
@@ -1,3 +1,5 @@
|
|
|
+import logging
|
|
|
+
|
|
|
import requests
|
|
|
|
|
|
|
|
@@ -6,17 +8,32 @@ class NasToolsClient:
|
|
|
self.base_url = base_url
|
|
|
self.api_key = api_key
|
|
|
self.token = None
|
|
|
+ # def login(self, username, password):
|
|
|
+ # """
|
|
|
+ # 登录以获取Token
|
|
|
+ # """
|
|
|
+ # url = f'{self.base_url}/api/v1/user/login'
|
|
|
+ # data = {'username': username, 'password': password}
|
|
|
+ # headers = {'Content-Type': 'application/x-www-form-urlencoded'}
|
|
|
+ # response = requests.post(url, data=data, headers=headers)
|
|
|
+ # if response.status_code == 200:
|
|
|
+ # self.token = response.json().get('token')
|
|
|
+ # return self.token
|
|
|
+ # else:
|
|
|
+ # raise Exception("Login failed with status code: " + str(response.status_code))
|
|
|
|
|
|
def login(self, username, password):
|
|
|
"""登录并获取令牌"""
|
|
|
- login_url = f"{self.base_url}/user/login"
|
|
|
+ url = f'{self.base_url}/api/v1/user/login'
|
|
|
data = {
|
|
|
"username": username,
|
|
|
"password": password
|
|
|
}
|
|
|
- response = requests.post(login_url, json=data)
|
|
|
+ headers = {'Content-Type': 'application/x-www-form-urlencoded'}
|
|
|
+ response = requests.post(url, data=data, headers=headers)
|
|
|
if response.status_code == 200:
|
|
|
self.token = response.json().get("data", {}).get("token")
|
|
|
+ logging.info(self.token)
|
|
|
return self.token
|
|
|
else:
|
|
|
raise Exception("Failed to login:", response.text)
|