SQLAlchemy ORM — удаление связанных объектов

Операцию удаления с одной таблицы легко выполнить. Все, что вам нужно сделать, это удалить объект сопоставленного класса из сеанса и зафиксировать действие. Однако операция удаления для нескольких связанных таблиц немного сложна.

В нашей базе данных sales.db классы Customer и Invoice сопоставляются с таблицей customer и invoice с типом взаимосвязи один-ко-многим. Мы постараемся удалить объект Customer и увидим результат.

В качестве краткой справки ниже приведены определения классов Customer и Invoice —

from sqlalchemy import create_engine, ForeignKey, Column, Integer, String
engine = create_engine('sqlite:///sales.db', echo = True)
from sqlalchemy.ext.declarative import declarative_base
Base = declarative_base()
from sqlalchemy.orm import relationship
class Customer(Base):
   __tablename__ = 'customers'

   id = Column(Integer, primary_key = True)
   name = Column(String)
   address = Column(String)
   email = Column(String)
   
class Invoice(Base):
   __tablename__ = 'invoices'

   id = Column(Integer, primary_key = True)
   custid = Column(Integer, ForeignKey('customers.id'))
   invno = Column(Integer)
   amount = Column(Integer)
   customer = relationship("Customer", back_populates = "invoices")
   
Customer.invoices = relationship("Invoice", order_by = Invoice.id, back_populates = "customer")

Мы настраиваем сеанс и получаем объект Customer, запрашивая его с помощью основного идентификатора, используя следующую программу:

from sqlalchemy.orm import sessionmaker
Session = sessionmaker(bind=engine)
session = Session()
x = session.query(Customer).get(2)

В нашей таблице примеров x.name — это «Гопал Кришна». Давайте удалим этот x из сеанса и посчитаем вхождение этого имени.

session.delete(x)
session.query(Customer).filter_by(name = 'Gopal Krishna').count()

Полученное выражение SQL вернет 0.

SELECT count(*) 
AS count_1
FROM (
   SELECT customers.id 
   AS customers_id, customers.name 
   AS customers_name, customers.address 
   AS customers_address, customers.email 
   AS customers_email
   FROM customers
   WHERE customers.name = ?) 
AS anon_1('Gopal Krishna',) 0

Однако связанные объекты Invoice для x все еще там. Это можно проверить с помощью следующего кода —

session.query(Invoice).filter(Invoice.invno.in_([10,14])).count()

Здесь 10 и 14 — номера счетов, принадлежащих клиенту Гопалу Кришне. Результатом вышеупомянутого запроса является 2, что означает, что связанные объекты не были удалены.

SELECT count(*) 
AS count_1
FROM (
   SELECT invoices.id 
   AS invoices_id, invoices.custid 
   AS invoices_custid, invoices.invno 
   AS invoices_invno, invoices.amount 
   AS invoices_amount
   FROM invoices
   WHERE invoices.invno IN (?, ?)) 
AS anon_1(10, 14) 2

Это потому, что SQLAlchemy не предполагает удаления каскада; мы должны дать команду, чтобы удалить его.

Чтобы изменить поведение, мы настраиваем параметры каскада в отношении User.addresses. Давайте закроем текущую сессию, воспользуемся новой Declarative_base () и повторно объявим класс User, добавив в отношения адресов, включая конфигурацию каскада.

Атрибут cascade в функции отношений представляет собой разделенный запятыми список правил каскада, который определяет, как операции «Сеанс» должны быть «каскадными» от родителя к потомку. По умолчанию это False, что означает «save-update, merge».

Доступны следующие каскады:

• копи-обновление
• сливаться
• вычеркивать
• удалять
• удалить-сирота
• обновить устареть

Часто используемой опцией является «all, delete-orphan», чтобы указать, что связанные объекты должны следовать вместе с родительским объектом во всех случаях, и удаляться при деассоциировании.

Следовательно, повторно объявленный класс Customer показан ниже —

class Customer(Base): 
   __tablename__ = 'customers'
   
   id = Column(Integer, primary_key = True) 
   name = Column(String) 
   address = Column(String) 
   email = Column(String) 
   invoices = relationship(
      "Invoice", 
      order_by = Invoice.id, 
      back_populates = "customer",
      cascade = "all, 
      delete, delete-orphan" 
   )

Давайте удалим Клиента с именем Gopal Krishna, используя приведенную ниже программу, и посмотрим количество связанных объектов Invoice —

from sqlalchemy.orm import sessionmaker
Session = sessionmaker(bind = engine)
session = Session()
x = session.query(Customer).get(2)
session.delete(x)
session.query(Customer).filter_by(name = 'Gopal Krishna').count()
session.query(Invoice).filter(Invoice.invno.in_([10,14])).count()