peewee外鍵錯誤

Question

我正在嘗試使用peewee為現有數據庫創建一個簡單模型。

本文描述了我正在使用的數據庫。

我已經用pwiz生成了一個可以正常工作的模型，但是它並沒有為我創建啟用聯接的外鍵。 因此，我對模型進行了編輯，如下所示：

from peewee import *

database = MySQLDatabase('enron', **{'password': '...', 'user': 'root'})


class UnknownField(object):
    pass


class BaseModel(Model):
    class Meta:
        database = database


class Employee(BaseModel):
    email = CharField(db_column='Email_id', unique=True)
    eid = PrimaryKeyField()    
    class Meta:
        db_table = 'employeelist'


class Message(BaseModel):
    mid = PrimaryKeyField()
    sender = ForeignKeyField(Employee,
                             related_name='messages',
                             to_field='email')  # was CharField() before my edit

    class Meta:
        db_table = 'message'

但是，當我嘗試運行時：

for message in Message.select():
    print message.mid

我收到一個錯誤：

Traceback (most recent call last):
  File "DBModelEnron.py", line 62, in <module>
    for message in Message.select():
  File "/usr/local/lib/python2.7/dist-packages/peewee.py", line 2514, in __iter__
    return iter(self.execute())
  File "/usr/local/lib/python2.7/dist-packages/peewee.py", line 2507, in execute
    self._qr = ResultWrapper(model_class, self._execute(), query_meta)
  File "/usr/local/lib/python2.7/dist-packages/peewee.py", line 2203, in _execute
    return self.database.execute_sql(sql, params, self.require_commit)
  File "/usr/local/lib/python2.7/dist-packages/peewee.py", line 2833, in execute_sql
    self.commit()
  File "/usr/local/lib/python2.7/dist-packages/peewee.py", line 2688, in __exit__
    reraise(new_type, new_type(*exc_value.args), traceback)
  File "/usr/local/lib/python2.7/dist-packages/peewee.py", line 2825, in execute_sql
    cursor.execute(sql, params or ())
  File "/usr/lib/python2.7/dist-packages/MySQLdb/cursors.py", line 174, in execute
    self.errorhandler(self, exc, value)
  File "/usr/lib/python2.7/dist-packages/MySQLdb/connections.py", line 36, in defaulterrorhandler
    raise errorclass, errorvalue
peewee.OperationalError: (1054, "Unknown column 't1.sender_id' in 'field list'")

我在這里發現了類似的問題，但是定義了正確的PrimaryKey。

Answer 1

只需在您的外鍵上添加db_column='' ：

sender = ForeignKeyField(Employee,
                         db_column='sender',  # Added this.
                         related_name='messages',
                         to_field='email')  # was CharField() before my edit