Dt8333
7dd95d8a59
* chore: auto fix by ruff * refactor: 统一修正返回类型注解为 None/bool 以匹配实现 * refactor: 将 _get_next_page 改为异步并移除多余的请求错误抛出 * refactor: 将 get_client 的返回类型改为 object * style: 为 LarkMessageEvent 的相关方法添加返回类型注解 None --------- Co-authored-by: Soulter <37870767+Soulter@users.noreply.github.com>
78 lines
2.5 KiB
Python
78 lines
2.5 KiB
Python
import os
|
|
import socket
|
|
import sys
|
|
import uuid
|
|
|
|
import aiohttp
|
|
|
|
from astrbot.core import db_helper, logger
|
|
from astrbot.core.config import VERSION
|
|
|
|
|
|
class Metric:
|
|
_iid_cache = None
|
|
|
|
@staticmethod
|
|
def get_installation_id():
|
|
"""获取或创建一个唯一的安装ID"""
|
|
if Metric._iid_cache is not None:
|
|
return Metric._iid_cache
|
|
|
|
config_dir = os.path.join(os.path.expanduser("~"), ".astrbot")
|
|
id_file = os.path.join(config_dir, ".installation_id")
|
|
|
|
if os.path.exists(id_file):
|
|
try:
|
|
with open(id_file) as f:
|
|
Metric._iid_cache = f.read().strip()
|
|
return Metric._iid_cache
|
|
except Exception:
|
|
pass
|
|
try:
|
|
os.makedirs(config_dir, exist_ok=True)
|
|
installation_id = str(uuid.uuid4())
|
|
with open(id_file, "w") as f:
|
|
f.write(installation_id)
|
|
Metric._iid_cache = installation_id
|
|
return installation_id
|
|
except Exception:
|
|
Metric._iid_cache = "null"
|
|
return "null"
|
|
|
|
@staticmethod
|
|
async def upload(**kwargs) -> None:
|
|
"""上传相关非敏感的指标以更好地了解 AstrBot 的使用情况。上传的指标不会包含任何有关消息文本、用户信息等敏感信息。
|
|
|
|
Powered by TickStats.
|
|
"""
|
|
if os.environ.get("ASTRBOT_DISABLE_METRICS", "0") == "1":
|
|
return
|
|
base_url = "https://tickstats.soulter.top/api/metric/90a6c2a1"
|
|
kwargs["v"] = VERSION
|
|
kwargs["os"] = sys.platform
|
|
payload = {"metrics_data": kwargs}
|
|
try:
|
|
kwargs["hn"] = socket.gethostname()
|
|
except Exception:
|
|
pass
|
|
try:
|
|
kwargs["iid"] = Metric.get_installation_id()
|
|
except Exception:
|
|
pass
|
|
try:
|
|
if "adapter_name" in kwargs:
|
|
await db_helper.insert_platform_stats(
|
|
platform_id=kwargs["adapter_name"],
|
|
platform_type=kwargs.get("adapter_type", "unknown"),
|
|
)
|
|
except Exception as e:
|
|
logger.error(f"保存指标到数据库失败: {e}")
|
|
|
|
try:
|
|
async with aiohttp.ClientSession(trust_env=True) as session:
|
|
async with session.post(base_url, json=payload, timeout=3) as response:
|
|
if response.status != 200:
|
|
pass
|
|
except Exception:
|
|
pass
|