У меня очень странная ошибка с SQLAlchemy. Я удалил как можно больше своего кода, чтобы сузить проблему, и удаление большего количества кода приведет к исчезновению ошибки. Мне удалось воспроизвести проблему на другом ПК с новой установкой SQLAlchemy (на Python 2.7).
Если я сделаю какой-либо вариант добавления чего-то вроде этого:
python = Application(name='Python')
python.versions.append(ApplicationVersion(version=27))
session.add(python)
session.commit()
#or
python = Application(name='Python')
session.add(python)
session.commit()
python.versions.append(ApplicationVersion(version=27))
session.commit()
Я получаю эту ошибку (если я ничего не удаляю из кода):
sqlalchemy.exc.IntegrityError: (sqlite3.IntegrityError) NOT NULL constraint failed: ApplicationVersion.application_id
[SQL: INSERT INTO "ApplicationVersion" (application_id, version_int) VALUES (?, ?)]
[parameters: (None, 27)]
Однако, как я уже упоминал, он отлично работает, если я что-нибудь удалю. Например, удалив строку документации этой функции ниже, она будет правильно назначать application_id
и работать должным образом.
@contextmanager
def Session():
"""Setup session to allow for usage with a context manager."""
session = _Session()
yield session
session.close()
Я буквально понятия не имею, что происходит. В качестве отказа от ответственности, другой ПК, который я тестировал, находится в той же рабочей сети, но, поскольку я тестирую с sqlite
, я не могу представить, что это что-то сетевое.
Вот код для воспроизведения ошибки (это один файл, объединенный из нескольких):
######### CONNECT.PY #######
import os
from contextlib import contextmanager
from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base, declared_attr
from sqlalchemy.orm import sessionmaker
class BaseTable(object):
"""General things to apply to each table.
Help: https://docs.sqlalchemy.org/en/13/orm/extensions/declarative/mixins.html
"""
@declared_attr
def __tablename__(cls):
"""Set the table name to that of the model."""
return cls.__name__
if 'DATABASE_URL' not in os.environ:
os.environ['DATABASE_URL'] = 'sqlite://'
Engine = create_engine(os.environ['DATABASE_URL'])
Base = declarative_base(bind=Engine, cls=BaseTable)
_Session = sessionmaker(bind=Base.metadata.bind)
@contextmanager
def Session():
"""Setup session to allow for usage with a context manager."""
session = _Session()
yield session
session.close()
########## MODELS.PY ###########
import time
import os
from sqlalchemy import Column, Integer, SmallInteger, String, Text
from sqlalchemy import ForeignKey, UniqueConstraint, Table, event
from sqlalchemy.ext.hybrid import hybrid_property
from sqlalchemy.orm import backref, relationship, validates
class Category(Base):
row_id = Column(Integer, primary_key=True)
name = Column(String(64), nullable=False)
parent_id = Column(Integer, ForeignKey('Category.row_id'), nullable=True)
parent = relationship('Category', foreign_keys=parent_id, remote_side=row_id)
children = relationship('Category')
__table_args__ = (
UniqueConstraint('name', 'parent_id', name='unique_name_parent'),
)
@hybrid_property
def fullname(self):
parent = self.parent
visited = set()
chain = [self.name]
while parent:
if parent in visited:
break
visited.add(parent)
chain.append(parent.name)
parent = parent.parent
return '.'.join(chain[::-1])
def __init__(self, name, collection, parent=None, creator=None, **kwargs):
super(Category, self).__init__(name=name, collection=collection, parent=parent, creator=creator, **kwargs)
def __repr__(self):
return '<{cls} "{fullname}">'.format(
cls=self.__class__.__name__,
fullname=self.fullname,
)
class Application(Base):
row_id = Column(Integer, primary_key=True)
name = Column(String(16), nullable=False)
versions = relationship('ApplicationVersion', order_by='ApplicationVersion.version_int')
class ApplicationVersion(Base):
row_id = Column(Integer, primary_key=True)
application_id = Column(Integer, ForeignKey('Application.row_id'), nullable=False)
version_int = Column(Integer, nullable=False)
application = relationship('Application', foreign_keys=application_id)
__table_args__ = (
UniqueConstraint('application_id', 'version_int', name='unique_application_version'),
)
def __init__(self, version, application=None, **kwargs):
super(ApplicationVersion, self).__init__(application=application, version_int=version, **kwargs)
def __repr__(self):
return '<{cls} "{application} {version}">'.format(
cls=self.__class__.__name__,
application=self.application.name,
version=self.version_int,
)
def __eq__(self, num):
return self.version_int == num
def __neq__(self, num):
return self.version_int != num
######## TEST.PY ########
Base.metadata.create_all()
if __name__ == '__main__':
with Session() as session:
# Setup programs and versions
python = Application(name='Python')
python.versions.append(ApplicationVersion(version=27))
session.add(python)
session.commit()
print python.versions
Это различные действия, которые остановят ошибку:
BaseTable
или Session
if 'DATABASE_URL' not in os.environ:
create_engine(os.environ['DATABASE_URL'])
на
create_engine('sqlite://')
Category
Category
fullname
, __init__
или __repr__
из Category
__init__
, __repr__
, __eq__
или __neq__
из
ApplicationVersion
Любая помощь будет оценена по достоинству, так как это сводит меня с ума. Я могу обойти проблему, используя session.add(ApplicationVersion(python, 27))
, но я хотел бы знать, что на самом деле здесь происходит, поскольку я никогда раньше не видел, чтобы Python вел себя так.
Я обнаружил, что проблема связана с пользовательским конструктором, который вы определили в ApplicationVersion
:
def __init__(self, version, application=None, **kwargs):
super(ApplicationVersion, self).__init__(application=application, version_int=version, **kwargs)
В частности, вы разрешаете значение по умолчанию None
для ApplicationVersion.application
. Я не уверен, какое значение это служит вам, поскольку конструктор по умолчанию не требует, чтобы вы передавали явное значение для любого из полей модели, и поэтому, если оно не предоставлено, оно будет None
при доступе в любом случае.
Затем в этой строке в вашем тесте:
python.versions.append(ApplicationVersion(version=27))
... вы явно создаете объект ApplicationVersion с application=None
из-за конструктора, но в то же время добавляете его в коллекцию python.versions
. Способ, которым эти отношения разрешают значения внешнего ключа, кажется непоследовательным, поэтому иногда он пытается сбросить с application_id=1
, который является pk нового объекта Application
, а иногда он пытается сбросить с application_id=None
, как диктует конструктор. Но application_id
нельзя обнулить:
application_id = Column(Integer, ForeignKey('Application.row_id'), nullable=False)
... и тогда вы получите IntegrityError
.
SQLAlchemy должен делать некоторое различие между явной установкой атрибута отношения в None
и тем, что он вообще никогда не устанавливался, потому что, если вы перестанете устанавливать application
в None
в своем конструкторе, проблема исчезнет:
def __init__(self, version, **kwargs):
super(ApplicationVersion, self).__init__(version_int=version, **kwargs)
Я смог сократить ваш пример до этого общего примера (извините, Python 3, поэтому вам нужно настроить вызовы print
):
from sqlalchemy import create_engine, Column, Integer
from sqlalchemy import ForeignKey
from sqlalchemy.exc import IntegrityError
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import relationship, sessionmaker
engine = create_engine('sqlite://')
Base = declarative_base()
Session = sessionmaker(bind=engine)
class Parent(Base):
__tablename__ = 'parent'
id = Column(Integer, primary_key=True)
children = relationship('Child')
class Child(Base):
__tablename__ = 'child'
id = Column(Integer, primary_key=True)
num = Column(Integer)
parent_id = Column(Integer, ForeignKey('parent.id'), nullable=False)
parent = relationship('Parent')
def __init__(self, parent=None, **kwargs):
super(Child, self).__init__(parent=parent, **kwargs)
if __name__ == '__main__':
Base.metadata.create_all(engine)
error_cnt = 0
success_cnt = 0
for _ in range(20):
s = Session()
try:
parent = Parent()
parent.children.append(Child())
s.add(parent)
s.commit()
except IntegrityError:
error_cnt += 1
else:
success_cnt += 1
finally:
s.close()
print('errors', error_cnt)
print('successes', success_cnt)
Когда вы запустите это, вы должны получить случайные подсчеты успехов и ошибок. Затем удалите метод Child.__init__()
, и он будет работать постоянно.
Ах, большое спасибо, что нашли время, чтобы проверить это, это очень неясно, но имеет общий смысл, ха-ха (интересно, это ошибка или просто ограничение). Я предполагаю, что при создании модели мне придется решить, является ли добавление или создание более питоническим, вместо того, чтобы пытаться учитывать и то, и другое.