コード例 #1
0
ファイル: admin.py プロジェクト: smajida/predator-2
def user_create():
    error = None
    form = UserForm(request.form)
    if request.method == 'POST' and form.validate(skip=['threshold']):
        username = request.form.get('username')
        password = werkzeug.generate_password_hash(request.form.get('password'))
        email = request.form.get('email')
        fullname = request.form.get('fullname')
        
        user = User.find_one({'username':username})
        if user:
            error = 'User %s already exists' % username
        else:
            user = User()
            user.update({'username':username, 'password':password, \
                         'email':email, 'fullname':fullname})
            user.validate()
            if user.validation_errors:
                error = str([k for k in user.validation_errors]) + \
                ' fields not valid'
            else:
                user.save()
                flash('User %s created.' % username, 'success')
                return redirect(url_for('.users_display'))
    flash(error, 'error')
    return render_template('create_user.html', form=form)
コード例 #2
0
ファイル: admin.py プロジェクト: smajida/predator-2
def user_delete(username):
    if username == 'admin':
        return 'admin cannot be deleted'
    user = User.find_one({'username':username})
    user.delete()
    flash('User %s deleted.' % username, 'success');
    return redirect(url_for('admin.users_display'))
コード例 #3
0
def get_my_conversations():
    username = get_current_user().username
    user = User.objects(username=username).first()
    conversation_ids = user.conversations
    return Conversation\
        .objects(conversation_id__in=conversation_ids)\
        .order_by('-timestamp')\
        .select_related()
コード例 #4
0
ファイル: cookies.py プロジェクト: smajida/predator-2
def make_rememberme_cookie():
    username = session['username'] # search for username key in custom cookie
    token = os.urandom(16).encode('hex')
    cookie = make_cookie(username=username, token=token)
    
    user = User.find_one({'username': username})
    user['last_login'][token] = datetime.datetime.now()
    user.save()

    return cookie
コード例 #5
0
ファイル: admin.py プロジェクト: smajida/predator-2
def user_edit(username):
    threshold = range(99)
    if request.method == 'POST':
        username = request.form.get('username')
      
    if not username:
        return "Invalid action"
    
    user = User.find_one({'username':username})
    if not user:
        return 'User does not exist'
    if session['username'] != 'admin' and username != session['username']:
        return 'You are not allowed to perform that action'
    
    form = UserForm(request.form)
    skip = request.form.get('skip')
    active = 'account'
    
    if request.method == 'POST' and form.validate(skip=skip):
        if request.form.get('password'):
            logging.warn('password received')
            active = 'password'
            if session['username']  != 'admin':
                previous_password = request.form.get('previous_password')
                if previous_password:
                    if werkzeug.check_password_hash(user['password'], previous_password):
                        pass
                else:
                    return render_template('edit_user.html', form=form, user=user, active=active)
            user['password'] = werkzeug.generate_password_hash(request.form.get('password'))
        elif request.form.get('email') and request.form.get('fullname'):
            logging.warn('email and fullname received')
            active = 'account'
            user['email'] = request.form.get('email')
            user['fullname'] = request.form.get('fullname')
        elif request.form.get('copy_labels'):
            logging.warn('labels received')
            active = 'label'
            user['labels'] = eval(request.form.get('copy_labels'))
        elif request.form.get('threshold'):
            logging.warn('settings received')
            active = 'settings'
            user['threshold'] = int(request.form.get('threshold'))

        if user.validation_errors:
            return str([k for k in user.validation_errors]) + ' fields not valid'
        user.save()
        flash('Information updated for user %s.' % username, 'success')
    
    if form.password.errors:
        active = 'password'
        
    return render_template('edit_user.html', form=form, user=user, threshold=threshold, active=active)
コード例 #6
0
ファイル: base.py プロジェクト: smajida/predator-2
def logout():
    me = session.pop('username', None)
    if me:
        response_body = redirect(url_for('base.login'))
        response = app.make_response(response_body)
        
        if request.cookies.get('TK'):
            cookie = unserialize_cookie(request.cookies['TK'])
            user = User.find_one({'username': me})
            if user and user['last_login'].get(cookie['token']):
                user['last_login'].pop(cookie['token'])
                user.save()
            response.delete_cookie('TK')
        return response
    return render_template('dashboard.html')
コード例 #7
0
ファイル: base.py プロジェクト: smajida/predator-2
def login():
    error = None
    remember = False
    cookie_flag = False
    
    next_url = request.args.get('next')
    if next_url:
        response_body = redirect(next_url)
    else:
        response_body = redirect(url_for("base.dashboard"))
    response = app.make_response(response_body)
                    
    if request.method == 'POST':
        username = request.form.get('username')
        password = request.form.get('password')
        remember_me = request.form.get('remember_me')
        if username and password:
            me = User.find_one({'username':username})
            if me:
                hash_password = me.get('password', '')
                if werkzeug.check_password_hash(hash_password, password):
                    session['username'] = me['username'] 
                    if remember_me:
                        remember = True
                else:
                    error = 'Invalid password'
            else:
                error = 'Invalid username'
                logger.warning('Invalid username')
                return render_template('login.html', error=error)
        else:
            error = "username and password fields are required."
            logger.warning("username and password fields are required")
            return render_template('login.html', error=error)

    if not session.get('username'):
        username = check_login_cookies(request.cookies)
        if username:
            cookie_flag = True
            session['username'] = username 

    if session.get('username'):
        if remember or cookie_flag:
            response.set_cookie('TK', make_rememberme_cookie(), 2592000) # 30 days
        logger.info("login successful")
        return response
    else:
        return render_template('login.html', error=error)
コード例 #8
0
ファイル: cop.py プロジェクト: smajida/predator-2
def get_user_prefs():
    """
    """
    label_scale = [None for i in range(100)]
    user = User.find_one({'username': session['username']})
    labels = user['labels']
    threshold = user['threshold']
    
    for label, value in labels.iteritems():
        ran, color = value
        start, end = ran.split('-')
        
        intermediate_values = range(int(start) + 1, int(end) + 1)
        for i in intermediate_values:
            label_scale[i-1] = label
            
    return labels, label_scale, threshold
コード例 #9
0
ファイル: cookies.py プロジェクト: smajida/predator-2
def check_login_cookies(cookie):
    if not cookie.get('TK'):
        return

    username_random = unserialize_cookie(cookie['TK'])
    username = username_random['username']
    token = username_random['token']

    user = User.find_one({'username': username})

    if user['last_login'].get(token):
        if (user['last_login'].get(token) + datetime.timedelta(days=30)) <  datetime.datetime.now():
            return
        user['last_login'].pop(token, None)
        user.save()
        return username
    return
コード例 #10
0
ファイル: admin.py プロジェクト: smajida/predator-2
def users_display():
    users = User.find()
    return render_template('display_users.html', users=users)
コード例 #11
0
def get_current_user_obj():
    username = get_current_user().username
    return User.objects(username=username).first()