async def create_suite(
project_id: int = Body(..., embed=True),
suite_name: str = Body(..., embed=True),
token_user: TokenUser = Depends(auth_token)
) -> BaseRes:
_, error = verify_project_filed(project_id)
if error:
return error
_, error = verify_project_member(token_user.user_id, project_id)
if error:
return error
suite = AtpProjectApiSuite(name=suite_name, project_id=project_id, creator=token_user.user_id,
create_time=datetime.datetime.now())
try:
Db.insert(suite)
return BaseRes()
except Exception as e:
logger.error(traceback.format_exc())
return BaseRes(status=0, error=str(e))
async def create_project_api_case(
project_api_case: ProjectApiCase,
token_user: TokenUser = Depends(auth_token)
) -> BaseRes:
_, error = verify_project_filed(project_api_case.project_id)
if error:
return error
_, error = verify_project_member(token_user.user_id, project_api_case.project_id)
if error:
return error
try:
request_headers: str = json.dumps(project_api_case.request_headers)
request_query: str = json.dumps(project_api_case.request_query)
request_body: str = project_api_case.request_body
except Exception as e:
logger.error(traceback.format_exc())
return BaseRes(status=0, error='json转换异常 : ' + str(e))
atp_project_api_case: AtpProjectApiCase = AtpProjectApiCase(
name=project_api_case.name,
method=project_api_case.request_method,
project_id=project_api_case.project_id,
suite_id=project_api_case.suite_id,
request_url=project_api_case.request_url,
is_use_env=project_api_case.request_host.is_user_env,
env_host=project_api_case.request_host.env_host,
request_headers=request_headers,
request_query=request_query,
request_body=request_body,
creator=token_user.user_id,
create_time=datetime.datetime.now()
)
try:
Db.insert(atp_project_api_case)
return BaseRes()
except Exception as e:
logger.error(traceback.format_exc())
return BaseRes(status=0, error=str(e))
async def create_module(
request: Module, token_user: TokenUser = Depends(auth_token)) -> BaseRes:
_, error = verify_project_filed(request.project_id)
if error:
return error
_, error = verify_project_member(token_user.user_id, request.project_id)
if error:
return error
try:
if request.parent_id != 0:
parent_module: AtpProjectModule = Db.select_by_primary_key(
AtpProjectModule, request.parent_id)
if not parent_module:
return BaseRes(status=0, error='父模块不存在')
Db.insert(
AtpProjectModule(name=request.name,
parent_id=request.parent_id,
project_id=request.project_id,
creator=token_user.user_id,
create_time=datetime.datetime.now()))
return BaseRes()
except Exception as e:
logger.error(e)
return BaseRes(status=0, error=str(e))
async def create_project_env(
env_info: ProjectEnv,
token_user: TokenUser = Depends(auth_token)
) -> BaseRes:
_, error = verify_project_filed(env_info.project_id)
if error:
return error
_, error = verify_project_member(token_user.user_id, env_info.project_id)
if error:
return error
try:
Db.insert(
AtpProjectEnv(
name=env_info.env_name,
host=env_info.env_host,
project_id=env_info.project_id,
creator=token_user.user_id,
create_time=datetime.datetime.now()
)
)
return BaseRes()
except Exception as e:
logger.error(traceback.format_exc())
return BaseRes(status=0, error=str(e))
