Циркулярный импорт модели SQLAlchemy
У меня две модели в том же модуле с именем models
. Они являются отношениями 1-1 и настроены для SQLAlchemy docs.
Vehicle.py
from models.AssetSetting import AssetSetting
class Vehicle(Base):
__tablename__ = 'vehicles'
vehicle_id = Column(Integer, primary_key=True)
...
settings = relationship('AssetSetting', backref=backref('asset_settings'))
AssetSetting.py
from models.Vehicle import Vehicle
class AssetSetting(Base):
__tablename__ = 'asset_settings'
asset_alert_setting_id = Column(Integer, primary_key=True, autoincrement=True)
...
vehicle = relationship('vehicles', foreign_keys=Column(ForeignKey('vehicles.vehicle_id')))
Если я использую строковое построение отношений (т.е. ForeignKey('vehicles.vehicle_id')
), я получаю ошибку:
sqlalchemy.exc.InvalidRequestError:
When initializing mapper Mapper|AssetSetting|asset_settings, expression 'vehicles' failed to locate a name ("name 'vehicles' is not defined").
If this is a class name, consider adding this relationship() to the <class 'models.AssetSetting.AssetSetting'> class after both dependent classes have been defined.
Если я использую сопоставление классов, я получаю классическую циклическую ошибку импорта:
Traceback (most recent call last):
File "tracking_data_runner.py", line 7, in <module>
from models.Tracker import Tracker
File "/.../models/Tracker.py", line 5, in <module>
from models.Vehicle import Vehicle
File "/.../models/Vehicle.py", line 13, in <module>
from models.Tracker import Tracker
ImportError: cannot import name 'Tracker'
Я считаю, что могу исправить эту проблему, поместив файлы в один и тот же пакет, но предпочел бы, чтобы они были разделены. Мысли?
Ответы
Ответ 1
Я обнаружил, что моя проблема была в два раза:
- Я неправильно ссылался на
Vehicles
в моих отношениях. Это должно быть relationship('Vehicle'
not relationship('vehicles'
- По-видимому, неверно объявлять FK внутри отношения, как это было в AssetSettings.py (
foreign_keys=Column(ForeignKey('vehicles.vehicle_id'))
). Я должен был объявить FK, а затем передать его в отношения.
Теперь мои конфигурации выглядят следующим образом:
Vehicle.py
class Vehicle(Base, IDiagnostable, IUsage, ITrackable):
__tablename__ = 'vehicles'
vehicle_id = Column(Integer, primary_key=True)_id = Column(Integer)
settings = relationship('AssetSetting', backref=backref('asset_settings'))
AssetSetting.py
class AssetSetting(Base):
__tablename__ = 'asset_settings'
asset_alert_setting_id = Column(Integer, primary_key=True, autoincrement=True)
vehicle_id = Column(ForeignKey('vehicles.vehicle_id'))
vehicle = relationship('Vehicle', foreign_keys=vehicle_id)
Ответ 2
Чтобы избежать циклических ошибок импорта, вы должны использовать построение строковых связей, но обе ваши модели должны использовать тот же Base
- тот же экземпляр declarative_base
. Создайте экземпляр Base
один раз и используйте его при инициализации как Vehicle
, так и AssetSetting
.
Или вы можете явно сопоставить имена и классы таблиц, чтобы помочь сопоставлять ваши модели:
Base = declarative_base(class_registry={"vehicles": Vehicle, "asset_settings": AssetSetting})
Ответ 3
Ваш __tablename__
ссылается на vehicles
, но ваш внешний ключ ссылается на vehicle.vehicle_id