mirror of https://github.com/Mai-with-u/MaiBot.git
74 lines
2.4 KiB
Python
74 lines
2.4 KiB
Python
#!/bin/python3
|
||
# 这个脚本的作用是在部署helm chart时动态生成adapter的配置文件，保存在configmap中
|
||
# 需要动态生成的原因是core服务的DNS名称是动态的，无法在adapter服务的配置文件中提前确定
|
||
# 一些与k8s现有资源冲突的配置也会在这里重置
|
||
|
||
import os
|
||
import toml
|
||
import base64
|
||
from kubernetes import client, config
|
||
from datetime import datetime, timezone
|
||
|
||
config.load_incluster_config()
|
||
core_api = client.CoreV1Api()
|
||
apps_api = client.AppsV1Api()
|
||
|
||
# 读取部署的关键信息
|
||
namespace = os.getenv("NAMESPACE")
|
||
release_name = os.getenv("RELEASE_NAME")
|
||
data_b64 = os.getenv("DATA_B64")
|
||
|
||
# 解析并覆盖关键配置
|
||
# 这里被覆盖的配置应当在helm chart中针对对应的k8s资源来灵活修改
|
||
data = toml.loads(base64.b64decode(data_b64).decode("utf-8"))
|
||
data.setdefault('napcat_server', {})
|
||
data['napcat_server']['host'] = '0.0.0.0'
|
||
data['napcat_server']['port'] = 8095
|
||
data.setdefault('maibot_server', {})
|
||
data['maibot_server']['host'] = f'{release_name}-maibot-core' # 根据release名称动态拼接core服务的DNS名称
|
||
data['maibot_server']['port'] = 8000
|
||
|
||
# 创建/修改configmap
|
||
cm_name = f'{release_name}-maibot-adapter-config'
|
||
cm = client.V1ConfigMap(
|
||
metadata=client.V1ObjectMeta(name=cm_name),
|
||
data={'config.toml': toml.dumps(data)}
|
||
)
|
||
try:
|
||
core_api.create_namespaced_config_map(namespace, cm)
|
||
print(f"ConfigMap `{cm_name}` created successfully")
|
||
except client.exceptions.ApiException as e:
|
||
if e.status == 409: # 已存在，更新
|
||
core_api.replace_namespaced_config_map(cm_name, namespace, cm)
|
||
print(f"ConfigMap `{cm_name}` replaced successfully")
|
||
else:
|
||
raise
|
||
|
||
# 重启adapter和core的statefulset
|
||
now = datetime.now(timezone.utc).isoformat()
|
||
body = {
|
||
"spec": {
|
||
"template": {
|
||
"metadata": {
|
||
"annotations": {
|
||
"kubectl.kubernetes.io/restartedAt": now
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
apps_api.patch_namespaced_stateful_set(
|
||
name=f'{release_name}-maibot-adapter',
|
||
namespace=namespace,
|
||
body=body,
|
||
)
|
||
print(f"StatefulSet `{release_name}-maibot-adapter` restarted successfully")
|
||
apps_api.patch_namespaced_stateful_set(
|
||
name=f'{release_name}-maibot-core',
|
||
namespace=namespace,
|
||
body=body,
|
||
)
|
||
print(f"StatefulSet `{release_name}-maibot-core` restarted successfully")
|
||
|
||
print('Job succeed.')
|