def update_user(id, data): user = get_session().query(User).filter(User.id == id).first() if not user: return user.save() get_session().flush() return
def update_role(id, data): role = get_session().query(Role).filter(Role.id == id).first() if not role: return role.save() get_session().flush() return
def http_error_handler(exception): get_session().remove() response = None logger.exception(exception) if isinstance(exception, APIException): response = exception elif isinstance(exception, HTTPException): response = HttpApiException(exception) else: response = InternalServerError(str(exception)) return response.get_response()
def get_user_list_by_page(query, page=1, page_size=10): query = (get_session().query(User).filter( User.account.like("%{}%".format(query["q"])) | User.name.like("%{}%".format(query["q"])))) total = query.count() result = query.paginate(page, page_size).all() return result, {"page": page, "page_size": page_size, "total": total}
def delete_role(id): user_role_count = (get_session().query(UserRole).filter( UserRole.role_id == id).count()) if user_role_count > 0: raise InternalServerError(gettext(u"role don`t delete by using")) role = get_session().query(Role).filter(Role.id == id).first() get_session().delete(role) get_session().flush() return
def create_user(data): user = User( account=data.get("account", ""), password=data.get("password", ""), name=data.get("name", ""), ) get_session().add(user) user_role_list = [] for role_id in data["role_id_list"]: role = get_session().query(Role).filter(Role.id == role_id).first() if not role: continue user_role = UserRole(user_id=user.id, role_id=role_id) user_role_list.append(user_role) get_session().bulk_save_objects(user_role_list) get_session().flush() return
def create_role(data): role = Role(name=data.get("name", ""), ) get_session().add(role) get_session().flush() return
def get_role_by_name(name): role = get_session().query(Role).filter(Role.name == name).first() return role
def get_role(id): role = get_session().query(Role).filter(Role.id == id).first() return role
def get_user_role_list(user_id): result = get_session().query(UserRole).filter( UserRole.user_id == user_id).all() return result
def delete_user(id): user = get_session().query(User).filter(User.id == id).first() user.is_frozen = True user.save() get_session().flush() return
def get_user_by_account(account): user = get_session().query(User).filter(User.account == account).first() return user
def get_user(id): user = get_session().query(User).filter(User.id == id).first() return user
def close_session(response): get_session().remove()
def commit_session(response): get_session().commit() return response