def on_post(self, req, resp):
params = req.params
ip = params.get('ip', '')
site_id = params.get('site_id', 0)
pool_id = params.get('pool_id', 0)
description = params.get('description', '')
clock = int(time.time())
kind_id = 1
level_id = params.get('level_id', 4) # 4是应用层
source_id = params.get('source_id', 0)
if ip:
server_module = create('server')
pool_id = server_module.search_poolid_by_ip(ip)
site_id = server_module.search_siteid_by_poolid(pool_id)
try:
description = json.loads(description)
event_module = create('event')
res = event_module.insert_event(ip, site_id, pool_id, description, clock, kind_id, level_id, source_id)
if res.id:
resp.body = super(AlertApi, self).format_out(1)
else:
resp.body = super(AlertApi, self).format_out(0)
resp.status = falcon.HTTP_OK
except Exception as e:
self.error_msg = "%s" % e
raise falcon.HTTPBadRequest("error", self.error_msg, code=0)
def process_resource(self, req, resp, resource):
self._api_module = create('api')
if hasattr(resource, 'no_auth') and resource.no_auth:
pass
else:
self._token_module = create('token')
try:
tmp = req.access_route
if isinstance(tmp, list):
if len(tmp) > 0:
self._from_ip = tmp[0]
except Exception as e:
print e
self._path = req.path
date = int(time.time())
token = req.get_param('token', default='')
token_result = self._token_module.get_token(token)
if token_result:
token_result = token_result['data']
if not token_result:
msg = "token is not found"
self._api_module.insert_api_log(self._from_ip, self._path, 0,
date, msg)
raise falcon.HTTPBadRequest("error", msg, code=0)
else:
self._user_id = token_result[0].get('id', 0)
def process_resource(self, req, resp, resource):
self._api_module = create('api')
if hasattr(resource, 'no_auth') and resource.no_auth:
pass
else:
self._token_module = create('token')
try:
tmp = req.access_route
if isinstance(tmp, list):
if len(tmp) > 0:
self._from_ip = tmp[0]
except Exception as e:
print e
self._path = req.path
date = int(time.time())
token = req.get_param('token', default='')
token_result = self._token_module.get_token(token)
if token_result:
token_result = token_result['data']
if not token_result:
msg = "token is not found"
self._api_module.insert_api_log(self._from_ip, self._path, 0, date, msg)
raise falcon.HTTPBadRequest("error", msg, code=0)
else:
self._user_id = token_result[0].get('id', 0)
def on_post(self, req, resp):
params = req.params
pool_id = params.get('pool_id', 0)
ip = params.get('ip', '')
key = params.get('key', '')
source_id = params.get('source_id', 0)
start_time = params.get('start_time', 0)
duration = params.get('duration', 0)
end_time = 0
if start_time:
if not fb.is_valid_date(start_time):
self.error_msg = "valid_date failed. PLS use this format:2015-12-23 23:11:01"
raise falcon.HTTPBadRequest("error", self.error_msg, code=0)
else:
start_time = int(time.mktime(time.strptime(start_time, '%Y-%m-%d %H:%M:%S')))
else:
c = int(time.time())
x = time.localtime(c)
a = time.strftime('%Y-%m-%d %H:%M:00', x)
start_time = int(time.mktime(time.strptime(a, '%Y-%m-%d %H:%M:%S')))
end_time = start_time + (duration if duration else 3600)
shield_module = create('shield')
res = shield_module.insert_shield(pool_id, ip, key, source_id, start_time, end_time)
if res.id:
resp.body = super(ShieldApi, self).format_out(1)
else:
resp.body = super(ShieldApi, self).format_out(0)
resp.status = falcon.HTTP_OK
