pythonCli_原始_禁止删除

import json
import builtins
import time
import subprocess
import site
from sys import argv
from argparse import ArgumentParser

## 安装 cmdb-sdk-python
#shell_script = """
#if [ -z "$( pip3 list | grep cmdb-sdk-python )" ]; then
#    (
#        rm -rf /tmp/cmdb-sdk-python && \
#        git clone git@atta-gitlab.abc.cn:atta-team/dev/cmdb-sdk-python.git /tmp/cmdb-sdk-python && \
#        cd /tmp/cmdb-sdk-python && \
#        pip3 install . && \
#        rm -rf /tmp/cmdb-sdk-python
#    ) > /dev/null 2> /tmp/cmdb-sdk-python_install.error
#    if [ $? -ne 0 ]; then
#        echo -e "[ERROR]: [cmdb-sdk-python] 安装失败!!!\n$( cat /tmp/cmdb-sdk-python_install.error )" 1>&2
#        rm -rf /tmp/cmdb-sdk-python_install.error
#        exit 1
#    fi
#fi
#"""
#result = subprocess.run(shell_script, shell=True, text=True, capture_output=True)
## 检查脚本是否成功执行
#if result.returncode != 0:
#    print(result.stderr)
#    exit(1)

from bkcmdb.core.auth import AccessKeyCredential
from bkcmdb.core.client import BkClient

class ParserObj(object):

    def __init__(self):
        self.parser = ArgumentParser()
        self.add_arguments()

    def add_arguments(self):
        self.parser.add_argument('--domain', '-d', dest='cmdb',  default='http://cmdbapi.abc.cn', help=u'CMDB API域名')
        self.parser.add_argument('--token',  '-t', dest='token', default='dev123',              help=u'CMDB API密码')

