106 lines
4.4 KiB
Python
106 lines
4.4 KiB
Python
from django.views.generic import View
|
|
from django.db.models import F
|
|
from libs import json_response, JsonParser, Argument, human_datetime, human_time
|
|
from apps.deploy.models import DeployRequest
|
|
from apps.deploy.utils import deploy_dispatch
|
|
from apps.app.models import App
|
|
from apps.host.models import Host
|
|
from threading import Thread
|
|
import json
|
|
import uuid
|
|
|
|
|
|
class RequestView(View):
|
|
def get(self, request):
|
|
data = []
|
|
for item in DeployRequest.objects.annotate(
|
|
env_name=F('app__env__name'),
|
|
app_name=F('app__name'),
|
|
app_host_ids=F('app__host_ids'),
|
|
app_extend=F('app__extend'),
|
|
created_by_user=F('created_by__nickname')):
|
|
tmp = item.to_dict()
|
|
tmp['env_name'] = item.env_name
|
|
tmp['app_name'] = item.app_name
|
|
tmp['app_extend'] = item.app_extend
|
|
tmp['extra'] = json.loads(item.extra)
|
|
tmp['host_ids'] = json.loads(item.host_ids)
|
|
tmp['app_host_ids'] = json.loads(item.app_host_ids)
|
|
tmp['status_alias'] = item.get_status_display()
|
|
tmp['created_by_user'] = item.created_by_user
|
|
data.append(tmp)
|
|
return json_response(data)
|
|
|
|
def post(self, request):
|
|
form, error = JsonParser(
|
|
Argument('id', type=int, required=False),
|
|
Argument('app_id', type=int, help='缺少必要参数'),
|
|
Argument('name', help='请输申请标题'),
|
|
Argument('extra', type=list, help='缺少必要参数'),
|
|
Argument('host_ids', type=list, filter=lambda x: len(x), help='请选择要部署的主机'),
|
|
Argument('desc', required=False),
|
|
).parse(request.body)
|
|
if error is None:
|
|
app = App.objects.filter(pk=form.app_id).first()
|
|
if not app:
|
|
return json_response(error='未找到该应用')
|
|
form.status = '1' if app.is_audit else '2'
|
|
form.extra = json.dumps(form.extra)
|
|
form.host_ids = json.dumps(form.host_ids)
|
|
if form.id:
|
|
DeployRequest.objects.filter(pk=form.id).update(
|
|
created_by=request.user,
|
|
reason=None,
|
|
**form
|
|
)
|
|
else:
|
|
DeployRequest.objects.create(created_by=request.user, **form)
|
|
return json_response(error=error)
|
|
|
|
|
|
class RequestDetailView(View):
|
|
def get(self, request, r_id):
|
|
req = DeployRequest.objects.filter(pk=r_id).first()
|
|
if not req:
|
|
return json_response(error='为找到指定发布申请')
|
|
return json_response({
|
|
'app_name': req.app.name,
|
|
'env_name': req.app.env.name,
|
|
'status': req.status,
|
|
'status_alias': req.get_status_display()
|
|
})
|
|
|
|
def post(self, request, r_id):
|
|
req = DeployRequest.objects.filter(pk=r_id).first()
|
|
if not req:
|
|
return json_response(error='未找到指定发布申请')
|
|
if req.status != '2':
|
|
return json_response(error='该申请单当前状态还不能执行发布')
|
|
hosts = Host.objects.filter(id__in=json.loads(req.host_ids))
|
|
token = uuid.uuid4().hex
|
|
Thread(target=deploy_dispatch, args=(request, req, token)).start()
|
|
outputs = {str(x.id): {'data': ''} for x in hosts}
|
|
outputs.update(local={'data': f'{human_time()} 建立接连... '})
|
|
targets = [{'id': x.id, 'title': f'{x.name}({x.hostname}:{x.port})'} for x in hosts]
|
|
return json_response({'token': token, 'outputs': outputs, 'targets': targets})
|
|
|
|
def patch(self, request, r_id):
|
|
form, error = JsonParser(
|
|
Argument('reason', required=False),
|
|
Argument('is_pass', type=bool, help='参数错误')
|
|
).parse(request.body)
|
|
if error is None:
|
|
req = DeployRequest.objects.filter(pk=r_id).first()
|
|
if not req:
|
|
return json_response(error='未找到指定申请')
|
|
if not form.is_pass and not form.reason:
|
|
return json_response(error='请输入驳回原因')
|
|
if req.status != '1':
|
|
return json_response(error='该申请当前状态不允许审核')
|
|
req.approve_at = human_datetime()
|
|
req.approve_by = request.user
|
|
req.status = '2' if form.is_pass else '-1'
|
|
req.reason = form.reason
|
|
req.save()
|
|
return json_response(error=error)
|