add gitlab-ci部署k8s集群

script/k8s-deploy.py

@@ -0,0 +1,139 @@
+#!/usr/bin/env python3
+
+import sys
+import os
+import getopt
+import yaml
+import pykube
+
+def usage():
+    print("Usage: %s [ -b | --build-arg | -f | --file ] product-line project-name docker_images_version dockerfile_name --no-cache" % sys.argv[0], "\n", "or",
+    "Usage: %s [ -b | --build-arg | -f | --file ] product-line project-name docker_images_version dockerfile_name" % sys.argv[0], "\n\n",
+    "-b, --build-arg 声明Dockerfile中环境变量，参数可以指定多次，示例 A=b, ", "\n",
+    "-f, --file 指定Helm values.yaml文件，参数只能指定一次，指定多个也只会取第一个值")
+
+def check_item_exists(project, server_name, docker_images_version, *args):
+    api = pykube.HTTPClient(pykube.KubeConfig.from_file("k8s config"))
+    deploy = pykube.Deployment.objects(api).filter(namespace=project)
+    os.environ['project'] = str(project)
+    os.environ['server_name'] = str(server_name)
+    os.environ['docker_images_version'] = str(docker_images_version)
+    os.environ['s_name'] = str(server_name)
+    if args:
+        os.environ['helm_values_file'] = str(args[0])
+
+    if os.system('helm status $server_name 1> /dev/null 2>&1') == 0:
+        os.system("echo '\033[1;32;40m' 'Helm滚动发布、超时为5分钟、失败在本次基础上自动回滚上一个版本。' '\033[0m \n'")
+        os.system('helm upgrade --timeout 300 --atomic --install $server_name --namespace $project \
+        --set serverName=$server_name --set image.project=$project \
+        --set serverFeatureNameReplace=$server_name \
+        --set image.tag=$docker_images_version "Helm warehouse address" $helm_values_file')
+        return
+    else:
+        # 兼容过去没有使用Helm部署
+        for deployment in deploy:
+            if server_name == str(deployment):
+                # deployment进行滚动升级
+                os.system('kubectl set image deployment/$server_name \
+                 $server_name=harbor.example.com/$project/$server_name:$docker_images_version \
+                 --namespace=$project')
+                return
+
+    # 部署新项目
+    # 使用helm模板部署服务
+    os.system('helm install --name $server_name --namespace $project \
+    --set serverName=$server_name --set image.project=$project \
+    --set serverFeatureNameReplace=$server_name \
+    --set image.tag=$docker_images_version "Helm warehouse address" $helm_values_file')
+    return
+
+def main():
+    build_arg_list = []
+    file_name_list = []
+    try:
+        opts, args = getopt.getopt(sys.argv[1:], "hb:f:", ["help", "build-arg=", "file="])
+        # check opts param
+        for opt, arg in opts:
+            if opt in ("-h", "--help"):
+                usage()
+                sys.exit(1)
+            elif opt in ("-b", "--build-arg"):
+                build_arg_list.append(arg)
+            elif opt in ("-f", "--file"):
+                file_name_list.append(arg)
+            else:
+                pass
+
+        if build_arg_list:
+            build_arg = '--build-arg ' + ' --build-arg '.join(build_arg_list)
+            os.environ['build_arg'] = str(build_arg)
+
+        if file_name_list:
+            try:
+                with open(file_name_list[0], "r", encoding='utf-8') as f:
+                    results = f.read()
+                    results_dict = yaml.load(results, Loader=yaml.FullLoader)
+                    if results_dict["replicaCount"] > 30 or results_dict["replicaCount"] <= 0:
+                        print("\033[31;31m" "replicaCount要求范围: 大于0小于30个Pod，请重新调整replicaCount参数!" "\033[0m \n")
+                        sys.exit(1)
+                    helm_values_file = '-f ' + ' -f '.join(file_name_list)
+            # 没有传入replicaCount参数，就走Helm默认replicaCount配置参数
+            except KeyError:
+                helm_values_file = '-f ' + ' -f '.join(file_name_list)
+            except FileNotFoundError:
+                print("\033[31;31m" "-f 或者 --file 指定的文件不存在!" "\033[0m \n")
+                sys.exit(1)
+
+        # check args param
+        if len(args) == 4:
+            os.environ['project'] = str(args[0])
+            os.environ['server_name'] = str(args[1])
+            os.environ['docker_images_version'] = str(args[2])
+            os.environ['dockerfile_name'] = str(args[3])
+            if os.system('docker build -t harbor.example.com/$project/$server_name:$docker_images_version . \
+            -f $dockerfile_name $build_arg && \
+            docker login -u admin -p ****** harbor.example.com && \
+            docker push harbor.example.com/$project/$server_name:$docker_images_version') != 0:
+                print("\033[31;31m" "Docker Build 失败!" "\033[0m \n")
+                sys.exit(1)
+
+            # 调用方法部署k8s
+            if file_name_list:
+                check_item_exists(args[0], args[1], args[2], helm_values_file)
+            else:
+                check_item_exists(args[0], args[1], args[2])
+
+        elif len(args) == 5:
+            # 判断第五个参数是不是 --no-cache
+            if args[4] == "--no-cache":
+                os.environ['project'] = str(args[0])
+                os.environ['server_name'] = str(args[1])
+                os.environ['docker_images_version'] = str(args[2])
+                os.environ['dockerfile_name'] = str(args[3])
+                if os.system('docker build --no-cache -t harbor.example.com/$project/$server_name:$docker_images_version . \
+                -f $dockerfile_name $build_arg && \
+                docker login -u admin -p ****** harbor.example.com && \
+                docker push \
+                harbor.example.com/$project/$server_name:$docker_images_version') != 0:
+                    print("\033[31;31m" "Docker Build 失败!" "\033[0m \n")
+                    sys.exit(1)
+
+                # 调用方法部署k8s
+                if file_name_list:
+                    check_item_exists(args[0], args[1], args[2], helm_values_file)
+                else:
+                    check_item_exists(args[0], args[1], args[2])
+            else:
+                print("\033[31;31m" "第五个参数不是 --no-cache !" "\033[0m \n")
+                sys.exit(1)
+        else:
+            usage()
+            sys.exit(1)
+
+    except getopt.GetoptError:
+        print("\033[31;31m" "输入参数错误!" "\033[0m \n")
+        usage()
+        sys.exit(1)
+
+if __name__ == '__main__':
+    main()