如何檢查數據庫中的現有值並更改它？

Question

我有一個表名叫「代碼」和3列：ID（主），碼號（串）和激活（布爾值，默認值= FALSE）

，我已經在這裏我要檢查值的形式，如果它存在，使其激活== False。

我想： myform.py：

#all imports here 
class CheckForm(Form): 
    some_code = StringField('Code: ',validators=[Required()]) 
    submit = SubmitField('Check it!') 

我views.py：

#all imports here 
@some.route('/sometest', methods=['GET', 'POST'] 
def check_function(): 
    form = CheckForm() 
    code=form.some_code.data 
    check_code = Codes.query.filter_by(code=code).first() 
    if check_code.activated == False: 
     check_code.activated = True 
     db.session.add(check_code) 
     db.session.commit() 
    elif check_code.activated == True: 
     print 'Code already used' 
    else: 
     print 'Code not found') 
return render_template('test.html') 

但我發現了錯誤：

AttributeError: 'NoneType' object has no attribute 'activated' 

我m使用燒瓶和sqlalchemy

Answer 1

只需爲check_code添加一個條件，以防在數據庫中找不到結果。

提示：處理上面的失敗條件和從函數返回時總是更具可讀性。

def check_function(): 
    form = CheckForm() 
    code=form.some_code.data 
    check_code = Codes.query.filter_by(code=code).first() 
    if not check_code: 
     print 'Code not found' 
     return # You don't have to worry about that again 

    if check_code.activated == True: 
     print 'Code already used' 
     return # Second case is already covered 

    # That's the only case left, no need for an `if` condition 
    check_code.activated = True 
    db.session.add(check_code) 
    db.session.commit()