66 lines
2.1 KiB
Python
66 lines
2.1 KiB
Python
import requests
|
||
import os
|
||
|
||
# 新增函数：从api.txt获取服务器地址（与登录模块保持一致）
|
||
def get_api_base_url():
|
||
try:
|
||
with open("api.txt", "r") as f:
|
||
lines = [line.strip() for line in f.readlines() if line.strip()]
|
||
if lines:
|
||
return lines[0].rstrip('/')
|
||
except FileNotFoundError:
|
||
print("api.txt文件未找到，使用默认地址")
|
||
except Exception as e:
|
||
print(f"读取api.txt失败: {e}，使用默认地址")
|
||
return "http://localhost:3000"
|
||
|
||
# 获取云盘信息（修改了URL生成方式）
|
||
def get_cloud_info(cookie):
|
||
base_url = get_api_base_url()
|
||
url = f"{base_url}/user/cloud"
|
||
|
||
# 保持参数传递方式与原始代码一致
|
||
params = {
|
||
"cookie": cookie,
|
||
"limit": 1
|
||
}
|
||
|
||
response = requests.get(url, params=params)
|
||
try:
|
||
response_data = response.json()
|
||
if response_data.get('code') == 200:
|
||
# 数据提取保持原始逻辑
|
||
total_size = response_data['size']
|
||
max_size = response_data['maxSize']
|
||
file_count = response_data['count']
|
||
|
||
print("\n云盘信息:")
|
||
print(f"▸ 已用空间: {convert_bytes(total_size)}")
|
||
print(f"▸ 总容量: {convert_bytes(max_size)}")
|
||
print(f"▸ 文件总数: {file_count} 个")
|
||
else:
|
||
print(f"获取失败: {response_data.get('message', '未知错误')}")
|
||
except requests.JSONDecodeError:
|
||
print("响应解析失败，原始内容:", response.text)
|
||
|
||
# 优化后的单位转换函数
|
||
def convert_bytes(size_in_bytes):
|
||
"""智能转换存储单位"""
|
||
units = ('B', 'KB', 'MB', 'GB', 'TB', 'PB')
|
||
try:
|
||
size = float(size_in_bytes)
|
||
for unit in units:
|
||
if size < 1024.0:
|
||
break
|
||
size /= 1024.0
|
||
return f"{size:.2f} {unit}"
|
||
except (TypeError, ValueError):
|
||
return "0.00 B"
|
||
|
||
# 使用示例
|
||
if __name__ == "__main__":
|
||
# 从文件读取cookie（模拟登录模块）
|
||
with open("cookies.txt", "r") as f:
|
||
test_cookie = f.read().strip()
|
||
|
||
get_cloud_info(test_cookie) |