使用python脚本自动生成K8S-YAML的方法示例

1、生成 servie.yaml

1.1、yaml转json

service模板yaml

apiVersion: v1

kind: Service

metadata:

name: ${jarName}

labels:

name: ${jarName}

version: v1

spec:

ports:

- port: ${port}

targetPort: ${port}

selector:

name: ${jarName}

转成json的结构

{

"apiVersion": "v1",

"kind": "Service",

"metadata": {

"name": "${jarName}",

"labels": {

"name": "${jarName}",

"version": "v1"

}

},

"spec": {

"ports": [

{

"port": "${port}",

"targetPort": "${port}"

}

],

"selector": {

"name": "${jarName}"

}

}

}

1.2、关键代码

# 通过传入service_name及ports列表

def create_service_yaml(service_name, ports):

# 将yaml读取为json,然后修改所有需要修改的${jarName}

service_data['metadata']['name'] = service_name

service_data['metadata']['labels']['name'] = service_name

service_data['spec']['selector']['name'] = service_name

# .spec.ports 比较特殊,是一个字典列表,由于传入的ports难以确定数量,难以直接修改

# 新建一个列表,遍历传入的ports列表,将传入的每个port都生成为一个字典,添加入新列表中

new_spec_ports = []

for port in ports:

port = int(port)

new_port = {'port': port, 'targetPort': port}

new_spec_ports.append(new_port)

# 修改.spec.ports为新列表

service_data['spec']['ports'] = new_spec_ports

2、生成 deployment.yaml

2.1、yaml转json

deployment模板yaml

apiVersion: apps/v1

kind: Deployment

metadata:

name: ${jarName}

labels:

name: ${jarName}

spec:

selector:

matchLabels:

name: ${jarName}

replicas: 1

template:

metadata:

labels:

name: ${jarName}

spec:

containers:

- name: ${jarName}

image: reg.test.local/library/${jarName}:${tag}

imagePullSecrets:

- name: registry-secret

转成的json结构

{

"apiVersion": "apps/v1",

"kind": "Deployment",

"metadata": {

"name": "${jarName}",

"labels": {

"name": "${jarName}"

}

},

"spec": {

"selector": {

"matchLabels": {

"name": "${jarName}"

}

},

"replicas": 1,

"template": {

"metadata": {

"labels": {

"name": "${jarName}"

}

},

"spec": {

"containers": [

{

"name": "${jarName}",

"image": "reg.test.local/library/${jarName}:${tag}"

}

],

"imagePullSecrets": [

{

"name": "registry-secret"

}

]

}

}

}

}

2.2、关键代码

# 传入service_name及image tag

def create_deploy_yaml(service_name, tag):

# 首先修改所有的${jarName}

deploy_data['metadata']['name'] = service_name

deploy_data['metadata']['labels']['name'] = service_name

deploy_data['spec']['selector']['matchLabels']['name'] = service_name

deploy_data['spec']['template']['metadata']['labels']['name'] = service_name

# 由于.spec.template.spec.containers的特殊性,我们采用直接修改的方式

# 首先拼接image字段

image = "reg.test.local/library/" + service_name + ":" + tag

# 创建new_containers字典列表

new_containers = [{'name': service_name, 'image': image}]

deploy_data['spec']['template']['spec']['containers'] = new_containers

3、完整脚本

#!/usr/bin/python

# encoding: utf-8

"""

The Script for Auto Create Deployment Yaml.

File: auto_create_deploy_yaml

User: miaocunfa

Create Date: 2020-06-10

Create Time: 17:06

"""

import os

from ruamel.yaml import YAML

yaml = YAML()

def create_service_yaml(service_name, ports):

service_mould_file = "mould/info-service-mould.yaml"

isServiceMould = os.path.isfile(service_mould_file)

if isServiceMould:

# read Service-mould yaml convert json

with open(service_mould_file, encoding='utf-8') as yaml_obj:

service_data = yaml.load(yaml_obj)

# Update jarName

service_data['metadata']['name'] = service_name

service_data['metadata']['labels']['name'] = service_name

service_data['spec']['selector']['name'] = service_name

# Update port

new_spec_ports = []

for port in ports:

port = int(port)

portname = 'port' + str(port)

new_port = {'name': portname, 'port': port, 'targetPort': port}

new_spec_ports.append(new_port)

service_data['spec']['ports'] = new_spec_ports

# json To service yaml

save_file = tag + '/' + service_name + '_svc.yaml'

with open(save_file, mode='w', encoding='utf-8') as yaml_obj:

yaml.dump(service_data, yaml_obj)

print(save_file + ": Success!")

else:

print("Service Mould File is Not Exist!")

def create_deploy_yaml(service_name, tag):

deploy_mould_file = "mould/info-deploy-mould.yaml"

isDeployMould = os.path.isfile(deploy_mould_file)

if isDeployMould:

with open(deploy_mould_file, encoding='utf-8') as yaml_obj:

deploy_data = yaml.load(yaml_obj)

# Update jarName

deploy_data['metadata']['name'] = service_name

deploy_data['metadata']['labels']['name'] = service_name

