Python current_user.name方法代码示例

# 需要导入模块: from flask_login import current_user [as 别名]
# 或者: from flask_login.current_user import name [as 别名]
def edit_profile_admin(id):
    user = User.query.get_or_404(id)
    form = EditProfileAdminForm(user=user)
    if form.validate_on_submit():
        user.email = form.email.data
        user.username = form.username.data
        user.confirmed = form.confirmed.data
        user.role = Role.query.get(form.role.data)
        user.name = form.name.data
        user.location = form.location.data
        user.about_me = form.about_me.data
        db.session.add(user)
        flash('The profile has been updated.')
        return redirect(url_for('.user', username=user.username))
    form.email.data = user.email
    form.username.data = user.username
    form.confirmed.data = user.confirmed
    form.role.data = user.role_id
    form.name.data = user.name
    form.location.data = user.location
    form.about_me.data = user.about_me
    return render_template('edit_profile.html', form=form, user=user) 

# 需要导入模块: from flask_login import current_user [as 别名]
# 或者: from flask_login.current_user import name [as 别名]
def dev_resource_cancel(id):
    """
    Dev users cancelled the application
    :param id:
    :return:
    """
    resource = Dbapply.query.get(id)
    resource.status = 2
    resource.finish_time = datetime.now()

    db.session.add(resource)
    db.session.commit()

    if current_app.config['MAIL_ON_OFF'] == 'ON':
        auditor = User.query.filter(User.name == resource.audit_name).first()
        mail_content = "<p>Proposer：" + resource.dev_name + "</p>" + "<p>Db instance's name：" + resource.db_name + \
                       "</p>" + "<p>Dev has cancelled the application.</p>"
        send_mail.delay('【inception_mysql】Db instance application cancelled', mail_content, auditor.email)

    return redirect(url_for('.dev_resource_status')) 

# 需要导入模块: from flask_login import current_user [as 别名]
# 或者: from flask_login.current_user import name [as 别名]
def audit_resource_alloc(id):
    """
    Alloc db instances to dev.
    :param id:
    :return:
    """
    resource = Dbapply.query.get(id)
    user = User.query.filter(User.name == resource.dev_name).first()
    db_config = Dbconfig.query.filter(Dbconfig.name == resource.db_name).first()
    user.dbs.append(db_config)
    resource.finish_time = datetime.now()
    resource.status = 0

    db.session.add(resource)
    db.session.commit()

    if current_app.config['MAIL_ON_OFF'] == 'ON':
        dev = User.query.filter(User.name == resource.dev_name).first()
        mail_content = "<p>Proposer：" + resource.dev_name + "</p>" + "<p>Db instance's name：" + resource.db_name + \
                       "</p>" + "<p>Your db instance request has been passed.</p>"
        send_mail.delay('【inception_mysql】Db instance request passed', mail_content, dev.email)

    return redirect(url_for('.audit_resource_pending')) 

# 需要导入模块: from flask_login import current_user [as 别名]
# 或者: from flask_login.current_user import name [as 别名]
def audit_resource_cancel(id):
    """
    Cancelled the application from dev.
    :param id:
    :return:
    """
    resource = Dbapply.query.get(id)
    resource.status = 3
    resource.finish_time = datetime.now()

    db.session.add(resource)
    db.session.commit()

    if current_app.config['MAIL_ON_OFF'] == 'ON':
        dev = User.query.filter(User.name == resource.dev_name).first()
        mail_content = "<p>Proposer：" + resource.dev_name + "</p>" + "<p>Db instance's name：" + resource.db_name + \
                       "</p>" + "<p>Your db instance request is disagreed.</p>"
        send_mail.delay('【inception_mysql】Db instance request failure', mail_content, dev.email)

    return redirect(url_for('.audit_resource_dealt')) 

