def create_group(group_id=None): name = request.values.get('name') cname = request.values.get('cname') selectors = request.values.get('selectors') if request.method == 'POST': try: group = Group(name, cname, selectors=selectors) db.session.add(group) db.session.commit() except Exception as e: logger.error(e) return jsonify(dict(status=False, desc='添加失败')) else: return jsonify(dict(status=True, desc='添加成功')) else: group = db.session.query(Group).filter(Group.id == group_id).first() if not group: return jsonify(dict(status=False, desc='用户组不存在')) try: group.name = name group.cname = cname group.selectors = selectors db.session.add(group) db.session.commit() except Exception as e: logger.error(e) return jsonify(dict(status=False, desc='更新失败')) else: return jsonify(dict(status=True, desc='更新成功'))
def create_selector(selector_id=None): name = request.values.get('name') cname = request.values.get('cname') kind = request.values.get('kind') if request.method == 'POST': try: selector = Selector(name, cname, kind=kind) db.session.add(selector) db.session.commit() except Exception as e: logger.error(e) return jsonify(dict(status=False, desc='添加失败')) else: return jsonify(dict(status=True, desc='添加成功')) else: selector = db.session.query(Selector).filter( Selector.id == selector_id).first() if not selector: return jsonify(dict(status=False, desc='权限不存在')) try: selector.name = name selector.cname = cname selector.kind = kind db.session.add(selector) db.session.commit() except Exception as e: logger.error(e) return jsonify(dict(status=False, desc='更新失败')) else: return jsonify(dict(status=True, desc='更新成功'))
def get(self): try: groups = db.session.query(Group).all() groups_list = [group._to_dict() for group in groups] except Exception, e: logger.error(e) db.session.rollback() return jsonify({"status": False, "desc": "获取用户组信息失败"})
def delete(self, id): try: db.session.query(User).filter(User.id == id).delete() db.session.commit() except Exception, e: logger.error(e) db.session.rollback() return jsonify({"status": False, "desc": "用户修改失败"})
def get(self): try: selectors = db.session.query(Selector).all() selectors_list = [selector._to_dict() for selector in selectors] except Exception, e: logger.error(e) db.session.rollback() return jsonify({"status": False, "desc": "获取权限信息失败"})
def delete_selector(selector_id): try: db.session.query(Selector).filter(Selector.id == selector_id).delete() db.session.commit() except Exception as e: logger.error(e) return jsonify(dict(status=False, desc='删除失败')) else: return jsonify(dict(status=True, desc='删除成功'))
def delete_group(group_id): try: db.session.query(Group).filter(Group.id == group_id).delete() db.session.commit() except Exception as e: logger.error(e) return jsonify(dict(status=False, desc='删除失败')) else: return jsonify(dict(status=True, desc='删除成功'))
def delete_user(user_id): try: user = db.session.query(User).filter(User.id == user_id).first() user.status = False db.session.commit() except Exception as e: logger.error(e) return jsonify(dict(status=False, desc='删除失败')) else: return jsonify(dict(status=True, desc='删除成功'))
def post(self): try: group_dict = request.get_json() group = Group._from_dict(group_dict) db.session.add(group) db.session.commit() except Exception, e: logger.error(e) db.session.rollback() return jsonify({"status": False, "desc": "用户组创建失败"})
def get(self): try: print 222222, request.cookies users = db.session.query(User).all() users_list = [user._to_dict() for user in users] # client_cookie = request.cookies.values() # print 222222, client_cookie except Exception, e: logger.error(e) db.session.rollback() return jsonify({"status": False, "desc": "获取用户信息失败"})
def post(self): try: selector_dict = request.get_json() selector = Selector._from_dict(selector_dict) db.session.add(selector) db.session.commit() except Exception, e: logger.error(e) db.session.rollback() return jsonify({"status": False, "desc": "权限创建失败"})
def put(self, id): try: group = db.session.query(Group).filter(Group.id == id).first() if not group: raise Exception group_dict = request.get_json() group.name = group_dict.get('name'), group.cname = group_dict.get('cname'), group.selectors = group_dict.get('selectors'), db.session.add(group) db.session.commit() except Exception, e: logger.error(e) db.session.rollback() return jsonify({"status": False, "desc": "用户组修改失败"})
def generate_auth_uuid(self): token = self.gen_auth_token() s_uuid = uuid.uuid1() l_uuid = str(s_uuid).split('-') s_uuid = ''.join(l_uuid) tokenMapping = TokenMapping( uuid=s_uuid, token=token, ) db.session.add(tokenMapping) try: db.session.commit() except Exception as e: db.session.rollback() logger.error(e) return s_uuid
def put(self, id): try: selector = db.session.query(Selector).filter(Selector.id == id).first() if not selector: raise Exception selector_dict = request.get_json() selector.name = selector_dict.get('name'), selector.cname = selector_dict.get('cname'), selector.kind = selector_dict.get('kind'), db.session.add(selector) db.session.commit() except Exception, e: logger.error(e) db.session.rollback() return jsonify({"status": False, "desc": "权限修改失败"})
def put(self, id): try: user = db.session.query(User).filter(User.id == id).first() if not user: raise Exception user_dict = request.get_json() user.name = user_dict.get('name'), user.cname = user_dict.get('cname'), user.email = user_dict.get('email'), user.mobile = user_dict.get('mobile'), user.department = user_dict.get('department'), user.group_ids = user_dict.get('group_ids'), user.status = user_dict.get('status') db.session.add(user) db.session.commit() except Exception, e: logger.error(e) db.session.rollback() return jsonify({"status": False, "desc": "用户修改失败"})
def post(self): try: user_dict = request.get_json() user = db.session.query(User).filter( User.name == user_dict.get('name')).first() if user: return jsonify(dict(status=False, desc='账号已存在')) user = User._from_dict(user_dict) db.session.add(user) db.session.commit() # 设置密码及scan_key user.password = user.gen_password_hash(user_dict.get('password')) user.scan_key = user.generate_auth_uuid() db.session.add(user) db.session.commit() except Exception, e: logger.error(e) db.session.rollback() return jsonify({"status": False, "desc": "用户创建失败"})
def post(self): try: auth_dict = request.get_json() if not auth_dict: raise Exception username = auth_dict.get('username') password = auth_dict.get('password') new_password = auth_dict.get('new_password') user = db.session.query(User).filter(User.name == username, User.status == True).first() if not user: return jsonify({"status": False, "desc": "用户名或密码错误"}) verify_res = user.check_password_hash(password) if not verify_res: return jsonify({"status": False, "desc": "用户名或密码错误"}) user.password = user.gen_password_hash(new_password) db.session.add(user) db.session.commit() except Exception, e: logger.error(e) db.session.rollback() return jsonify({"status": False, "desc": "密码修改失败"})
def create_user(user_id=None): name = request.values.get('name') cname = request.values.get('cname') email = request.values.get('email') mobile = request.values.get('mobile') # company = request.values.get('company') password = request.values.get('password') repassword = request.values.get('repassword') department = request.values.get('department') # role_name = request.values.get('user_role') status = True if request.values.get('user_status') else False groups = request.values.get('groups') if request.method == 'POST': user = db.session.query(User).filter(User.name == name).first() if user: return jsonify(dict(status=False, desc='账号已存在')) if password != repassword: return jsonify(dict(status=False, desc='两次输入密码不一致')) try: user = User(name, cname, email, mobile, department, group_ids=groups) db.session.add(user) db.session.commit() # 设置密码及scan_key user.password = user.gen_password_hash(password) user.scan_key = user.generate_auth_uuid() db.session.add(user) db.session.commit() except Exception as e: logger.error(e) return jsonify(dict(status=False, desc='添加失败')) else: return jsonify(dict(status=True, desc='添加成功')) else: user = db.session.query(User).filter(User.id == user_id).first() if not user: return jsonify(dict(status=False, desc='账号不存在')) try: # user.name = name user.cname = cname user.email = email user.mobile = mobile # user.company = company user.department = department # role = db.session.query(Role).filter(Role.cname == role_name).first() # user.rid = role.id user.status = status user.group_ids = groups db.session.add(user) db.session.commit() except Exception as e: logger.error(e) return jsonify(dict(status=False, desc='更新失败')) else: return jsonify(dict(status=True, desc='更新成功'))