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

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

В нашей базе данных sales.db классы «Клиент» и «Счет-фактура» сопоставляются с таблицей «Клиент» и «Счет-фактура» с помощью отношений «один ко многим». Мы попробуем удалить объект 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
  • expunge
  • delete
  • delete-orphan
  • refresh-expire

Часто используется опция «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" 
   )

Давайте удалим клиента с именем Гопал Кришна, используя приведенную ниже программу, и посмотрим количество связанных с ним объектов счета-фактуры -

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()

Счетчик теперь равен 0 со следующим SQL, выпущенным вышеуказанным скриптом -

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.id = ?
(2,)
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.custid 
ORDER BY invoices.id (2,)
DELETE FROM invoices 
WHERE invoices.id = ? ((1,), (2,))
DELETE FROM customers 
WHERE customers.id = ? (2,)
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',)
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)
0