# 需要导入模块: from flask_login import current_user [as 别名]
# 或者: from flask_login.current_user import name [as 别名]
def audit_work_cancel(id):
    """
    Cancel the work order by auditor.
    :param id:
    :return:
    """
    work = Work.query.get(id)
    work.status = 6
    work.finish_time = datetime.now()
    db.session.add(work)
    db.session.commit()

    if current_app.config['MAIL_ON_OFF'] == 'ON':
        dev = User.query.filter(User.name == work.dev_name).first()
        mail_content = "<p>Work Sheet：" + work.name + " is cancelled by auditor. Please contact with your auditor.</p>"
        send_mail.delay('【inception_mysql】Work Sheet Cancelled', mail_content, dev.email)

    return redirect(url_for('.audit_work_dealt')) 

# 需要导入模块: from flask_login import current_user [as 别名]
# 或者: from flask_login.current_user import name [as 别名]
def audit_work_reject(id):
    """
    Reject the work order by auditor.
    :param id:
    :return:
    """
    work = Work.query.get(id)
    work.status = 7
    work.finish_time = datetime.now()
    db.session.add(work)
    db.session.commit()

    if current_app.config['MAIL_ON_OFF'] == 'ON':
        dev = User.query.filter(User.name == work.dev_name).first()
        mail_content = "<p>Work Sheet：" + work.name + " is rejected by auditor. Please modify it.</p>"
        send_mail.delay('【inception_mysql】Work Sheet Rejected', mail_content, dev.email)

    return redirect(url_for('.audit_work_dealt')) 

# 需要导入模块: from flask_login import current_user [as 别名]
# 或者: from flask_login.current_user import name [as 别名]
def audit_work_execute():
    data = request.form
    id = int(data['workflowid'])
    work = Work.query.filter(Work.id == id).first()

    if work.timer is None:
        work.task_id = str(id)
        work.status = 3
        work.man_review_time = datetime.now()

        # 执行之前重新split并check一遍，更新SHA1缓存；因为如果在执行中，其他进程去做这一步操作的话，会导致inception core dump挂掉
        split_review_result = sql_auto_review(work.sql_content, work.db_name, is_split='yes')
        work.auto_review = json.dumps(split_review_result)

        db.session.add(work)
        db.session.commit()

        async_result = execute_task.apply_async(args=[id], task_id=str(id))

        if current_app.config['MAIL_ON_OFF'] == 'ON':
            dev = User.query.filter(User.name == work.dev_name).first()
            mail_content = "<p>Work Sheet：" + work.name + " is executing. </p>" + "<p>Please view it</p>"
            send_mail.delay('【inception_mysql】Work Sheet Inform', mail_content, dev.email)

    return jsonify({}), 202, {'Location': url_for('.audit_work_detail', id=id)} 

# 需要导入模块: from flask_login import current_user [as 别名]
# 或者: from flask_login.current_user import name [as 别名]
def test_create_user(db, client):
    role = UserRole.security_team
    resp = client.post(url_for('tracker.create_user'), follow_redirects=True,
                       data=dict(username=USERNAME, password=PASSWORD,
                                 email=EMAIL, active=True, role=role.name))
    assert resp.status_code == 200

    resp = client.post(url_for('tracker.logout'), follow_redirects=True)
    assert_not_logged_in(resp)

    resp = client.post(url_for('tracker.login'), follow_redirects=True,
                       data=dict(username=USERNAME, password=PASSWORD))
    assert_logged_in(resp)
    assert USERNAME == current_user.name
    assert EMAIL == current_user.email
    assert role == current_user.role 

# 需要导入模块: from flask_login import current_user [as 别名]
# 或者: from flask_login.current_user import name [as 别名]
def test_edit_user(db, client):
    new_password = random_string()
    new_email = '{}foo'.format(EMAIL)
    new_role = UserRole.security_team
    resp = client.post(url_for('tracker.edit_user', username=USERNAME), follow_redirects=True,
                       data=dict(username=USERNAME, email=new_email, password=new_password,
                       role=new_role.name, active=True))
    assert resp.status_code == 200

    resp = client.post(url_for('tracker.logout'), follow_redirects=True)
    assert_not_logged_in(resp)

    resp = client.post(url_for('tracker.login'), follow_redirects=True,
                       data={'username': USERNAME, 'password': new_password})
    assert_logged_in(resp)
    assert USERNAME == current_user.name
    assert new_email == current_user.email
    assert new_role == current_user.role 

