2018-08-16 12:36:27 +02:00
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
|
|
|
|
import sys
|
|
|
|
import base
|
|
|
|
import swagger_client
|
|
|
|
|
|
|
|
class Project(base.Base):
|
2018-10-31 06:13:31 +01:00
|
|
|
def create_project(self, name=None, metadata=None, **kwargs):
|
2018-09-19 11:49:43 +02:00
|
|
|
if name is None:
|
|
|
|
name = base._random_name("project")
|
2018-10-31 06:13:31 +01:00
|
|
|
if metadata is None:
|
|
|
|
metadata = {}
|
2018-08-16 12:36:27 +02:00
|
|
|
client = self._get_client(**kwargs)
|
2018-10-31 06:13:31 +01:00
|
|
|
_, status_code, header = client.projects_post_with_http_info(
|
2018-08-16 12:36:27 +02:00
|
|
|
swagger_client.ProjectReq(name, metadata))
|
2018-10-31 06:13:31 +01:00
|
|
|
base._assert_status_code(201, status_code)
|
|
|
|
project_id = base._get_id_from_header(header)
|
|
|
|
|
|
|
|
return name, project_id
|
|
|
|
|
|
|
|
def get_projects(self, params, **kwargs):
|
|
|
|
client = self._get_client(**kwargs)
|
|
|
|
data = []
|
|
|
|
data, status_code, _ = client.projects_get_with_http_info(**params)
|
|
|
|
base._assert_status_code(200, status_code)
|
|
|
|
return data
|
|
|
|
|
|
|
|
def check_project_name_exist(self, name=None, **kwargs):
|
|
|
|
client = self._get_client(**kwargs)
|
|
|
|
_, status_code, _ = client.projects_head_with_http_info(name)
|
|
|
|
|
|
|
|
return {
|
|
|
|
200: True,
|
|
|
|
404: False,
|
|
|
|
}.get(status_code,'error')
|
|
|
|
|
|
|
|
|
|
|
|
def get_project(self, project_id, **kwargs):
|
|
|
|
client = self._get_client(**kwargs)
|
|
|
|
data, status_code, _ = client.projects_project_id_get_with_http_info(project_id)
|
|
|
|
base._assert_status_code(200, status_code)
|
|
|
|
|
|
|
|
return data
|
|
|
|
|
|
|
|
def update_project(self, project_id, metadata, **kwargs):
|
|
|
|
client = self._get_client(**kwargs)
|
|
|
|
project = swagger_client.Project(project_id, None, None, None, None, None, None, None, None, None, None, metadata)
|
|
|
|
_, status_code, _ = client.projects_project_id_put_with_http_info(project_id, project)
|
|
|
|
base._assert_status_code(200, status_code)
|
|
|
|
|
|
|
|
def delete_project(self, project_id, **kwargs):
|
|
|
|
client = self._get_client(**kwargs)
|
|
|
|
_, status_code, _ = client.projects_project_id_delete_with_http_info(project_id)
|
|
|
|
|
|
|
|
base._assert_status_code(200, status_code)
|
|
|
|
|
|
|
|
def get_project_metadata_by_name(self, project_id, meta_name, **kwargs):
|
|
|
|
client = self._get_client(**kwargs)
|
|
|
|
ProjectMetadata = swagger_client.ProjectMetadata()
|
|
|
|
ProjectMetadata, status_code, _ = client.projects_project_id_metadatas_meta_name_get_with_http_info(project_id, meta_name)
|
|
|
|
|
|
|
|
base._assert_status_code(200, status_code)
|
|
|
|
return {
|
|
|
|
'public': ProjectMetadata.public,
|
|
|
|
'enable_content_trust': ProjectMetadata.enable_content_trust,
|
|
|
|
'prevent_vul': ProjectMetadata.prevent_vul,
|
|
|
|
'auto_scan': ProjectMetadata.auto_scan,
|
|
|
|
'severity': ProjectMetadata.severity,
|
|
|
|
}.get(meta_name,'error')
|
|
|
|
|
|
|
|
|
|
|
|
def get_project_members(self, project_id, **kwargs):
|
|
|
|
client = self._get_client(**kwargs)
|
|
|
|
data = []
|
|
|
|
data, status_code, _ = client.projects_project_id_members_get_with_http_info(project_id)
|
|
|
|
print "****************data for get_project_members:", data
|
|
|
|
base._assert_status_code(200, status_code)
|
|
|
|
return data
|
|
|
|
|
|
|
|
def add_project_members(self, project_id, user_id, member_role_id = None, **kwargs):
|
|
|
|
if member_role_id is None:
|
|
|
|
member_role_id = 1
|
|
|
|
_member_user = {"user_id": int(user_id)}
|
|
|
|
projectMember = swagger_client.ProjectMember(member_role_id, member_user = _member_user)
|
|
|
|
client = self._get_client(**kwargs)
|
|
|
|
data = []
|
|
|
|
data, status_code, _ = client.projects_project_id_members_post_with_http_info(project_id, project_member = projectMember)
|
|
|
|
base._assert_status_code(201, status_code)
|
|
|
|
return data
|
|
|
|
|