乐趣区

关于python:k8s在线和离线批量修改镜像地址的两种方式

背景介绍

有时往 k8s 集群里部署一堆服务的时候,须要拷贝一堆 yaml 文件,当然还有其余形式部署,例如通过建设一个一个的流水线等形式,然而这太慢了,尽管是一劳永逸,然而如果说仅仅是部署一次那就可太吃力了,那么间接拷贝一堆 yaml 文件就简略了许多,然而如果之前的环境中镜像地址是走的内网,而当初放到另一个环境中,网络不通时,就须要更改其中的镜像地址了

如果之前的网络环境到阿里云内网镜像地址是通的,也就是长这样:registry-vpc.cn-beijing.aliyuncs.com/xxxx/xxxxx

而后当初要部署到华为云上时,间接启动 yaml 文件必定是起不来的,须要批改下此镜像地址,改为:registry.cn-beijing.aliyuncs.com/xxxx/xxxx,改成这样就能够拉了。

那么该如何实现呢?

调用 k8s 接口实现镜像替换

上面这段代码是通过 k8s 的接口,先去查以后命名空间下都有哪些服务,而后把这些服务都写道一个数组里,而后再去迭代这个数组,而后去批改整个命名空间下的所有 pod 的镜像地址

import re
from kubernetes import client, config


class deployServer:
    def __init__(self, kubeconfig):
        self.kubeconfig = kubeconfig
        config.kube_config.load_kube_config(config_file=self.kubeconfig)
        self._AppsV1Api = client.AppsV1Api()
        # 这俩临时用不到
        # self._CoreV1Api = client.CoreV1Api()
        # self._ExtensionsV1beta1Api = client.ExtensionsV1beta1Api()

    def list_deploy(self, namespace_name):
        '''
        :param namespace_name: 指定命名空间
        :return: 返回一个 deployment 的列表
        '''
        deploySubj = self._AppsV1Api.list_namespaced_deployment(namespace_name)
        return [subj.metadata.name for subj in deploySubj.items]

    def patchMultiDeploy(self, namespace_name, newimgdomain):
        for i in self.list_deploy(namespace_name):
            self.patchDeploy(namespace_name, i, newimgdomain)
            print("{}: {} patch success.".format(namespace_name, i))

    def patchDeploy(self, deploy_namespace, deploy_name, newimgdomain, keyword='vpc'):
        img_addr_rule = r"[^?]{1,}\.[^?]{1,}\.com(\/.*\:.*)"
        img_domain_rule = r"([^?]{1,}\.[^?]{1,}\.com).*\:.*"
        old_deploy = self._AppsV1Api.read_namespaced_deployment(
            name=deploy_name,
            namespace=deploy_namespace,
        )
        old_deploy_container = old_deploy.spec.template.spec.containers
        # 迭代 deployment 的列表,获取 deployment 的名称以及它所在索引地位
        for i, k in enumerate(old_deploy_container):
            # 捕捉处域名之外的镜像地址
            oldImgAddr = re.findall(img_addr_rule, old_deploy_container[i].image)[0]
            # 捕捉镜像地址的域名,也能够不获取,然而获取会更精确些
            oldImgDoamin = re.findall(img_domain_rule, old_deploy_container[i].image)[0]
            # 符合条件的才会进行替换
            if keyword in oldImgDoamin:
                old_deploy_container[i].image = newimgdomain + oldImgAddr
                self._AppsV1Api.patch_namespaced_deployment(
                    name=deploy_name,
                    namespace=deploy_namespace,
                    body=old_deploy
                )
        return


if __name__ == '__main__':
    kubeconfig = r'kubeconfig/config'
    k8s = deployServer(kubeconfig)
    # 指定服务所在命名空间
    namespace = 'ops-logging'
    # 指定要替换的镜像的域名
    img = 'docker.elastic.co'
    k8s.patchMultiDeploy(namespace, img)

留神这里须要指定 kubeconfig 文件

批改文件里的镜像地址

下面那个是间接调接口,通过 patch 来进行批改,上面是间接批改 yaml 文件

import os
import re

def modifyImagesFromFile(originpath, src_img, dst_img):
    _re_rule = r".*image: ([^?]{1,}\.[^?]{1,}\.com).*\:.*"
    if os.path.isdir(originpath):
        filesubj = os.walk(originpath)
        filename = []
        # 迭代指定门路下的所有文件
        for path, _, file in filesubj:
            # 把所有的文件拼成一个残缺的门路,并写道一个列表里
            filename = [os.path.join(path, f) for f in file]
        if filename:
            for f in filename:
                with open(f, 'r+', encoding='utf-8') as fr:
                    fr_data = fr.read()
                    replace_list = re.findall(_re_rule, fr_data)
                    if replace_list:
                        for pattern in replace_list:
                            if src_img in pattern:
                                # 把匹配到的内容进行替换
                                fr_data = fr_data.replace(pattern, dst_img)
                        fr.seek(0)
                        fr.write(fr_data)
                        print("{} replace success".format(f))
                    else:
                        print("Unknown")
    else:
        print("{} not a directory".format(originpath))


if __name__ == '__main__':
    originpath = r"E:\ 我的项目 \DEV 环境 \sg-saas-pro-hbali\gldsg-gvs\deployments"
    dst_img = 'registry.cn-beijing.aliyuncs.com'
    src_img = 'registry-vpc.cn-beijing.aliyuncs.com'
    modifyImagesFromFile(originpath)

留神:须要把所有的 deployment 文件都放到一个目录里

总结一哈

如果是在线批改的话,就用下面调 k8s 接口通过 patch 来批改,间接粘贴复制,指定好命名空间,指定好 kubeconfig 的文件地位即可
如果是把所有的 yaml 文件都 down 下来了,那么就间接迭代目录下的所有文件,通过正则去捕捉镜像地址,而后进行替换,再写入源文件就能够了。

本文由博客群发一文多发等经营工具平台 OpenWrite 公布

退出移动版