StarsL.cn
2 years ago
parent
033d2bb80e
commit
c39bee1e21
Binary file not shown.
@ -0,0 +1,115 @@
|
||||
from flask import Blueprint
|
||||
from flask_restful import reqparse, Resource, Api
|
||||
from flask_apscheduler import APScheduler
|
||||
from config import vendors,regions
|
||||
from units import token_auth,consul_kv
|
||||
from .jobs import deljob,addjob,runjob,modjob_interval
|
||||
import json
|
||||
blueprint = Blueprint('edit_cloud',__name__)
|
||||
api = Api(blueprint)
|
||||
|
||||
parser = reqparse.RequestParser()
|
||||
parser.add_argument('vendor',type=str)
|
||||
parser.add_argument('account',type=str)
|
||||
parser.add_argument('region',type=str)
|
||||
parser.add_argument('editJob',type=dict)
|
||||
class Edit(Resource):
|
||||
decorators = [token_auth.auth.login_required]
|
||||
job_list = list(consul_kv.get_kv_dict(f'ConsulManager/jobs').values())
|
||||
def get(self,stype):
|
||||
if stype == 'cloud':
|
||||
cloud_dict = {}
|
||||
for i in self.job_list:
|
||||
vendor,account = i['id'].split('/')[0:2]
|
||||
if vendor in cloud_dict:
|
||||
if account not in cloud_dict[vendor]:
|
||||
cloud_dict[vendor].append(account)
|
||||
else:
|
||||
cloud_dict[vendor] = [account]
|
||||
return {'code': 20000,'cloud_dict': cloud_dict}
|
||||
if stype == 'find':
|
||||
args = parser.parse_args()
|
||||
vendor = args['vendor']
|
||||
account = args['account']
|
||||
region = args['region']
|
||||
restype = ['group']
|
||||
interval = {'proj_interval': 60, 'ecs_interval': 5, 'rds_interval': 5}
|
||||
for i in self.job_list:
|
||||
if f'{vendor}/{account}/group' == i['id']:
|
||||
interval['proj_interval'] = i['minutes']
|
||||
elif f'{vendor}/{account}/ecs/{region}' == i['id']:
|
||||
restype.append('ecs')
|
||||
interval['ecs_interval'] = i['minutes']
|
||||
elif f'{vendor}/{account}/rds/{region}' == i['id']:
|
||||
restype.append('rds')
|
||||
interval['rds_interval'] = i['minutes']
|
||||
return {'code': 20000, 'restype': restype, 'interval': interval}
|
||||
def post(self,stype):
|
||||
if stype == 'commit':
|
||||
args = parser.parse_args()
|
||||
editjob_dict = args['editJob']
|
||||
vendor = editjob_dict['vendor']
|
||||
account = editjob_dict['account']
|
||||
region = editjob_dict['region']
|
||||
restype = editjob_dict['restype']
|
||||
proj_interval = int(editjob_dict['proj_interval'])
|
||||
ecs_interval = int(editjob_dict['ecs_interval'])
|
||||
rds_interval = int(editjob_dict['rds_interval'])
|
||||
print(editjob_dict)
|
||||
if editjob_dict['akskswitch']:
|
||||
ak = editjob_dict['ak']
|
||||
sk = editjob_dict['sk']
|
||||
consul_kv.put_aksk(editjob_dict['vendor'],editjob_dict['account'],ak,sk)
|
||||
|
||||
jobgroup = [x for x in self.job_list if x['id'] == f'{vendor}/{account}/group'][0]
|
||||
if proj_interval != jobgroup['minutes']:
|
||||
jobgroup['minutes'] = proj_interval
|
||||
jobid = f'{vendor}/{account}/group'
|
||||
consul_kv.put_kv(f'ConsulManager/jobs/{jobid}',jobgroup)
|
||||
modjob_interval(jobid,proj_interval)
|
||||
|
||||
ecs_jobid = f'{vendor}/{account}/ecs/{region}'
|
||||
rds_jobid = f'{vendor}/{account}/rds/{region}'
|
||||
if 'ecs' in restype:
|
||||
isecs = [x for x in self.job_list if x['id'] == f'{vendor}/{account}/ecs/{region}']
|
||||
if len(isecs) == 1:
|
||||
if ecs_interval != isecs[0]['minutes']:
|
||||
consul_kv.put_kv(f'ConsulManager/jobs/{ecs_jobid}',isecs[0])
|
||||
modjob_interval(ecs_jobid,ecs_interval)
|
||||
else:
|
||||
job_func = f"__main__:{vendor}.ecs"
|
||||
job_args = [account,region]
|
||||
job_interval = ecs_interval
|
||||
addjob(ecs_jobid, job_func, job_args, job_interval)
|
||||
job_dict = {'id':ecs_jobid,'func':job_func,'args':job_args,'minutes':job_interval,
|
||||
"trigger": "interval","replace_existing": True}
|
||||
consul_kv.put_kv(f'ConsulManager/jobs/{ecs_jobid}',job_dict)
|
||||
else:
|
||||
try:
|
||||
consul_kv.del_key(f'ConsulManager/jobs/{ecs_jobid}')
|
||||
deljob(ecs_jobid)
|
||||
except:
|
||||
pass
|
||||
|
||||
if 'rds' in restype:
|
||||
isrds = [x for x in self.job_list if x['id'] == f'{vendor}/{account}/rds/{region}']
|
||||
if len(isrds) == 1:
|
||||
if rds_interval != isrds[0]['minutes']:
|
||||
consul_kv.put_kv(f'ConsulManager/jobs/{rds_jobid}',isrds[0])
|
||||
modjob_interval(rds_jobid,rds_interval)
|
||||
else:
|
||||
job_func = f"__main__:{vendor}.rds"
|
||||
job_args = [account,region]
|
||||
job_interval = rds_interval
|
||||
addjob(rds_jobid, job_func, job_args, job_interval)
|
||||
job_dict = {'id':rds_jobid,'func':job_func,'args':job_args,'minutes':job_interval,
|
||||
"trigger": "interval","replace_existing": True}
|
||||
consul_kv.put_kv(f'ConsulManager/jobs/{rds_jobid}',job_dict)
|
||||
else:
|
||||
try:
|
||||
consul_kv.del_key(f'ConsulManager/jobs/{rds_jobid}')
|
||||
deljob(rds_jobid)
|
||||
except:
|
||||
pass
|
||||
return {'code': 20000, 'data': f'{vendor}/{account}/{region}：编辑成功！'}
|
||||
api.add_resource(Edit, '/api/edit/<stype>')
|
||||
@ -0,0 +1,22 @@
|
||||
import request from '@/utils/request-ops'
|
||||
|
||||
export function getCloud() {
|
||||
return request({
|
||||
url: '/api/edit/cloud',
|
||||
method: 'get'
|
||||
})
|
||||
}
|
||||
export function findGroup(vendor, account, region) {
|
||||
return request({
|
||||
url: '/api/edit/find',
|
||||
method: 'get',
|
||||
params: { vendor, account, region }
|
||||
})
|
||||
}
|
||||
export function PostEditJob(editJob) {
|
||||
return request({
|
||||
url: '/api/edit/commit',
|
||||
method: 'post',
|
||||
data: { editJob }
|
||||
})
|
||||
}
|
||||
Binary file not shown.
Binary file not shown.
Loading…
Reference in new issue