def check_users():
userid = request.args.get('userid')
print(userid)
if r.get(userid) == None:
return ("1")
else:
username = r.get(userid)
return ("2")
def get_is_data():
userid = request.args.get('userID')
eq_id = request.args.get('equipID')
user_name = r.get(userid)
wu = wx_user.query.filter(wx_user.name == user_name).first() #获取用户名
wxuser_id = wu.id #获取用户ID
sql = 'select image_id from information where user_id=%s and equip_id=%s' % (
wxuser_id, eq_id)
cursor.execute(sql)
image_id = cursor.fetchall()
all_id = []
unread_id = []
for (i, ) in image_id:
unread_id.append(i)
imgs_id = srch_hash_data('upload_img', 'equip_id', eq_id)
for i in imgs_id:
all_id.append(i)
# 将所有的图片id减去未读图片的id获得已读图片的ID
is_id = set(all_id) - set(unread_id)
read_arr = []
for i in is_id:
read_data = srch_hash_data('upload_img', 'id', str(i))
read_arr.append(read_data[0][0])
read_arr.append(read_data[0][1])
read_arr.append(read_data[0][2])
json_data = json.dumps(read_arr, cls=CJsonEncoder)
return json_data
def getimformation():
Eqid = request.values.get('eqid')
database.ping(reconnect=True)
counter_lock2 = threading.Lock()
counter_lock2.acquire()
exist = r.exists("eqimfo" + Eqid)
if exist == 0:
sql = 'select * from upload_img where equip_id=%s order by date desc' % Eqid
try:
cursor.execute(sql)
except:
database.ping(reconnect=True)
cursor.execute(sql)
data = cursor.fetchall()
message = []
y = []
for i in data:
wxuser = wx_user.query.filter(wx_user.id == i[1]).first()
name = wxuser.name
y.append(i[0])
y.append(name)
y.append(i[2])
y.append(i[3])
y.append(i[5])
y.append(i[6])
message.append(y)
y = []
json_data = json.dumps(message, cls=CJsonEncoder)
counter_lock2.release()
r.set("eqimfo" + Eqid, json_data)
return json_data
else:
return r.get("eqimfo" + Eqid)
def get_deserve():
exist = r.exists('eq_data')
if exist == 0:
sql = 'select *from equipment'
cursor.execute(sql)
user_data = cursor.fetchall()
json_data = json.dumps(user_data, cls=CJsonEncoder)
r.set('eq_data', json_data)
else:
json_data = r.get('eq_data')
return json_data
def change_emails():
userid = request.form.get('userID')
newmail = request.form.get('newMail')
mailcode = request.form.get('mailcode') #用户输入的验证码
usermail = request.form.get('userMail') #用户接收验证码的邮箱
codeKey = "" + usermail + ":code"
code = r.get(codeKey)
if mailcode == '%s' % code:
if len(r.keys('wx_user*')) != 0:
for k in r.scan_iter('wx_user*'):
r_username = r.hget(k, 'name')
if r.get(userid) == r_username:
user_infor = {'email': newmail}
r.hset(k, user_infor)
sql = 'update wx_user set email="' + newmail + '" where name="' + r.get(
userid) + '"'
cursor.execute(sql)
database.commit()
return ("0")
else:
return ("1")
def del_unread_data():
userid = request.args.get('userID')
infor_id = request.args.get('inforID')
read_flag = request.args.get('read_flag')
if read_flag == 'unread_id':
user_name = r.get(userid)
wu = wx_user.query.filter(wx_user.name == user_name).first() #获取用户名
wxuser_id = wu.id #获取用户ID
sql = 'delete from information where image_id=%s and user_id=%s' % (
infor_id, wxuser_id)
cursor.execute(sql)
database.commit()
data = srch_hash_data('imgs', 'id', infor_id)
return data
def get_equip_data():
userid = request.args.get('userID')
user_name = r.get(userid)
wu = wx_user.query.filter(wx_user.name == user_name).first() #获取用户名
wxuser_id = wu.id #获取用户ID
equip_unread = []
data = r.sort('equip:id', by='equipment:*->id', get=['equipment:*->id'])
for equip_id in data:
sql = 'select count(*) from information where user_id=%s and equip_id=%s' % (
wxuser_id, equip_id)
cursor.execute(sql)
one = cursor.fetchall()
for (i, ) in one:
equip_unread.append(i)
equip_unread_data = json.dumps(equip_unread)
return equip_unread_data
def web_register(user_name):
username = request.form.get("username")
pwd = request.form.get('pwd')
repwd = request.form.get('repwd')
userphone = request.form.get("userphone")
usermail = request.form.get("usermail")
mailcode = request.form.get('mailcode')
codeKey = "" + usermail + ":code"
code = r.get(codeKey)
# 检验参数完整
if not all([username, pwd, repwd, userphone, usermail, mailcode]):
# return jsonify({'status': '失败'}, {'erroMsg': '参数不完整'})
return ("3")
else:
user = wx_user.query.filter(wx_user.name == username).first()
#找到
if user:
return ("2")
else:
# 判断邮箱验证码是否正确
if mailcode == '%s' % code:
for k in r.scan_iter('wx_user*'):
r.delete(k)
webuser = wx_user(name=username,
email=usermail,
phone=userphone,
password=pwd,
wx_id=1,
info_number=0)
db.session.add(webuser)
db.session.commit()
for k in r.scan_iter('wx_user*'):
r.delete(k)
return ("1")
else:
return ("0")
def getuserval():
userid = request.args.get('userID')
userData = srch_hash_data('wx_user', 'name', r.get(userid))
return userData
