def test_get_cluster_simple(self, project_id, cluster_id, requests_mock): requests_mock.get(ANY, json={'foo': 'bar'}) client = PaaSCCClient(ComponentAuth('token')) resp = client.get_cluster(project_id, cluster_id) assert resp == {'foo': 'bar'} assert requests_mock.called
def test_get_node_list(self, project_id, cluster_id, requests_mock): requests_mock.get(ANY, json={"code": 0, "data": {"count": 1, "results": [{"inner_ip": "127.0.0.1"}]}}) client = PaaSCCClient(ComponentAuth("token")) resp = client.get_node_list(project_id, cluster_id) assert resp == {"count": 1, "results": [{"inner_ip": "127.0.0.1"}]} assert "desire_all_data" in requests_mock.last_request.qs
def test_list_all_projects(self, requests_mock): requests_mock.get( ANY, json={ 'code': 0, 'data': [ { 'description': 'asdfsdf', 'english_name': 'a1', 'project_id': 'e5e7c623798f41f18211c8b25895ef6f', 'project_name': 'asdfasdf', }, { 'description': 'public scan', 'english_name': 'CODE_716498', 'project_id': '47ceebac77214c77b794f7a6bb63ba14', 'project_name': 'CODE_716498', }, ], }, ) client = PaaSCCClient(ComponentAuth("token")) data = client.list_all_projects() assert len(data) == 2 assert "desire_all_data" in requests_mock.last_request.qs
def test_get_cluster_by_id(self, cluster_id, requests_mock): requests_mock.get(ANY, json={'code': 0, 'data': {'cluster_id': cluster_id}}) client = PaaSCCClient(ComponentAuth('token')) resp = client.get_cluster_by_id(cluster_id) assert resp == {'cluster_id': cluster_id} assert requests_mock.called
def test_delete_cluster(self, project_id, cluster_id, requests_mock): requests_mock.delete(ANY, json={"code": 0, "data": None}) client = PaaSCCClient(ComponentAuth('token')) resp = client.delete_cluster(project_id, cluster_id) assert resp is None assert requests_mock.called assert requests_mock.request_history[0].method == "DELETE"
def test_update_node_list(self, project_id, cluster_id, requests_mock): requests_mock.patch( ANY, json={ "code": 0, "data": [{ "inner_ip": "127.0.0.1", "cluster_id": cluster_id, "project_id": project_id, "status": "normal" }], }, ) client = PaaSCCClient(ComponentAuth('token')) resp = client.update_node_list( project_id, cluster_id, [UpdateNodesData(inner_ip="127.0.0.1", status="normal")]) assert resp == [{ "inner_ip": "127.0.0.1", "cluster_id": cluster_id, "project_id": project_id, "status": "normal" }] assert requests_mock.called assert requests_mock.request_history[0].method == "PATCH"
def test_update_cluster(self, project_id, cluster_id, requests_mock): requests_mock.put( ANY, json={"code": 0, "data": {"cluster_id": cluster_id, "project_id": project_id, "status": "normal"}} ) client = PaaSCCClient(ComponentAuth('token')) resp = client.update_cluster(project_id, cluster_id, {"status": "normal"}) assert resp == {"cluster_id": cluster_id, "project_id": project_id, "status": "normal"} assert requests_mock.called
def query_bcs_cc_nodes(ctx_cluster: CtxCluster) -> List: """查询bcs cc中的节点数据""" client = PaaSCCClient( ComponentAuth(access_token=ctx_cluster.context.auth.access_token)) node_data = client.get_node_list(ctx_cluster.project_id, ctx_cluster.id) return { node["inner_ip"]: node for node in (node_data.get("results") or []) if node["status"] not in [NodeStatus.Removed] }
def _list_namespaces(self, cluster_id: str, shared_cluster_ids: List[str]) -> List[Dict]: paas_cc = PaaSCCClient(auth=ComponentAuth(get_system_token())) if cluster_id not in shared_cluster_ids: cluster = paas_cc.get_cluster_by_id(cluster_id=cluster_id) ns_data = paas_cc.get_cluster_namespace_list(project_id=cluster['project_id'], cluster_id=cluster_id) return ns_data['results'] or [] # 共享集群获取 ns_data = paas_cc.list_namespaces_in_shared_cluster(cluster_id) return ns_data['results'] or []
def _get_project_id(self, access_token, project_id_or_code: str) -> str: cache_key = f'BK_DEVOPS_BCS:PROJECT_ID:{project_id_or_code}' project_id = region.get(cache_key, expiration_time=EXPIRATION_TIME) if not project_id: paas_cc = PaaSCCClient(auth=ComponentAuth(access_token)) project_data = paas_cc.get_project(project_id_or_code) project_id = project_data['project_id'] region.set(cache_key, project_id) return project_id
def get_ns_id(access_token: str, project_id: str, cluster_id: str, namespace: str) -> int: """获取命名空间ID""" client = PaaSCCClient(ComponentAuth(access_token)) data = client.get_cluster_namespace_list(project_id, cluster_id) # 匹配命名空间名称 for ns in data.get("results") or []: if ns["name"] == namespace: return ns["id"] raise ValidationError( _("集群:{}下没有查询到命名空间:{}").format(cluster_id, namespace))
def _get_namespace_id(self): """获取命名空间ID,用于权限相关验证""" client = PaaSCCClient( ComponentAuth( access_token=self.ctx_cluster.context.auth.access_token)) namespaces = client.get_cluster_namespace_list( project_id=self.ctx_cluster.project_id, cluster_id=self.ctx_cluster.id) for ns in namespaces["results"]: if self.namespace != ns["name"]: continue return ns["id"] return -1
def fetch_instance_info(self, filter_obj: FancyDict, **options) -> ListResult: """ 批量获取集群属性详情 :param filter_obj: 查询参数字典 :return: ListResult 类型的实例列表 """ cluster_ids = filter_obj.ids paas_cc = PaaSCCClient(auth=ComponentAuth(get_system_token())) cluster_list = paas_cc.list_clusters(cluster_ids) results = [{ 'id': cluster['cluster_id'], 'display_name': cluster['name'] } for cluster in cluster_list] return ListResult(results=results, count=len(results))
def convert_ip_used_data( access_token: str, project_id: str, cluster_id: str, ip_used_data: Dict[str, bool] ) -> Dict[str, bool]: """转换为ip信息 NOTE: 历史数据可能为{ip_id: 是否使用},需要把ip_id转换为ip :param access_token: access_token :param project_id: 项目ID :param cluster_id: 集群ID :param ip_used_data: IP信息,格式为{ip: True} :return: 返回IP信息 """ nodes = PaaSCCClient(auth=ComponentAuth(access_token)).get_node_list(project_id, cluster_id) node_id_ip = {info["id"]: info["inner_ip"] for info in nodes["results"] or []} # 通过ip id, 获取ip _ip_used_data = {} for ip, used in ip_used_data.items(): try: # 临时数据中IP为节点ID,需要转换为对应的IP ip_id = int(ip) if node_id_ip.get(ip_id): _ip_used_data[node_id_ip[ip_id]] = used except ValueError: _ip_used_data[ip] = used return _ip_used_data
def filter_namespaces(self, cluster_id: str): paas_cc = PaaSCCClient(auth=ComponentAuth(self.access_token)) ns_data = paas_cc.get_cluster_namespace_list( project_id=self.project_id, cluster_id=cluster_id) ns_list = ns_data['results'] or [] # 过滤掉 k8s 系统和 bcs 平台使用的命名空间 return [ns for ns in ns_list if ns['name'] not in K8S_PLAT_NAMESPACE]
def _get_enabled_project(self, access_token, project_id_or_code: str) -> Optional[FancyDict]: cache_key = f"BK_DEVOPS_BCS:ENABLED_BCS_PROJECT:{project_id_or_code}" project = region.get(cache_key, expiration_time=EXPIRATION_TIME) if project and isinstance(project, FancyDict): return project paas_cc = PaaSCCClient(auth=ComponentAuth(access_token)) project_data = paas_cc.get_project(project_id_or_code) project = FancyDict(**project_data) self._refine_project(project) # 用户绑定了项目, 并且选择了编排类型 if project.cc_app_id != 0 and project.kind in ClusterType: region.set(cache_key, project) return project return None
def _get_enabled_project(self, access_token, project_id_or_code: str) -> Optional[FancyDict]: cache_key = bcs_project_cache_key.format( project_id_or_code=project_id_or_code) project = region.get(cache_key, expiration_time=EXPIRATION_TIME) if project and isinstance(project, FancyDict): return project paas_cc = PaaSCCClient(auth=ComponentAuth(access_token)) project_data = paas_cc.get_project(project_id_or_code) project = FancyDict(**project_data) self._refine_project(project) # 项目绑定了业务,即开启容器服务 if project.cc_app_id != 0: region.set(cache_key, project) return project return None
def list_auth_projects(access_token: str, username: str = '') -> Dict: """获取用户有权限(project_view)的所有项目""" if not username: authorization = ssm.get_authorization_by_access_token(access_token) username = authorization['identity']['username'] perm_filter = ProjectFilter().make_view_perm_filter(username) if not perm_filter: return {'code': 0, 'data': []} # 如果是 any, 表示所有项目 if ProjectFilter.op_is_any(perm_filter): if settings.REGION != 'ce': # 非 ce 版本可能项目很多, PaaSCCClient 接口拉取超时, 降级处理不返回 logger.error(f'{username} project filter match any!') return {'code': 0, 'data': []} client = PaaSCCClient(auth=ComponentAuth(access_token)) projects = client.list_all_projects() else: project_id_list = perm_filter.get('value') if not project_id_list: return {'code': 0, 'data': []} client = PaaSCCClient(auth=ComponentAuth(access_token)) projects = client.list_projects_by_ids(project_id_list) for p in projects: p['project_code'] = p['english_name'] return {'code': 0, 'data': projects}
def _list_namespaces(self, cluster_id: str) -> List[Dict]: paas_cc = PaaSCCClient(auth=ComponentAuth(get_system_token())) cluster = paas_cc.get_cluster_by_id(cluster_id=cluster_id) ns_data = paas_cc.get_cluster_namespace_list( project_id=cluster['project_id'], cluster_id=cluster_id) return ns_data['results']