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.
98 lines
5.5 KiB
98 lines
5.5 KiB
from flask import Blueprint
|
|
from flask_restful import reqparse, Resource, Api
|
|
from flask_apscheduler import APScheduler
|
|
from units import token_auth,consul_kv
|
|
from config import vendors
|
|
import json
|
|
from .jobs import deljob,addjob,runjob
|
|
blueprint = Blueprint('exp',__name__)
|
|
api = Api(blueprint)
|
|
|
|
parser = reqparse.RequestParser()
|
|
parser.add_argument('query_dict',type=str)
|
|
parser.add_argument('exp_config_dict',type=dict)
|
|
parser.add_argument('isnotify_dict',type=dict)
|
|
|
|
class Exp(Resource):
|
|
decorators = [token_auth.auth.login_required]
|
|
def get(self,stype):
|
|
if stype == 'list':
|
|
switch = consul_kv.get_value(f'ConsulManager/exp/config').get('switch',False)
|
|
if not switch:
|
|
return({'code': 20000,'exp_list':[],'vendor_list':[],'account_list':[],'amount_list':[]})
|
|
args = parser.parse_args()
|
|
query_dict = json.loads(args['query_dict'])
|
|
if query_dict['vendor'] != '':
|
|
query_dict['vendor'] = {v : k for k, v in vendors.items()}[query_dict['vendor']]
|
|
query_set = set({k:v for k,v in query_dict.items() if v != ''}.items())
|
|
cloud_job_list = consul_kv.get_keys_list('ConsulManager/jobs')
|
|
cloud_list = [i for i in cloud_job_list if i.endswith('/group')]
|
|
exp_list = []
|
|
amount_list = []
|
|
for i in cloud_list:
|
|
vendor,account = i.split('/')[2:4]
|
|
cloud_info_dict = {'vendor':vendor,'account':account}
|
|
if query_set.issubset(cloud_info_dict.items()):
|
|
pass
|
|
else:
|
|
continue
|
|
amount = consul_kv.get_value(f'ConsulManager/exp/lists/{vendor}/{account}/amount')['amount']
|
|
amount_list.append({'vendor':vendors[vendor],'account':account,'amount':amount})
|
|
exp_dict = consul_kv.get_value(f'ConsulManager/exp/lists/{vendor}/{account}/exp')
|
|
for k,v in exp_dict.items():
|
|
isnotify = consul_kv.get_value(f"ConsulManager/exp/isnotify/{vendor}/{account}/{v['notify_id']}").get('isnotify',True)
|
|
exp_list.append({'vendor':vendors[vendor],'account':account,'id':k,'Region':v['Region'],
|
|
'Product':v['Product'],'Name':v.get('Name','Null'),'EndTime':v['EndTime'],
|
|
'Ptype':v['Ptype'].replace('hws.resource.type.',''),
|
|
'notify_id': v['notify_id'],'isnotify':isnotify})
|
|
vendor_list = sorted(list(set([i['vendor'] for i in exp_list])))
|
|
account_list = sorted(list(set([i['account'] for i in exp_list])))
|
|
return {'code': 20000,'exp_list':exp_list,'vendor_list':vendor_list,'account_list':account_list,'amount_list':amount_list}
|
|
if stype == 'config':
|
|
exp_config = consul_kv.get_value('ConsulManager/exp/config')
|
|
return {'code': 20000, 'exp_config': exp_config}
|
|
def post(self,stype):
|
|
if stype == 'isnotify':
|
|
args = parser.parse_args()
|
|
isnotify_dict = args['isnotify_dict']
|
|
vendor = {v : k for k, v in vendors.items()}[isnotify_dict['vendor']]
|
|
account = isnotify_dict['account']
|
|
notify_id = isnotify_dict['notify_id']
|
|
isnotify = isnotify_dict['isnotify']
|
|
if not isnotify:
|
|
consul_kv.put_kv(f'ConsulManager/exp/isnotify/{vendor}/{account}/{notify_id}',{'isnotify':isnotify})
|
|
return {'code': 20000, 'data': '此条资源告警关闭！', 'type': 'warning'}
|
|
else:
|
|
consul_kv.del_key(f'ConsulManager/exp/isnotify/{vendor}/{account}/{notify_id}')
|
|
return {'code': 20000, 'data': '此条资源告警开启！', 'type': 'success'}
|
|
if stype == 'config':
|
|
args = parser.parse_args()
|
|
exp_config_dict = args['exp_config_dict']
|
|
consul_kv.put_kv('ConsulManager/exp/config',exp_config_dict)
|
|
cloud_job_list = consul_kv.get_keys_list('ConsulManager/jobs')
|
|
cloud_list = [i for i in cloud_job_list if i.endswith('/group')]
|
|
collect_days = exp_config_dict['collect_days']
|
|
notify_days = exp_config_dict['notify_days']
|
|
notify_amount = exp_config_dict['notify_amount']
|
|
if exp_config_dict['switch']:
|
|
for i in cloud_list:
|
|
vendor,account = i.split('/')[2:4]
|
|
exp_job_id = f'{vendor}/{account}/exp'
|
|
exp_job_func = f'__main__:{vendor}.exp'
|
|
exp_job_args = [account,collect_days,notify_days,notify_amount]
|
|
exp_job_interval = 60
|
|
addjob(exp_job_id,exp_job_func,exp_job_args,exp_job_interval)
|
|
exp_job_dict = {'id':exp_job_id,'func':exp_job_func,'args':exp_job_args,'minutes':exp_job_interval,'trigger': 'interval','replace_existing': True}
|
|
consul_kv.put_kv(f'ConsulManager/exp/jobs/{vendor}/{account}',exp_job_dict)
|
|
runjob(exp_job_id)
|
|
return {'code': 20000, 'data': '到期日与余额采集功能开启！'}
|
|
else:
|
|
for i in cloud_list:
|
|
vendor,account = i.split('/')[2:4]
|
|
exp_job_id = f'{vendor}/{account}/exp'
|
|
deljob(exp_job_id)
|
|
consul_kv.del_key_all('ConsulManager/exp/jobs/')
|
|
consul_kv.del_key_all('ConsulManager/exp/lists/')
|
|
return {'code': 20000, 'data': '到期日与余额采集功能关闭！'}
|
|
api.add_resource(Exp, '/api/exp/<stype>')
|