You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
116 lines
5.4 KiB
116 lines
5.4 KiB
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>')
|