def register():
"""
用户注册
:param username:用户名
:param password:密码
:return: json
"""
User = currentUser()
if User[0] == False or User[1]['username'] != 'admin':
rtn = {'code': -1}
return falseReturn(rtn, '没有权限，访问被拒绝')
username = request.form.get('username')
password = request.form.get('password')
usr = Users.query.filter_by(username=username).first()
if usr is not None:
return falseReturn(-2, '用户名已被注册')
user = Users(username=username,
password=Users.set_password(username, password))
result = Users.add(username, user)
if user.id is not None:
returnUser = {
'id': user.id,
'username': user.username,
'login_time': user.login_time
}
return jsonify(trueReturn(returnUser, "用户注册成功"))
else:
return jsonify(falseReturn('', '用户注册失败'))
def pay():
'''
车辆支付过路费
:param: car:下道车辆
:return:json
'''
User = currentUser()
if User[0] == False:
return falseReturn(User[1])
car = request.form.get("car")
if car is None:
rtn = {
'code': -1,
}
return falseReturn(rtn, "参数不完整")
toll = Toll.query.filter_by(car=car, status=0).first()
rtn = {}
if toll is None:
rtn = {
'code': -4,
}
return falseReturn(rtn, "车辆没有上道记录")
if toll.fee is not None:
toll.status = 1
db.session.add(toll)
db.session.commit()
return trueReturn("success", "支付成功")
rtn = {
'code': -3,
}
return falseReturn(rtn, "车辆没有下道记录")
def godown():
'''
车辆离开高速
:param car:车牌号
:param pos:下道口
:return:json
'''
User = currentUser()
if User[0] == False:
return falseReturn(User[1])
car = request.form.get("car")
pos = request.form.get("pos")
if car is None or pos is None:
rtn = {
'code': -1,
}
return falseReturn(rtn, "参数不完整")
toll = Toll.query.filter_by(car=car, status=0).first()
rtn = {}
if toll is None:
rtn = {
'code': -4,
}
return falseReturn(rtn, "车辆没有上道记录")
if toll.downp is not None:
if toll.status is not None:
if toll.status == 1:
rtn = {
'code': 3,
'fee': toll.fee,
'upp': toll.upp,
'upt': toll.upt.strftime("%Y-%m-%d %H:%M:%S"),
'id': toll.id
}
return falseReturn(rtn, "车辆已下道且完成付费")
rtn = {
'code': 2,
'fee': toll.fee,
'upp': toll.upp,
'upt': toll.upt.strftime("%Y-%m-%d %H:%M:%S"),
'id': toll.id
}
return falseReturn(rtn, "车辆当前已在下道口但未付费")
toll.downp = pos
toll.downt = datetime.datetime.now()
toll.downu = User[1]['id']
toll.fee = calmoney(toll.upp, pos)
db.session.add(toll)
db.session.commit()
rtn = {
'code': 1,
'fee': toll.fee,
'upp': toll.upp,
'upt': toll.upt.strftime("%Y-%m-%d %H:%M:%S"),
'id': toll.id
}
return trueReturn(rtn)
def get():
"""
获取用户信息
:return: json
"""
User = currentUser()
if User[0] == False or User[1]['username'] != 'admin':
rtn = {'code': -1}
return falseReturn(rtn, '没有权限，访问被拒绝')
usrs = db.session.query(Users).all()
returnUser = []
for usr in usrs:
urd = {'id': usr.id, 'name': usr.username}
returnUser.append(urd)
return trueReturn(returnUser, "请求成功")
def deleteu():
"""
删除用户
:param:username:用户名
:return:json
"""
User = currentUser()
if User[0] == False or User[1]['username'] != 'admin':
rtn = {'code': -1}
return falseReturn(rtn, '没有权限，访问被拒绝')
username = request.form.get('username')
if username == 'admin':
return falseReturn(-1, '不能删除自己')
usr = Users.query.filter_by(username=username).first()
if usr is None:
return falseReturn(-2, '找不到用户')
db.session.delete(usr)
db.session.commit()
return trueReturn("success")
def goup():
'''
车辆进入高速
:param car:车牌号
:param pos:上道口
:return:json
'''
User = currentUser()
if User[0] == False:
return falseReturn(User[1])
car = request.form.get("car")
pos = request.form.get("pos")
if car is None or pos is None:
rtn = {
'code': -1,
}
return falseReturn(rtn, "参数不完整")
checkhistory = Toll.query.filter_by(car=car, status=0).first()
rtn = {}
if checkhistory is not None:
rtn = {
'code': -5,
'time': checkhistory.upt.strftime("%Y-%m-%d %H:%M:%S"),
'pos': checkhistory.upp,
'id': checkhistory.id
}
return falseReturn(rtn, "车辆有未驶出的记录")
column = Toll(car=car,
upu=[User[1]['id']],
upt=datetime.datetime.now(),
upp=pos,
status=0)
db.session.add(column)
db.session.commit()
return trueReturn({"id": column.id})
def uppic():
'''
上传照片证据
:param: id:goup或godown接口中返回的id
:param: type:指明是上道还是下道，整数类型，0为上道照片，1为下道照片
:param: file:照片 png，jepg，jpg，gif 其中的一种格式
:return:json
'''
User = currentUser()
if User[0] == False:
return falseReturn(User[1])
id = request.form.get("id")
type = int(request.form.get("type"))
f = request.files["file"]
if id is None or type is None or f is None:
rtn = {
'code': -1,
}
return falseReturn(rtn, "参数不完整")
file = ['png', 'jpg', 'gif', 'jpeg']
ffilename = f.filename
hou = f.filename.split('.')[1].lower()
if hou not in file:
return falseReturn({'code': -1}, '不合法的图片格式')
fpath = os.path.join('app/upload', ffilename)
f.save(fpath)
nfilename = filehash(hashlib.md5, fpath) + filehash(hashlib.sha1,
fpath) + "." + hou
nfilepath = os.path.join('app/pic', nfilename)
check = os.path.exists(nfilepath)
if check:
os.unlink(fpath)
return falseReturn({'code': -2}, '图片已经存在')
data = Toll.query.filter_by(id=id).first()
if type == 0:
if data.uppic is not None:
if data.uppic == nfilename:
os.unlink(fpath)
return falseReturn({'code': -2}, '重复上传')
else:
os.unlink(os.path.join('app/pic', data.uppic))
data.uppic = nfilename
shutil.copy(fpath, nfilepath)
os.unlink(fpath)
db.session.add(data)
db.session.commit()
return trueReturn({'code': 1})
else:
if data.downpic is not None:
if data.downpic == nfilename:
os.unlink(fpath)
return falseReturn({'code': -2}, '重复上传')
else:
os.unlink(os.path.join('app/pic', data.downpic))
data.downpic = nfilename
shutil.copy(fpath, nfilepath)
os.unlink(fpath)
db.session.add(data)
db.session.commit()
return trueReturn({'code': 1})
def test():
User = currentUser()
if User[0]:
return trueReturn(User[1])
else:
return falseReturn(User[1])
