mirror of
https://github.com/snowykami/server-status-client.git
synced 2026-01-25 13:11:54 +00:00
Compare commits
17 Commits
22feee38ee
...
main
| Author | SHA1 | Date | |
|---|---|---|---|
| 7ee3eb377e | |||
| 90b4952ea9 | |||
| 0559212d03 | |||
| e74324e21b | |||
| 48ac46bbb3 | |||
| b5ba568a66 | |||
| b30b3429ad | |||
| 7cbca85a60 | |||
| ce3a07e350 | |||
| 9160e7c4b4 | |||
| 8e2c5c24b8 | |||
| c63d84106b | |||
| 0d7b7700de | |||
| b5bf2c1de6 | |||
| 87418edeea | |||
| 5c089873ca | |||
| 79242f9dd6 |
@@ -31,7 +31,7 @@ _✨ 服务器状态 - 客户端 ✨_
|
|||||||
```shell
|
```shell
|
||||||
sudo bash -c "$(curl -sSL https://raw.githubusercontent.com/snowykami/server-status-client/refs/heads/main/deploy.sh)"
|
sudo bash -c "$(curl -sSL https://raw.githubusercontent.com/snowykami/server-status-client/refs/heads/main/deploy.sh)"
|
||||||
```
|
```
|
||||||
如果位于中国大陆,可使用国内镜像
|
如果位于中国大陆无法访问GitHub,可使用中国版脚本
|
||||||
```shell
|
```shell
|
||||||
sudo bash -c "$(curl -sSL https://git.liteyuki.icu/snowykami/server-status-client/raw/branch/main/deploy-cn.sh)"
|
sudo bash -c "$(curl -sSL https://git.liteyuki.icu/snowykami/server-status-client/raw/branch/main/deploy-cn.sh)"
|
||||||
```
|
```
|
||||||
@@ -57,8 +57,8 @@ _✨ 服务器状态 - 客户端 ✨_
|
|||||||
|
|
||||||
### 命令
|
### 命令
|
||||||
|
|
||||||
- `server-status <server> <token> <id> run` - 运行客户端
|
- `python main.py <server> <token> <id> run` - 运行客户端
|
||||||
- `server-status <server> <token> <id> rm` - 从服务端移除主机
|
- `python.main.py <server> <token> <id> rm` - 从服务端移除主机
|
||||||
|
|
||||||
#### 可选项
|
#### 可选项
|
||||||
|
|
||||||
@@ -71,7 +71,7 @@ _✨ 服务器状态 - 客户端 ✨_
|
|||||||
#### 示例
|
#### 示例
|
||||||
|
|
||||||
```shell
|
```shell
|
||||||
server_status https://status.liteyuki.icu 114514 myhost run -n "MyHost" --labels "标签1,标签2" --interval 5 --location "Chongqing" --link "https://example.com"
|
python main.py https://status.liteyuki.icu 114514 myhost run -n "MyHost" --labels "标签1,标签2" --interval 5 --location "Chongqing" --link "https://example.com"
|
||||||
```
|
```
|
||||||
|
|
||||||
## 📝 其他
|
## 📝 其他
|
||||||
|
|||||||
34
deploy-cn.sh
34
deploy-cn.sh
@@ -10,40 +10,36 @@ fi
|
|||||||
|
|
||||||
# check install dir
|
# check install dir
|
||||||
install_dir="/opt"
|
install_dir="/opt"
|
||||||
echo -n "Install directory? (default: $install_dir/server-status-client): " && read -r install_dir_input
|
echo -n "安装目录? (默认: $install_dir/server-status-client): " && read -r install_dir_input
|
||||||
if [ -n "$install_dir_input" ]; then
|
if [ -n "$install_dir_input" ]; then
|
||||||
install_dir="$install_dir_input"
|
install_dir="$install_dir_input"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
# check server
|
# check server
|
||||||
echo -n "Server? (required): " && read -r server
|
echo -n "服务端地址? (必须): " && read -r server
|
||||||
if [ -z "$server" ]; then
|
if [ -z "$server" ]; then
|
||||||
echo "Server is required"
|
echo "服务端地址是必须的"
|
||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
# check token
|
# check token
|
||||||
echo -n "Token? (required): " && read -r token
|
echo -n "令牌? (必须或留空): " && read -r token
|
||||||
if [ -z "$token" ]; then
|
|
||||||
echo "Token is required"
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
# check hostname
|
# check hostname
|
||||||
hostname=$(hostname)
|
hostname=$(hostname)
|
||||||
echo -n "Hostname? (default: $hostname): " && read -r hostname_input
|
echo -n "此主机名? (默认: $hostname): " && read -r hostname_input
|
||||||
if [ -n "$hostname_input" ]; then
|
if [ -n "$hostname_input" ]; then
|
||||||
hostname="$hostname_input"
|
hostname="$hostname_input"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
# labels
|
# labels
|
||||||
echo -n "Labels? (space separated): " && read -r labels_input
|
echo -n "标签们? (空格分隔): " && read -r labels_input
|
||||||
if [ -n "$labels_input" ]; then
|
if [ -n "$labels_input" ]; then
|
||||||
labels="$labels_input"
|
labels="$labels_input"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
# location
|
# location
|
||||||
echo -n "Location? (optional): " && read -r location_input
|
echo -n "地理位置? (可选|自定义): " && read -r location_input
|
||||||
if [ -n "$location_input" ]; then
|
if [ -n "$location_input" ]; then
|
||||||
location="$location_input"
|
location="$location_input"
|
||||||
fi
|
fi
|
||||||
@@ -52,7 +48,7 @@ fi
|
|||||||
repo2="https://git.liteyuki.icu/snowykami/server-status-client"
|
repo2="https://git.liteyuki.icu/snowykami/server-status-client"
|
||||||
# try 1 if failed try 2
|
# try 1 if failed try 2
|
||||||
git clone "$repo2" "$install_dir/server-status-client"
|
git clone "$repo2" "$install_dir/server-status-client"
|
||||||
cd "$install_dir/server-status-client" || { echo "Failed to clone repo"; exit 1; }
|
cd "$install_dir/server-status-client" || { echo "克隆失败"; exit 1; }
|
||||||
|
|
||||||
# create venv
|
# create venv
|
||||||
python3 -m venv venv
|
python3 -m venv venv
|
||||||
@@ -60,22 +56,22 @@ python_exe="./venv/bin/python"
|
|||||||
|
|
||||||
# check if venv is created
|
# check if venv is created
|
||||||
if [ ! -f "$python_exe" ]; then
|
if [ ! -f "$python_exe" ]; then
|
||||||
echo "Failed to create venv"
|
echo "创建虚拟环境失败"
|
||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
echo "venv created successfully"
|
echo "虚拟环境创建成功"
|
||||||
|
|
||||||
# install the required packages
|
# install the required packages
|
||||||
echo "Installing the required packages"
|
echo "正在安装依赖包..."
|
||||||
$python_exe -m pip install pdm -i https://pypi.tuna.tsinghua.edu.cn/simple
|
$python_exe -m pip install pdm -i https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple
|
||||||
$python_exe -m pdm install
|
$python_exe -m pdm install
|
||||||
|
|
||||||
# create the systemd service
|
# create the systemd service
|
||||||
echo "Creating the systemd service"
|
echo "正在创建服务..."
|
||||||
|
|
||||||
# generate random id
|
# generate random id
|
||||||
# shellcheck disable=SC2002
|
# shellcheck disable=SC2002
|
||||||
id=$(cat /dev/urandom | tr -dc 'a-zA-Z0-9' | fold -w 32 | head -n 1)
|
id=$(cat /dev/urandom | tr -dc 'a-zA-Z0-9' | fold -w 8 | head -n 1)
|
||||||
|
|
||||||
bash -c "cat <<EOF > /etc/systemd/system/server-status-client.service
|
bash -c "cat <<EOF > /etc/systemd/system/server-status-client.service
|
||||||
[Unit]
|
[Unit]
|
||||||
@@ -97,4 +93,4 @@ EOF"
|
|||||||
systemctl enable server-status-client
|
systemctl enable server-status-client
|
||||||
systemctl start server-status-client
|
systemctl start server-status-client
|
||||||
|
|
||||||
echo "server-status-client installed successfully"
|
echo "安装完成,服务已启动"
|
||||||
@@ -74,7 +74,7 @@ echo "Creating the systemd service"
|
|||||||
|
|
||||||
# generate random id
|
# generate random id
|
||||||
# shellcheck disable=SC2002
|
# shellcheck disable=SC2002
|
||||||
id=$(cat /dev/urandom | tr -dc 'a-zA-Z0-9' | fold -w 32 | head -n 1)
|
id=$(cat /dev/urandom | tr -dc 'a-zA-Z0-9' | fold -w 8 | head -n 1)
|
||||||
|
|
||||||
bash -c "cat <<EOF > /etc/systemd/system/server-status-client.service
|
bash -c "cat <<EOF > /etc/systemd/system/server-status-client.service
|
||||||
[Unit]
|
[Unit]
|
||||||
|
|||||||
@@ -31,5 +31,5 @@ tag_filter = "v*"
|
|||||||
tag_regex = '^v(?:\D*)?(?P<version>([1-9][0-9]*!)?(0|[1-9][0-9]*)(\.(0|[1-9][0-9]*))*((a|b|c|rc)(0|[1-9][0-9]*))?(\.post(0|[1-9][0-9]*))?(\.dev(0|[1-9][0-9]*))?$)$'
|
tag_regex = '^v(?:\D*)?(?P<version>([1-9][0-9]*!)?(0|[1-9][0-9]*)(\.(0|[1-9][0-9]*))*((a|b|c|rc)(0|[1-9][0-9]*))?(\.post(0|[1-9][0-9]*))?(\.dev(0|[1-9][0-9]*))?$)$'
|
||||||
|
|
||||||
[[tool.pdm.source]]
|
[[tool.pdm.source]]
|
||||||
name = "tuna"
|
name = "pypi"
|
||||||
url = "https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple"
|
url = "https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple"
|
||||||
@@ -8,7 +8,41 @@ import requests
|
|||||||
|
|
||||||
from server_status.timezone import get_timezone
|
from server_status.timezone import get_timezone
|
||||||
|
|
||||||
excluded_partition_prefix = ("/var", "/boot", "/run", "/proc", "/sys", "/dev", "/tmp", "/snap")
|
excluded_partition_prefix = (
|
||||||
|
"/var",
|
||||||
|
"/boot",
|
||||||
|
"/run",
|
||||||
|
"/proc",
|
||||||
|
"/sys",
|
||||||
|
"/dev",
|
||||||
|
"/tmp",
|
||||||
|
"/snap",
|
||||||
|
|
||||||
|
"/System",
|
||||||
|
"/Applications",
|
||||||
|
"/private",
|
||||||
|
"/Library",
|
||||||
|
)
|
||||||
|
|
||||||
|
include_partition_prefix_mac = ("/Volumes")
|
||||||
|
|
||||||
|
os_name = "" # linux下为发行版名称,windows下为Windows, macOS下为Darwin
|
||||||
|
os_version = "" # linux下为发行版版本,windows下为Windows版本
|
||||||
|
try:
|
||||||
|
# read /etc/os-release
|
||||||
|
with open("/etc/os-release") as f:
|
||||||
|
os_release = f.read()
|
||||||
|
# 找到NAME=和VERSION=的行
|
||||||
|
for line in os_release.split("\n"):
|
||||||
|
if line.startswith("NAME="):
|
||||||
|
os_name = line.split("=")[1].replace('"', "")
|
||||||
|
elif line.startswith("VERSION_ID="):
|
||||||
|
os_version = line.split("=")[1].replace('"', "")
|
||||||
|
except FileNotFoundError:
|
||||||
|
os_name = platform.system()
|
||||||
|
os_version = platform.release()
|
||||||
|
|
||||||
|
print("Current OS:", os_name, os_version)
|
||||||
|
|
||||||
|
|
||||||
def log(*args):
|
def log(*args):
|
||||||
@@ -29,6 +63,7 @@ def get_network_speed(interval) -> tuple[int, int]:
|
|||||||
|
|
||||||
|
|
||||||
class Hardware:
|
class Hardware:
|
||||||
|
os_release: str = ""
|
||||||
mem_total: int = psutil.virtual_memory().total
|
mem_total: int = psutil.virtual_memory().total
|
||||||
mem_used: int = psutil.virtual_memory().used
|
mem_used: int = psutil.virtual_memory().used
|
||||||
|
|
||||||
@@ -112,7 +147,9 @@ class Api:
|
|||||||
"""
|
"""
|
||||||
self.headers.update(self.format(headers))
|
self.headers.update(self.format(headers))
|
||||||
|
|
||||||
def format(self, obj: str | list[str] | dict[str, Any]) -> str | list[str] | dict[str, Any]:
|
def format(
|
||||||
|
self, obj: str | list[str] | dict[str, Any]
|
||||||
|
) -> str | list[str] | dict[str, Any]:
|
||||||
if isinstance(obj, str):
|
if isinstance(obj, str):
|
||||||
obj = obj.format(**self.variables)
|
obj = obj.format(**self.variables)
|
||||||
elif isinstance(obj, dict):
|
elif isinstance(obj, dict):
|
||||||
@@ -125,8 +162,17 @@ class Api:
|
|||||||
|
|
||||||
|
|
||||||
class Client:
|
class Client:
|
||||||
def __init__(self, addr: str, token: str, client_id: str, name: str = "", location: str = "", labels: list[str] = [], link: str = "",
|
def __init__(
|
||||||
interval: int = 2):
|
self,
|
||||||
|
addr: str,
|
||||||
|
token: str,
|
||||||
|
client_id: str,
|
||||||
|
name: str = "",
|
||||||
|
location: str = "",
|
||||||
|
labels: list[str] = [],
|
||||||
|
link: str = "",
|
||||||
|
interval: int = 2,
|
||||||
|
):
|
||||||
self.api = Api(addr, {"token": token, "id": client_id})
|
self.api = Api(addr, {"token": token, "id": client_id})
|
||||||
self.api = self.api.group("/client")
|
self.api = self.api.group("/client")
|
||||||
self.api.add_headers(Authorization="{token}")
|
self.api.add_headers(Authorization="{token}")
|
||||||
@@ -140,15 +186,66 @@ class Client:
|
|||||||
self.link = link
|
self.link = link
|
||||||
self.interval = interval
|
self.interval = interval
|
||||||
|
|
||||||
self.start_time = psutil.boot_time()
|
self.start_time: float = psutil.boot_time()
|
||||||
|
|
||||||
self.hardware = Hardware()
|
self.hardware = Hardware()
|
||||||
|
|
||||||
log("Client initialized",
|
log(
|
||||||
f"Name: {self.name}({self.client_id}), Location: {self.location}, Labels: {self.labels}")
|
"Client initialized",
|
||||||
|
f"Name: {self.name}({self.client_id}), Location: {self.location}, Labels: {self.labels}",
|
||||||
|
)
|
||||||
|
|
||||||
def start(self):
|
def start(self):
|
||||||
self.observe()
|
log("Starting client")
|
||||||
|
threading.Thread(target=self._start_obs, daemon=True).start()
|
||||||
|
threading.Thread(target=self._start_post, daemon=True).start()
|
||||||
|
|
||||||
|
while True:
|
||||||
|
time.sleep(1)
|
||||||
|
|
||||||
|
def _start_obs(self):
|
||||||
|
"""启动监控记录线程"""
|
||||||
|
while True:
|
||||||
|
try:
|
||||||
|
self.hardware.mem_total = psutil.virtual_memory().total
|
||||||
|
self.hardware.mem_used = psutil.virtual_memory().used
|
||||||
|
self.hardware.swap_total = psutil.swap_memory().total
|
||||||
|
self.hardware.swap_used = psutil.swap_memory().used
|
||||||
|
self.hardware.cpu_cores = psutil.cpu_count(logical=False)
|
||||||
|
self.hardware.cpu_logics = psutil.cpu_count(logical=True)
|
||||||
|
for part in psutil.disk_partitions():
|
||||||
|
try:
|
||||||
|
usage = psutil.disk_usage(part.mountpoint)
|
||||||
|
|
||||||
|
if (
|
||||||
|
(
|
||||||
|
platform.system() in ("Linux", "Darwin")
|
||||||
|
and (
|
||||||
|
part.mountpoint.startswith(
|
||||||
|
excluded_partition_prefix
|
||||||
|
)
|
||||||
|
)
|
||||||
|
)
|
||||||
|
):
|
||||||
|
continue
|
||||||
|
|
||||||
|
self.hardware.disks[part.device] = {
|
||||||
|
"mountpoint": part.mountpoint,
|
||||||
|
"device": part.device,
|
||||||
|
"fstype": part.fstype,
|
||||||
|
"total": usage.total,
|
||||||
|
"used": usage.used,
|
||||||
|
}
|
||||||
|
except:
|
||||||
|
pass
|
||||||
|
self.hardware.cpu_percent = psutil.cpu_percent(1)
|
||||||
|
self.hardware.net_up, self.hardware.net_down = get_network_speed(1)
|
||||||
|
log("Observed")
|
||||||
|
except Exception as e:
|
||||||
|
log(f"Failed to observe: {e}")
|
||||||
|
|
||||||
|
def _start_post(self):
|
||||||
|
"""启动上报进程"""
|
||||||
while True:
|
while True:
|
||||||
try:
|
try:
|
||||||
resp = self.get_ping()
|
resp = self.get_ping()
|
||||||
@@ -156,9 +253,13 @@ class Client:
|
|||||||
log(f"Connected to server {self.addr}")
|
log(f"Connected to server {self.addr}")
|
||||||
break
|
break
|
||||||
else:
|
else:
|
||||||
log(f"Failed to connect to server {self.addr}, retrying in 5 seconds: {resp.text}")
|
log(
|
||||||
|
f"Failed to connect to server {self.addr}, retrying in 5 seconds: {resp.text}"
|
||||||
|
)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
log(f"Failed to connect to server {self.addr}, retrying in 5 seconds: {e}")
|
log(
|
||||||
|
f"Failed to connect to server {self.addr}, retrying in 5 seconds: {e}"
|
||||||
|
)
|
||||||
time.sleep(5)
|
time.sleep(5)
|
||||||
|
|
||||||
while True:
|
while True:
|
||||||
@@ -185,10 +286,11 @@ class Client:
|
|||||||
"id": self.client_id,
|
"id": self.client_id,
|
||||||
"name": self.name,
|
"name": self.name,
|
||||||
"os": {
|
"os": {
|
||||||
"name": platform.system(),
|
"name": platform.system(), # 系统类型 linux|windows|darwin
|
||||||
"version": platform.version(),
|
"version": os_name + os_version,
|
||||||
"machine": platform.machine(),
|
# 系统版本复杂描述 #1 SMP PREEMPT_DYNAMIC Fri Sep 13 10:42:50 UTC 2024 (5c05eeb)
|
||||||
"release": platform.release(),
|
"machine": platform.machine(), # 机器类型 x86_64
|
||||||
|
"release": os_version, # 系统版本
|
||||||
},
|
},
|
||||||
"labels": self.labels,
|
"labels": self.labels,
|
||||||
"location": self.location,
|
"location": self.location,
|
||||||
@@ -221,42 +323,5 @@ class Client:
|
|||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
def observe(self):
|
|
||||||
"""
|
|
||||||
观察硬件状态并更新
|
|
||||||
Returns:
|
|
||||||
|
|
||||||
"""
|
|
||||||
|
|
||||||
def _observe():
|
|
||||||
while True:
|
|
||||||
self.hardware.mem_total = psutil.virtual_memory().total
|
|
||||||
self.hardware.mem_used = psutil.virtual_memory().used
|
|
||||||
self.hardware.swap_total = psutil.swap_memory().total
|
|
||||||
self.hardware.swap_used = psutil.swap_memory().used
|
|
||||||
self.hardware.cpu_cores = psutil.cpu_count(logical=False)
|
|
||||||
self.hardware.cpu_logics = psutil.cpu_count(logical=True)
|
|
||||||
for part in psutil.disk_partitions():
|
|
||||||
try:
|
|
||||||
usage = psutil.disk_usage(part.mountpoint)
|
|
||||||
|
|
||||||
if part.mountpoint.startswith(excluded_partition_prefix) or usage.total == 0:
|
|
||||||
continue
|
|
||||||
|
|
||||||
self.hardware.disks[part.device] = {
|
|
||||||
"mountpoint": part.mountpoint,
|
|
||||||
"device": part.device,
|
|
||||||
"fstype": part.fstype,
|
|
||||||
"total": usage.total,
|
|
||||||
"used": usage.used,
|
|
||||||
}
|
|
||||||
except:
|
|
||||||
pass
|
|
||||||
self.hardware.cpu_percent = psutil.cpu_percent(1)
|
|
||||||
self.hardware.net_up, self.hardware.net_down = get_network_speed(1)
|
|
||||||
log("Observed")
|
|
||||||
|
|
||||||
threading.Thread(target=_observe, daemon=True).start()
|
|
||||||
|
|
||||||
def remove(self, client_id) -> requests.Response:
|
def remove(self, client_id) -> requests.Response:
|
||||||
return self.api.delete("/host", data={"id": client_id})
|
return self.api.delete("/host", data={"id": client_id})
|
||||||
|
|||||||
@@ -3,7 +3,7 @@ import socket
|
|||||||
|
|
||||||
from arclet.alconna import Alconna, Subcommand, Option, Args, MultiVar
|
from arclet.alconna import Alconna, Subcommand, Option, Args, MultiVar
|
||||||
|
|
||||||
server_status_alc = Alconna(
|
server_status_alc = Alconna( # type: ignore
|
||||||
"server_status",
|
"server_status",
|
||||||
Args["server", str]["token", str]["id", str],
|
Args["server", str]["token", str]["id", str],
|
||||||
Subcommand(
|
Subcommand(
|
||||||
|
|||||||
Reference in New Issue
Block a user