72 lines
2.3 KiB
Python
72 lines
2.3 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,121.36.13.175,1883,mxpoty5rco1jhwde|${device_SN}|2,pykqg5qpzspfce3w,0,121.36.13.175,http://121.36.13.175/dlp/v1/device/backupAndRestore?sn=${device_SN},,
|
|
|
|
csv_line = (
|
|
f"{current_sn},Horseman-A21,H21.0.0.1,121.36.13.175,1885,mxpoty5rco1jhwde|{current_sn}|2,pykqg5qpzspfce3w,0,121.36.13.175"
|
|
)
|
|
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 = "YZXHRAX86NN2604030001" # 起始序列号
|
|
COUNT = 2 # 需要生成的个数
|
|
# ----------------
|
|
|
|
generate_files(START_SN, COUNT)
|