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,modjob_args
import json
from units.config_log import *
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]
def get(self,stype):
job_list = list(consul_kv.get_kv_dict(f'ConsulManager/jobs').values())
if stype == 'cloud':
cloud_dict = {}
for i in 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': 10, 'rds_interval': 20, 'redis_interval': 20}
isextip = False
for i in 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']
isextip = i["args"][-1] if len(i["args"]) == 3 else False
elif f'{vendor}/{account}/rds/{region}' == i['id']:
restype.append('rds')
interval['rds_interval'] = i['minutes']
elif f'{vendor}/{account}/redis/{region}' == i['id']:
restype.append('redis')
interval['redis_interval'] = i['minutes']
return {'code': 20000, 'restype': restype, 'interval': interval, 'isextip': isextip}
def post(self,stype):
job_list = list(consul_kv.get_kv_dict(f'ConsulManager/jobs').values())
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']
isextip = editjob_dict['isextip']
proj_interval = int(editjob_dict['proj_interval'])
ecs_interval = int(editjob_dict['ecs_interval'])
rds_interval = int(editjob_dict['rds_interval'])
redis_interval = int(editjob_dict['redis_interval'])
logger.info(f'{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 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}'
redis_jobid = f'{vendor}/{account}/redis/{region}'
if 'ecs' in restype:
isecs = [x for x in job_list if x['id'] == f'{vendor}/{account}/ecs/{region}']
if len(isecs) == 1:
if ecs_interval != isecs[0]['minutes']:
isecs[0]['minutes'] = ecs_interval
consul_kv.put_kv(f'ConsulManager/jobs/{ecs_jobid}',isecs[0])
modjob_interval(ecs_jobid,ecs_interval)
if len(isecs[0]['args']) != 3 or isextip != isecs[0]['args'][2]:
try:
isecs[0]['args'][2] = isextip
except:
isecs[0]['args'].append(isextip)
consul_kv.put_kv(f'ConsulManager/jobs/{ecs_jobid}',isecs[0])
modjob_args(ecs_jobid,isecs[0]['args'])
else:
job_func = f"__main__:{vendor}.ecs"
job_args = [account,region,isextip]
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 job_list if x['id'] == f'{vendor}/{account}/rds/{region}']
if len(isrds) == 1:
if rds_interval != isrds[0]['minutes']:
isrds[0]['minutes'] = rds_interval
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
if 'redis' in restype:
isredis = [x for x in job_list if x['id'] == f'{vendor}/{account}/redis/{region}']
if len(isredis) == 1:
if redis_interval != isredis[0]['minutes']:
isredis[0]['minutes'] = redis_interval
consul_kv.put_kv(f'ConsulManager/jobs/{redis_jobid}',isredis[0])
modjob_interval(redis_jobid,redis_interval)
else:
job_func = f"__main__:{vendor}.redis"
job_args = [account,region]
job_interval = redis_interval
addjob(redis_jobid, job_func, job_args, job_interval)
job_dict = {'id':redis_jobid,'func':job_func,'args':job_args,'minutes':job_interval,
"trigger": "interval","replace_existing": True}
consul_kv.put_kv(f'ConsulManager/jobs/{redis_jobid}',job_dict)
else:
try:
consul_kv.del_key(f'ConsulManager/jobs/{redis_jobid}')
deljob(redis_jobid)
except:
pass
return {'code': 20000, 'data': f'{vendor}/{account}/{region}：编辑成功！'}
api.add_resource(Edit, '/api/edit/<stype>')