Почему я получаю NameError для столбца, определенного в модели SQLAlchemy - PullRequest
Новая
       13

Почему я получаю NameError для столбца, определенного в модели SQLAlchemy

0 голосов
/ 08 мая 2018

Я разрабатываю REST API с помощью Flask. Одна из таблиц моделируется следующим образом: 

class AudioSessionModel(db.Model):
    __tablename__ = 'audio_session'
    id = db.Column('audio_session_id', db.Integer, primary_key = True)
    cs_id = db.Column(db.Integer)
    session_id = db.Column(db.Integer)
    facility = db.Column(db.Integer)
    description = db.Column(db.String(400))

    def __init__(self, cs_id, session_id, facility):
        self.cs_id = cs_id
        self.session_id = session_id
        self.facility = facility

Бизнес-логика определяется в классе DAO: 

class AudioSessionDAO(object):
    def update(self, data):
        audio = AudioSessionModel.query.filter(cs_id == data['CSID'], session_id == data['Session'])
        audio.description = data['Desc']
        db.session.commit()
        return audio

Эта upate функция вызывается в моей конечной точке для PUT запроса: 

@api.route('/OperatorAccessment')
class OperatorAssessment(Resource):

    @api.expect(assessment)
    def put(self):
        as_dao = AudioSessionDAO()
        as_dao.update(request.json)

Модель assessment выглядит так: 

assessment = api.model('Operator Assessment', {
    'CSID': fields.Integer(required=True, description='Central Station ID'),
    'Session': fields.Integer(required=True, description='Session ID'),
    'Desc': fields.String(description='Description')
})

Когда я проверяю запрос PUT со следующим json в теле запроса: 

{
   "CSID": 1,  
   "Session": 1, 
   "Desc": "Siren"
 }

Я получил следующую ошибку: 

  File "C:\Users\xxx_app\model\dao.py", line 63, in update
    audio = AudioSessionModel.query.filter(cs_id == data['CSID'], session_id == data['Session'])
NameError: name 'cs_id' is not defined

Очевидно, cs_id определено. Почему я все еще получаю эту ошибку?

1 Ответ

0 голосов
/ 08 мая 2018

Вы должны использовать атрибуты класса, т.е. 

AudioSessionModel.query.filter(
    AudioSessionModel.cs_id == data['CSID'],
    AudioSessionModel.session_id == data['Session'])

или filter_by с аргументами ключевых слов, использующими только =: 

AudioSessionModel.query.filter_by(
    cs_id=data['CSID'],
    session_id=data['Session'])

См. В чем разница между filter и filter_by в SQLAlchemy?

...