def post(self):
rsp = RspInfo()
userId = self.get_argument('userId', None)
userPwd = self.get_argument('userPwd', None)
if userId and userPwd:
log = LoginDao()
list, total = log.checkLogin(userId, userPwd)
if 1 == total:
info = {
'userId': list[0].get('USER_ID'),
'userName': list[0].get('USER_NAME'),
'userAuth': list[0].get('USER_AUTH')
}
token = Utils.makeToken(info)
info['token'] = token
# 登录信息 token 保存 Redis
MyRedis().set(info['userId'], token)
rsp.setSuccess()
rsp.setObj(info)
else:
rsp.setInfo("登录失败, 请检查用户名或密码是否正确")
else:
pass
self.write(rsp.toDict())
return
def add(self):
"""
添加 project 记录
"""
token = self.get_argument('token', None)
proName = self.get_argument('proName', None)
proType = self.get_argument('proType', None)
proUse = self.get_argument('proUse', None)
proDes = self.get_argument('proDes', None)
userCreator = self.get_current_user()['userId'] or None
rsp = RspInfo()
try:
dao = ProjectDao()
res = dao.insert(proName, proType, proUse, proDes, userCreator)
if res:
rsp.setSuccess()
rsp.setInfo("添加项目成功")
rsp.setObj(token)
except Exception as e:
print('ERROR {}'.format(e))
Utils.log('ERROR {}'.format(e))
rsp.setInfo("添加项目失败")
finally:
del (dao)
return rsp
def add(self):
"""
添加用户记录
"""
token = self.get_argument('token', None)
userId = self.get_argument('userId', None)
userName = self.get_argument('userName', None)
userAuth = self.get_argument('userAuth', None)
userLogin = self.get_argument('userLogin', None)
userCreator = self.get_current_user()['userId'] or None
rsp = RspInfo()
try:
dao = UserDao()
valid = dao.validInsert(userId)
if valid:
rsp.setInfo('该用户名已存在, 添加用户失败')
raise Exception("该用户名已存在, 添加用户失败")
res = dao.insert(userId, userName, userAuth, userLogin,
userCreator)
if res:
rsp.setSuccess()
rsp.setInfo("添加用户成功")
rsp.setObj(token)
except Exception as e:
print('ERROR {}'.format(e))
Utils.log('ERROR {}'.format(e))
rsp.setInfo("添加用户失败")
finally:
del (dao)
return rsp
def post(self):
rsp = RspInfo()
token = self.get_argument('token', None)
userInfo = Utils.verifyToken(token)
rsp.setSuccess()
rsp.setObj(userInfo)
self.write(rsp.toDict())
return
def write_error(self, status_code, **kwargs):
"""
重写错误状态码信息
"""
rsp = RspInfo()
rsp.setObj(status_code)
rsp.setInfo("错误的请求")
self.write(respose_info)
def query(self):
"""
查询任务进度
"""
userId = self.get_current_user()['userId'] or ''
rsp = RspInfo()
try:
dao = ProjectRateDao()
res = dao.query(userId)
rsp.setSuccess()
rsp.setObj(res)
except Exception as e:
print('ERROR {}'.format(e))
Utils.log('ERROR 查询任务进度失败 {}'.format(e))
rsp.setInfo("查询任务进度失败")
finally:
del (dao)
return rsp
