72 lines
2.4 KiB
Python
72 lines
2.4 KiB
Python
import os
|
||
|
||
def generate_files(start_sn, count):
|
||
# 1. 解析起始序列号 (分离前缀和最后4位数字)
|
||
# 假设格式固定，最后4位为数字
|
||
prefix = start_sn[:-4]
|
||
try:
|
||
start_num = int(start_sn[-4:])
|
||
except ValueError:
|
||
print("错误: 序列号最后4位必须是数字。")
|
||
return
|
||
|
||
csv_filename = "yz_project.csv"
|
||
docker_filename = "docker-compose.yml"
|
||
|
||
csv_content = []
|
||
docker_services = []
|
||
|
||
# docker-compose 头部定义
|
||
docker_header = """version: '3'
|
||
x-service-common: &service-common
|
||
image: swr.cn-north-4.myhuaweicloud.com/cloud-yuzhen/yuzhen_base:v1.0.11
|
||
restart: always
|
||
volumes:
|
||
- ./yz_project.csv:/home/yz_project.csv
|
||
|
||
services:"""
|
||
|
||
print(f"正在生成 {count} 个序列号，起始于: {start_sn}...")
|
||
|
||
for i in range(count):
|
||
# 计算当前数字并补零至4位 (例如: 1 -> 0001)
|
||
current_num = start_num + i
|
||
current_sn = f"{prefix}{current_num:04d}"
|
||
|
||
# --- 生成 CSV 行 ---
|
||
# 格式: ${device_SN},Horseman-A21,H21.0.0.1,192.168.1.244,1883,mxpoty5rco1jhwde|${device_SN}|2,pykqg5qpzspfce3w,0,192.168.1.244,http://192.168.1.244/dlp/v1/device/backupAndRestore?sn=${device_SN},,
|
||
|
||
csv_line = (
|
||
f"{current_sn},Horseman-A21,H21.0.0.1,192.168.1.244,1883,mxpoty5rco1jhwde|{current_sn}|2,pykqg5qpzspfce3w,0,192.168.1.244,http://192.168.1.244/dlp/v1/device/backupAndRestore?sn={current_sn}"
|
||
)
|
||
csv_content.append(csv_line)
|
||
|
||
# --- 生成 Docker Compose 服务块 ---
|
||
# 注意 YAML 缩进
|
||
service_block = f"""
|
||
{current_sn}:
|
||
<<: *service-common
|
||
command: [ "/home/start.sh", "{current_sn}" ]"""
|
||
docker_services.append(service_block)
|
||
|
||
# 2. 写入 CSV 文件
|
||
with open(csv_filename, 'w', encoding='utf-8') as f:
|
||
f.write("\n".join(csv_content))
|
||
# 如果文件末尾需要换行符，取消下面注释
|
||
# f.write("\n")
|
||
print(f"已创建: {csv_filename}")
|
||
|
||
# 3. 写入 Docker Compose 文件
|
||
with open(docker_filename, 'w', encoding='utf-8') as f:
|
||
f.write(docker_header)
|
||
f.write("".join(docker_services))
|
||
print(f"已创建: {docker_filename}")
|
||
|
||
if __name__ == "__main__":
|
||
# --- 配置区域 ---
|
||
START_SN = "YZXHRAX86NN2511250001" # 起始序列号
|
||
COUNT = 166 # 需要生成的个数
|
||
# ----------------
|
||
|
||
generate_files(START_SN, COUNT)
|