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