deploy_data['spec']['selector']['matchLabels']['name'] = service_name

deploy_data['spec']['template']['metadata']['labels']['name'] = service_name

# Update containers

image = "reg.test.local/library/" + service_name + ":" + tag

new_containers = [{'name': service_name, 'image': image}]

deploy_data['spec']['template']['spec']['containers'] = new_containers

# json To service yaml

save_file = tag + '/' + service_name + '_deploy.yaml'

with open(save_file, mode='w', encoding='utf-8') as yaml_obj:

yaml.dump(deploy_data, yaml_obj)

print(save_file + ": Success!")

else:

print("Deploy Mould File is Not Exist!")

services = {

'info-gateway': ['9999'],

'info-admin': ['7777'],

'info-config': ['8888'],

'info-message-service': ['8555', '9666'],

'info-auth-service': ['8666'],

'info-scheduler-service': ['8777'],

'info-uc-service': ['8800'],

'info-ad-service': ['8801'],

'info-community-service': ['8802'],

'info-groupon-service': ['8803'],

'info-hotel-service': ['8804'],

'info-nearby-service': ['8805'],

'info-news-service': ['8806'],

'info-store-service': ['8807'],

'info-payment-service': ['8808'],

'info-agent-service': ['8809'],

'info-consumer-service': ['8090'],

}

prompt = "\n请输入要生成的tag: "

answer = input(prompt)

print("")

if os.path.isdir(answer):

raise SystemExit(answer + ': is Already exists!')

else:

tag = answer

os.makedirs(tag)

for service_name, service_ports in services.items():

create_service_yaml(service_name, service_ports)

create_deploy_yaml(service_name, tag)

4、执行效果

python3 Auto_Create_K8S_YAML.py

请输入要生成的tag: 0.0.1

0.0.1/info-gateway_svc.yaml: Success!

0.0.1/info-gateway_deploy.yaml: Success!

0.0.1/info-admin_svc.yaml: Success!

0.0.1/info-admin_deploy.yaml: Success!

0.0.1/info-config_svc.yaml: Success!

0.0.1/info-config_deploy.yaml: Success!

0.0.1/info-message-service_svc.yaml: Success!

0.0.1/info-message-service_deploy.yaml: Success!

0.0.1/info-auth-service_svc.yaml: Success!

0.0.1/info-auth-service_deploy.yaml: Success!

0.0.1/info-scheduler-service_svc.yaml: Success!

0.0.1/info-scheduler-service_deploy.yaml: Success!

0.0.1/info-uc-service_svc.yaml: Success!

0.0.1/info-uc-service_deploy.yaml: Success!

0.0.1/info-ad-service_svc.yaml: Success!

0.0.1/info-ad-service_deploy.yaml: Success!

0.0.1/info-community-service_svc.yaml: Success!

0.0.1/info-community-service_deploy.yaml: Success!

0.0.1/info-groupon-service_svc.yaml: Success!

0.0.1/info-groupon-service_deploy.yaml: Success!

0.0.1/info-hotel-service_svc.yaml: Success!

0.0.1/info-hotel-service_deploy.yaml: Success!

0.0.1/info-nearby-service_svc.yaml: Success!

0.0.1/info-nearby-service_deploy.yaml: Success!

0.0.1/info-news-service_svc.yaml: Success!

0.0.1/info-news-service_deploy.yaml: Success!

0.0.1/info-store-service_svc.yaml: Success!

0.0.1/info-store-service_deploy.yaml: Success!

0.0.1/info-payment-service_svc.yaml: Success!

0.0.1/info-payment-service_deploy.yaml: Success!

0.0.1/info-agent-service_svc.yaml: Success!

0.0.1/info-agent-service_deploy.yaml: Success!

0.0.1/info-consumer-service_svc.yaml: Success!

0.0.1/info-consumer-service_deploy.yaml: Success!

➜ ll

total 12

drwxr-xr-x. 2 root root 4096 Jun 29 18:24 0.0.1

# 生成的 service yaml

➜ cat info-message-service_svc.yaml

apiVersion: v1

kind: Service

metadata:

name: info-message-service

labels:

name: info-message-service

version: v1

spec:

ports:

- name: port8555

port: 8555

targetPort: 8555

- name: port9666

port: 9666

targetPort: 9666

selector:

name: info-message-service

# 生成的 deployment yaml

➜ cat info-message-service_deploy.yaml

apiVersion: apps/v1

kind: Deployment

metadata:

name: info-message-service

labels:

name: info-message-service

spec:

selector:

matchLabels:

name: info-message-service

replicas: 2

template:

metadata:

labels:

name: info-message-service

spec:

containers:

- name: info-message-service

image: reg.test.local/library/info-message-service:0.0.1

imagePullSecrets:

- name: registry-secret

到此这篇关于使用python脚本自动生成K8S-YAML的方法示例的文章就介绍到这了,更多相关python自动生成K8S-YAML内容请搜索以前的文章或继续浏览下面的相关文章希望大家以后多多支持!

以上是 使用python脚本自动生成K8S-YAML的方法示例 的全部内容, 来源链接: utcz.com/z/335905.html

回到顶部