# 需要导入模块: from flask_login import current_user [as 别名]
# 或者: from flask_login.current_user import name [as 别名]
def validate(self):
        rv = BaseForm.validate(self)
        if not rv:
            return False

        if current_user.name in self.password.data:
            self.password.errors.append(ERROR_PASSWORD_CONTAINS_USERNAME)
            return False

        if self.password.data != self.password_repeat.data:
            self.password_repeat.errors.append(ERROR_PASSWORD_REPEAT_MISMATCHES)
            return False

        if not compare_digest(current_user.password, hash_password(self.password_current.data, current_user.salt)):
            self.password_current.errors.append(ERROR_PASSWORD_INCORRECT)
            return False

        return True 

# 需要导入模块: from flask_login import current_user [as 别名]
# 或者: from flask_login.current_user import name [as 别名]
def list_user():
    users = User.query.order_by(User.name).all()
    users = sorted(users, key=lambda u: u.name)

    if not current_user.role.is_administrator:
        masked = []
        for user in users:
            guest = Guest()
            guest.name = user.name
            guest.email = user.email
            guest.role = user.role if not user.role.is_administrator else UserRole.security_team
            guest.active = user.active
            if user.active:
                masked.append(guest)
        users = masked

    users = sorted(users, key=lambda u: u.role)
    return render_template('admin/user.html',
                           title='User list',
                           users=users) 

# 需要导入模块: from flask_login import current_user [as 别名]
# 或者: from flask_login.current_user import name [as 别名]
def _store_prefs(self, key, val):
        """Store the prefs if persistent storage is enabled"""
        if bui.config['WITH_SQL'] and not bui.config['BUI_DEMO']:
            from ..ext.sql import db
            from ..models import Pref
            pref = Pref.query.filter_by(user=current_user.name, key=key).first()
            if pref:
                if val:
                    pref.value = val
                else:
                    db.session.delete(pref)
            elif val:
                pref = Pref(current_user.name, key, val)
                db.session.add(pref)
            try:
                db.session.commit()
            except:  # pragma: no cover
                db.session.rollback() 

# 需要导入模块: from flask_login import current_user [as 别名]
# 或者: from flask_login.current_user import name [as 别名]
def delete(self):
        """Delete prefs"""
        args = self.parser.parse_args()
        sess = session
        ret = {}
        for key in args.keys():
            temp = args.get(key)
            if temp:
                del sess[key]
                if bui.config['WITH_SQL']:
                    from ..ext.sql import db
                    from ..models import Pref
                    try:
                        Pref.query.filter_by(
                            user=current_user.name,
                            key=key
                        ).delete()
                        db.session.commit()
                    except:  # pragma: no cover
                        db.session.rollback()
            ret[key] = sess.get(key)

        return ret 

# 需要导入模块: from flask_login import current_user [as 别名]
# 或者: from flask_login.current_user import name [as 别名]
def delete(self, task_type, task_id, server=None):
        """Cancel a given task"""
        if task_type not in task_types:
            self.abort(400)
        task_obj, _ = task_types[task_type]
        task = task_obj.AsyncResult(task_id)
        user = task.result.get('user')
        dst_server = task.result.get('server')

        if (current_user.name != user or (dst_server and dst_server != server)) and \
                not current_user.acl.is_admin():
            self.abort(403, 'Unauthorized access')

        # do not remove the task from db yet since we may need to remove
        # some temporary files afterward. The "cleanup_restore" task will take
        # care of this
        task.revoke()
        return '', 201 

# 需要导入模块: from flask_login import current_user [as 别名]
# 或者: from flask_login.current_user import name [as 别名]
def get(self, task_id, server=None):
        """Returns the generated archive"""
        task = load_all_tree.AsyncResult(task_id)
        if task.state != 'SUCCESS':
            if task.state == 'FAILURE':
                self.abort(
                    500,
                    'Unsuccessful task: {}'.format(task.result.get('error'))
                )
            self.abort(400, 'Task not processed yet: {}'.format(task.state))

        user = task.result.get('user')
        dst_server = task.result.get('server')
        resp = task.result.get('tree')

        if current_user.name != user or (dst_server and dst_server != server):
            self.abort(403, 'Unauthorized access')

        task.revoke()

        return resp