class Models():

    def __init__(self):
        parser_cls = ParserObj()
        self.parser = parser_cls.parser  # 合并参数
        self._add_method_arguments()
        self.options = options = parser_cls.parser.parse_args(argv[1:])
        cred = AccessKeyCredential("access_key_id", options.token)
        self.bkclient = BkClient(cred, options.cmdb)

    def _add_method_arguments(self):
        """
        添加参数: 
        1. 调用所有的方法
        2. 设定所选方法的参数
        """
        methods = [method for method in dir(self) if callable(getattr(self, method)) and not method.startswith(("__","_"))]
        self.parser.add_argument('method', choices=methods, help=u'选择方法执行')
        self.parser.add_argument('--modal',    '-m',     dest='modal',          default='',   help=u'CMDB模型名称')
        self.parser.add_argument('--instname', '-i',     dest='instName',       default='',   help=u'指定所选模型的实例名(多个以,分割)')
        self.parser.add_argument('--param',    '-p',     dest='param',          default='',   help=u'指定所选方法的参数')
        self.parser.add_argument('--model_props',        dest='modelProps',     default='',   help=u'指定所选模型的属性')
        self.parser.add_argument('--asst_modal',         dest='asstModal',      default='',   help=u'指定关联模型名称')
        self.parser.add_argument('--asst_model_props',   dest='asstModelProps', default='',   help=u'指定所选关联模型的属性')
        self.parser.add_argument('--return_model_props', dest='rtnModelProps',  default='',   help=u'指定所选模型需返回的属性')

    def _string_split(self, param):
        # 默认拆分字符串规则: ","
        self.input_string = param
        self.input_string_split = self.input_string.split(',')
        return self.input_string_split

    def _get_inst_name(self, param):
        input_string_split = self._string_split(args.instName)

        # 获取服务列表信息
        ins = self.bkclient.Instance(args.modal)
        ins.condition = {"bk_inst_name": {"$in": input_string_split}}
        service_info = ins.get()

        # 判断当前服务数: 输入服务数量，匹配到的服务数量
        input_count = len(input_string_split)
        match_count = service_info.get('count')

        if input_count > match_count:
            #match_servname = service_info['info'].values
            match_servname = [item['bk_inst_name'] for item in service_info['info']]
            unmatch_servname = [ value for value in input_string_split if value not in match_servname ]
            raise ValueError(f"[ERROR]: 请先将以下[ {input_count - match_count} ]个服务: [ {unmatch_servname} ] 录入CMDB模型[ {args.modal} ], "
                             f"请提供组件名及仓库地址并联系运维值班人【技术部群中每天会发送】添加服务！！！")

        return ' '.join(item['bk_inst_name'] for item in service_info['info'])

    def __get_inst_info(self, param):
        """
        python内部调用，返回多个实例信息
        """
        # 判断服务是否存在
        is_exit = self._get_inst_name(args.instName)
        if not is_exit:
            exit()

        # 获取服务列表信息
        ins = self.bkclient.Instance(args.modal)
        ins.condition = {"bk_inst_name": {"$in": self._string_split(args.instName)}}
        service_info = ins.get()

        # 返回实例列表
        return json.dumps(service_info['info']).encode('utf-8').decode('unicode-escape')


    def _get_inst_info(self, param):
        """
        返回实例的信息，json格式
        """
        # 判断服务是否存在
        is_exit = self._get_inst_name(args.instName)
        if not is_exit:
            exit()

        # 只能输入一个实例名
        if len(self._string_split(args.instName)) > 1:
            raise ValueError("[ERROR]: 只能输入一个实例名!!!")

        # 获取服务列表信息
        ins = self.bkclient.Instance(args.modal)
        ins.condition = {"bk_inst_name": {"$in": self._string_split(args.instName)}}
        service_info = ins.get()

        # 返回中文字符
        return json.dumps(service_info['info'][0]).encode('utf-8').decode('unicode-escape')

    def _get_service_k8s_info(self, param):
        # 解析获取的服务k8s配置信息
        input_string_split = self._string_split(args.instName)
        if len(input_string_split) < 4:
            raise ValueError(f"[ERROR]: 请以[,]分隔写入对应服务的k8s资源信息, 格式如下: appname,cluster_tag,env,nsname. 如:python,sitaiserver,dev,aisit1")

        appname     = input_string_split[0]
        cluster_tag = input_string_split[1]
        env         = input_string_split[2]
        nsname      = input_string_split[3]
        #print(f'{appname}|{cluster_tag}|{env}|{nsname}')

        # 获取服务列表信息
        ins = self.bkclient.Instance(args.modal)
        ins.condition =  {
            "bk_inst_name": {"$eq": appname},
            "cluster_tag": { "$eq": cluster_tag},
            "env": {         "$eq": env},
            "namespace": {   "$regex": nsname}
        }
        service_info = ins.get()

        if not service_info['info']:
            ins.condition =  {
                "bk_inst_name": {"$eq": appname},
                "cluster_tag": { "$eq": cluster_tag},
                "env": {         "$eq": env}
            }
            service_info = ins.get()

        return json.dumps(service_info['info'][0])

    def _get_inst_property_value(self, param):
        property_value = json.loads(param).get(args.modelProps)

        return property_value
    
    def get_inst_name(self, param):
        # 特殊模型处理: xt_service_k8s
        if args.modal == 'xt_service_k8s':
            self.get_service_k8s_name(args.instName)
            exit()
        input_string_split = self._string_split(args.instName)

        # 获取服务列表信息
        ins = self.bkclient.Instance(args.modal)
        ins.condition = {"bk_inst_name": {"$in": input_string_split}}
        service_info = ins.get()

        # 判断当前服务数: 输入服务数量，匹配到的服务数量
        input_count = len(input_string_split)
        match_count = service_info.get('count')

        if input_count > match_count:
            #match_servname = service_info['info'].values
            match_servname = [item['bk_inst_name'] for item in service_info['info']]
            unmatch_servname = [ value for value in input_string_split if value not in match_servname ]
            raise ValueError(f"[ERROR]: 请先将以下[ {input_count - match_count} ]个服务: [ {unmatch_servname} ] 录入CMDB[ {args.modal} ], "
                             f"请提供组件名及仓库地址并联系运维值班人【技术部群中每天会发送】添加服务！！！")
            exit(1)
         
        print(*[item['bk_inst_name'] for item in service_info['info']], sep=',')

    def get_service_k8s_name(self, param):
        input_string_split = self._string_split(args.instName)

        # 获取服务列表信息
        ins = self.bkclient.Instance(args.modal)
        ins.condition = {"bk_inst_name": {"$in": input_string_split}}
        service_info = ins.get()

        print(*[item['bk_inst_name'] for item in service_info['info']], sep=',')

    def get_inst_info(self, param):
        # 特殊模型处理: xt_service_k8s
        if args.modal == 'xt_service_k8s':
            self.get_service_k8s_info(args.instName)
            exit()

        # 只能输入一个实例名
        if len(self._string_split(args.instName)) > 1:
            raise ValueError("[ERROR]: 只能输入一个实例名!!!")

        # 判断服务是否存在
        is_exit = self._get_inst_name(args.instName)
        if not is_exit:
            exit()

        # 获取服务列表信息
        ins = self.bkclient.Instance(args.modal)
        #ins.condition = {"bk_inst_name": {"$in": self._string_split(args.instName)}}
        ins.condition = {"bk_inst_name": {"$eq": args.instName}}
        service_info = ins.get()

        # 返回中文字符
        print(json.dumps(service_info['info'][0]).encode('utf-8').decode('unicode-escape'))
    
    def get_service_k8s_info(self, param):
        # 解析获取的服务k8s配置信息
        input_string_split = self._string_split(args.instName)
        if len(input_string_split) < 4:
            raise ValueError(f"[ERROR]: 请以[,]分隔写入对应服务的k8s资源信息, 格式如下: appname,cluster_tag,env,nsname. 如:python,sitaiserver,dev,aisit1")

        appname     = input_string_split[0]
        cluster_tag = input_string_split[1]
        env         = input_string_split[2]
        nsname      = input_string_split[3]
        #print(f'{appname}|{cluster_tag}|{env}|{nsname}')

        # 获取服务列表信息
        ins = self.bkclient.Instance(args.modal)
        ins.condition =  {
            "bk_inst_name": {"$eq": appname},
            "cluster_tag": { "$eq": cluster_tag},
            "env": {         "$eq": env},
            "namespace": {   "$eq": nsname}
        }
        service_info = ins.get()

        # 当指定{nsname} 不存在时，返回下面的参数对应的值
        if not service_info['info']:
            ins.condition =  {
                "bk_inst_name": {"$eq": appname},
                "cluster_tag": { "$eq": cluster_tag},
                "env": {         "$eq": env}
            }
            service_info = ins.get()

        print(json.dumps(service_info['info'][0]))

    def get_inst_property_value(self, param):
        #for json_str in json.loads(param):
        #    property_value = json_str.get(args.modelProps)
        #    print(property_value)
        print(param)
        property_value = json.loads(param).get(args.modelProps)
        print(property_value)

    def get_inst_property_single(self, param):
        service_info = self._get_inst_info(args.instName)
        property_value = json.loads(service_info).get(args.modelProps)

        print(property_value)

    def get_service_k8s_property_single(self, param):
        ## 解析获取的服务k8s配置信息
        service_info = self._get_service_k8s_info(args.instName)
        property_value = json.loads(service_info).get(args.modelProps)

        print(property_value)

    def get_inst_property_keys(self, param):
        service_info = self._get_inst_info(args.instName)
        property_key = ' '.join(json.loads(service_info).keys())

        print(property_key)

    def get_service_k8s_property_keys(self, param):
        service_info = self._get_service_k8s_info(args.instName)
        property_key = ' '.join(json.loads(service_info).keys())

        print(property_key)

    def get_inst_model(self, param):
        # 获取服务列表信息
        #ins = self.bkclient.Instance(param)
        ins = self.bkclient.Instance(args.modal)
        #for item in ins.objectattr:
        #    print(item)
        #model_property = ins.objectattr()

        # 获取列表里面的key
        property_key = '\n'.join(item['bk_property_id'] for item in ins.objectattr if 'bk_property_id' in item)
        print(property_key)

    def query_property_single(self, param):
        """
        通过单一字段查询获取服务信息
        """
        # 解析匹配内容
        input_string_split = self._string_split(param)
        if len(input_string_split) < 2:
            raise ValueError(f"[ERROR]: 请以[,]分隔写入对应匹配内容, 格式如下: rule,match, rule:[ in|regex|not ]  如:eq,python")

        match_rule  = input_string_split[0]
        match_value = input_string_split[1:]

        #input_string_split = self._string_split(param)
        #property_key   = input_string_split[0]
        #fuzzy_value    = input_string_split[1]
        #query_property = input_string_split[2]
        #print(f"{property_key}|{fuzzy_value}|{query_property}")

        property_key   = args.modelProps
        query_property = args.rtnModelProps

        ins = self.bkclient.Instance(args.modal)
        ins.limit = 9999
        if match_rule == "regex" or match_rule == "eq":
            # regex、eq值: 字符串
            match_value = ','.join([item for item in match_value])
            ins.condition = {property_key: {f"${match_rule}": match_value}}
        elif match_rule == "in":
            # in 值: 列表
            ins.condition = {property_key: {f"${match_rule}": match_value}}
        elif match_rule == "not":
            # not值: 比in多加一层条件
            ins.condition = {property_key: {f"${match_rule}": {"$in": match_value}}}
        else:
            raise ValueError(f"[ERROR]: 匹配规则:[{match_rule}]未找到, 请以正确输入, 格式如下: rule,match, rule:[ in|regex|not ]  如:eq,python")

        service_info = ins.get()

        if not service_info['info']:
            raise ValueError(f"[ERROR]: 通过模型[{args.modal}]的列属性[{property_key}], "
                             f"{match_rule}:[ {match_value} ], 查找列属性[{query_property}], 返回结果为空, "
                             f"即CMDB中未找到匹配服务的信息, 请检查服务名或者联系运维同学！！！")

        print(*[item[query_property] for item in service_info['info']], sep='\n')

    #def accurate_query_property_single(self, param):
    #    """
    #    通过单一字段精确查询获取服务信息
    #    """
    #    ## 解析获取的服务k8s配置信息
    #    #input_string_split = self._string_split(param)
    #    #if len(input_string_split) < 3:
    #    #    print(f"[ERROR]: 请以[,]分隔写入对应服务的k8s资源信息, 格式如下: appname,cluster_tag,env,nsname. 如:")
    #    #    exit(1)
    #    property_key   = args.modelProps
    #    accurate_value = param
    #    query_property = args.rtnModelProps
    #    #print(f"{property_key}|{accurate_value}|{query_property}")

    #    ins = self.bkclient.Instance(args.modal)
    #    ins.condition = {property_key: {"$eq": accurate_value}}
    #    service_info = ins.get()

    #    if not service_info['info']:
    #        raise ValueError(f"[ERROR]: 通过模型[{args.modal}]的列属性[{property_key}], "
    #                         f"精确匹配:[{accurate_value}], 查找列属性[{query_property}], 返回结果为空, "
    #                         f"即CMDB中未找到匹配服务的信息，请检查服务名或者联系运维同学！！！")

    #    # CMDB中有bk_inst_name唯一性校验
    #    #if len(service_info['info']) > 1 and query_property == "bk_inst_name":
    #    #    print(f"[ERROR] 在CMDB可能存在同名[{service_info}]服务，请检查服务名是否正确或联系运维同学！！！")

    #    print(*[item[query_property] for item in service_info['info']], sep='\n')

    def get_inst_asst(self, param):
        """
        获取实例的关联关系
        """
        service_info = self._get_inst_info(args.instName)
        args.modelProps = "bk_inst_id"
        service_info_id = self._get_inst_property_value(service_info)

        ins = self.bkclient.InstanceObj(args.modal, service_info_id)
        service_info = ins.get_asst_object(args.asstModal)
        print(json.dumps(service_info).encode('utf-8').decode('unicode-escape'))

    def get_inst_asst_property_single(self, param):
        """
        获取实例的关联模型中 指定值的指定属性值
        : 通过单个或多个实例值, 获取对应的关联模型的值, 多 -> 一
        """

        # 判断实例是否存在
        is_exit = self._get_inst_name(args.instName)
        if not is_exit:
            exit()

        # 获取实例列表信息
        ins = self.bkclient.Instance(args.modal)
        ins.condition = {"bk_inst_name": {"$in": self._string_split(args.instName)}}
        service_info = ins.get()

        # 获取实例的ID
        args.modelProps = "bk_inst_id"
        service_info_id = [item[args.modelProps] for item in service_info['info']]

        # 获取输入模型对应的关联关系id
        ## 判断是否是第一次执行: 为了位运算asst_inst_info_tmp
        times = 0
        for item in service_info_id:
            ins = self.bkclient.InstanceObj(args.modal, item)
            service_info = ins.get_passive_asst(args.asstModal)
            asst_inst_info = [asst_item['bk_inst_id'] for asst_item in service_info]
            if times == 0:
                asst_inst_info_tmp = asst_inst_info
            asst_inst_info_tmp = set(asst_inst_info) & set(asst_inst_info_tmp)
            times += 1

        # 获取关联实例的信息
        ins = self.bkclient.Instance(args.asstModal)
        ins.condition = {"bk_inst_id": {"$in": list(asst_inst_info_tmp)}}
        service_info = ins.get()
        property_key = ' '.join([item[args.asstModelProps] for item in service_info['info']])

        print(property_key)


if __name__ == "__main__":
    #install_cmdb_jdk()
    runModels = Models()
    args = runModels.parser.parse_args()

    method_to_call = getattr(runModels, args.method, None)
    
    ## 调用方法
    start_time = time.time()
    if method_to_call and callable(method_to_call):
        method_to_call(args.param)
    else:
        print("指定的方法无效.")
    end_time = time.time()
    print(f"耗时: {end_time - start_time:.3f}s")