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='更新成功'))
