def add():
if not session.get('log_in'):
return redirect(url_for('login'))
if request.method == 'GET':
return render_template('add.html')
way = request.form['way']
conn = F.db_get_conn()
cursor = conn.cursor()
if way == '0':
ip = request.form['ip']
sql = 'insert into xpp_list (type,ip1) values (%s,"%s")'%(way,ip)
n = cursor.execute(sql)
if n:
flash(u'添加成功！')
conn.commit()
cursor.close()
conn.close()
return render_template('add.html')
elif way == '1':
start = request.form['start']
end = request.form['end']
sql = 'insert into xpp_list (type,ip1,ip2) values (%s,"%s","%s")'%(way,start,end)
n = cursor.execute(sql)
fid = cursor.lastrowid
if n:
for ip in F.gen_ip(start,end):
sql = 'insert into xpp_list_son (fid,ip) values (%s,"%s")'%(fid,ip)
cursor.execute(sql)
conn.commit()
cursor.close()
conn.close()
flash(u'添加成功！')
return render_template('add.html')
def list_son(id):
if not session.get('log_in'):
return redirect(url_for('login'))
sql = 'select ip1,type from xpp_list where id = %s'%id
conn = F.db_get_conn()
cursor = conn.cursor()
cursor.execute(sql)
rez = cursor.fetchone()
if rez[1] == 0:
sql = 'select ip,status,create_time from xpp_log where ip = "%s" order by create_time desc limit 50'%rez[0]
cursor.execute(sql)
data = cursor.fetchall()
data2 = []
for i in data:
t = []
t.append(i[0])
t.append(i[1])
t.append(time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(i[2])))
data2.append(t)
return render_template('list_son.html',data=data2)
elif rez[1] == 1:
sql = 'select ip from xpp_list_son where fid = %s'%id
n = cursor.execute(sql)
if n:
data = cursor.fetchall()
data2 = []
for i in data:
t = []
t.append(i[0])
t.append(F.ip_get_status(i[0]))
t.append(u'now')
data2.append(t)
return render_template('list_son.html',data=data2)
def list():
if not session.get('log_in'):
return redirect(url_for('login'))
conn = F.db_get_conn()
cursor = conn.cursor()
sql = 'select id,ip1,ip2,type from xpp_list'
cursor.execute(sql)
data = cursor.fetchall()
return render_template('list.html',data=data)
def detect_ip(data):
conn = F.db_get_conn()
cursor = conn.cursor()
for ip in data:
rez = F.ping_host(ip)
if rez:
sql = 'insert into xpp_log (ip,status,create_time) values ("%s",1,%s)'%(ip,str(time.time())[:10])
else:
sql = 'insert into xpp_log (ip,status,create_time) values ("%s",0,%s)'%(ip,str(time.time())[:10])
cursor.execute(sql)
conn.commit()
cursor.close()
conn.close()
def list_del(id):
if not session.get('log_in'):
return redirect(url_for('login'))
sql = 'delete from xpp_list where id = %s'%id
conn = F.db_get_conn()
cursor = conn.cursor()
n = cursor.execute(sql)
if n:
flash(u'删除成功！')
else:
flash(u'删除失败！')
conn.commit()
cursor.close()
conn.close()
return redirect(url_for('list'))
def get_all_ip(): host_list = [] sql = 'select ip1 from xpp_list where type = 0' conn = F.db_get_conn() cursor = conn.cursor() n = cursor.execute(sql) if n: host_list1 = cursor.fetchall() for i in host_list1: host_list.append(i[0]) sql = 'select ip from xpp_list_son' n = cursor.execute(sql) if n: host_list2 = cursor.fetchall() for i in host_list2: host_list.append(i[0]) cursor.close() conn.close() return host_list
def login():
if request.method == 'POST':
user_name = request.form['username']
pass_word = request.form['password']
conn=F.db_get_conn()
cursor=conn.cursor()
sql = 'select password from xpp_user where username = "******"'%user_name
n = cursor.execute(sql)
if n:
rows = cursor.fetchone()
cursor.close()
conn.close()
if pass_word == rows[0]:
session['log_in'] = True
return redirect(url_for('index'))
else:
flash(u'密码错误！')
else:
flash(u'用户不存在！')
cursor.close()
conn.close()
return render_template('login.html')
else:
return render_template('login.html')
