def get_sig_count(sig_priority):
try:
db = sql.DataBase()
except pymysql.err.OperationalError as e:
return jsonify({'code': 50027, 'error': 'db error'})
count = db.get_event_count_by_sig(sig_priority)
return jsonify({'code': 20000, 'sig': sig_priority, 'count': count})
def get_sig_count_by_time(time, sig_priority):
db = sql.DataBase()
counts = db.get_event_count_by_time_sig(time, sig_priority)
return jsonify({
'code': 20000,
'sig': sig_priority,
'time': time,
'counts': counts
})
def get_all_count_by_time(time):
db = sql.DataBase()
count1 = db.get_event_count_by_time_sig(time, 1)
count2 = db.get_event_count_by_time_sig(time, 2)
count3 = db.get_event_count_by_time_sig(time, 3)
result = [[] for i in range(4)]
for i in range(len(count1)):
result[0].append(count1[i][0])
result[1].append(count2[i][0])
result[2].append(count3[i][0])
print(count3[i][1:])
result[3].append(".".join(str(d) for d in count3[i][1:]))
return jsonify({'code': 20000, 'time': time, 'counts': result})
def update_loc():
"""
更新入侵事件位置信息
"""
mongo = MongoClient()
db = mongo.ids
db.drop_collection('location')
mongo_loc = db.location
db = sql.DataBase()
ip_count = db.get_event_loc()
for ip, count in ip_count:
print(ip)
try:
info = get_loc(ip)
except Exception:
continue
if info is None:
continue
values = list(info[:2])
values.append(count)
mongo_loc.insert({
'name': info[2],
'value': values
})
def get_system_score():
db = sql.DataBase()
score = db.get_score()
return jsonify({'code': 20000, 'score': score})
def get_event_count_by_num(num):
db = sql.DataBase()
counts = db.get_event_count_top(num)
return jsonify({'code': 20000, 'counts': counts})