Фильтрация SQLAlchemy M2MPython

Программы на Python
Ответить
Anonymous
 Фильтрация SQLAlchemy M2M

Сообщение Anonymous »

У меня плавные отношения M2M:
user_accounts = Table(
"user_accounts",
Base.metadata,
Column("user_id", ForeignKey("user.id", ondelete="CASCADE"), index=True),
Column("account_id", ForeignKey("accounts.id", ondelete="CASCADE"), index=True),
)

class User(Base, TimestampedModelMixin):
__tablename__ = "user"
__allow_unmapped__ = True
accounts: Mapped[list["Account"]] = relationship(secondary=user_accounts)

class Account(Base, TimestampedModelMixin):
__tablename__ = "accounts"
__allow_unmapped__ = True

external_id: Mapped[str] = mapped_column(String, index=True, unique=True)
users: Mapped[list["Users"]] = relationship(secondary=user_accounts, back_populates="accounts")


Я пытаюсь найти все учетные записи, перечисленные в account_ids, но которым не назначен пользователь. Я делаю это с помощью асинхронного механизма:
class AccountQuery:
def __init__(self, db: Annotated[AsyncSession, get_db]):
self._db = db

async def get_accounts_without_user(self, user_id: UUID, accounts_ids: list[str]) -> list[Account]:
stmt = (
select(AdAccount)
.where(
Account.external_id.in_(accounts_ids),
user_accounts.columns.user_id != user_id
)
.join(user_accounts, )
# .options(selectinload(Account.users))
)
result = await self._db.execute(stmt)
return list(result.scalars())

И тесты для этой цели:
@pytest_asyncio.fixture()
async def account(db_session):
account = Account(external_id="123")
db_session.add(account)
await db_session.commit()
return account

@pytest_asyncio.fixture()
async def user(db_session) -> Users:
return await Users.create(db_session, ...)

@pytest.mark.asyncio
async def test_get_accounts_without_user_assigned(account, user, db_session):
query = AccountQuery(db=db_session)
accounts = await query.get_accounts_without_user(user_id=user.id, accounts_ids=["123"])
assert accounts == [account]
await query.assign_user_to_accounts(user_id=user.id, accounts=accounts)
accounts = await query.get_accounts_without_user(user_id=user.id, accounts_ids=["123"])
assert accounts == []

Думаю, проблема в какой-то глупой неправильной настройке соединения, но я застрял в этом коде.
Также для справки: сгенерированный SQL:< /p>
SELECT ...
FROM accounts JOIN user_accounts ON accounts.id = user_accounts.ad_account_id
WHERE accounts.external_id IN (__[POSTCOMPILE_external_id_1]) AND user_accounts.user_id != :user_id_1


Подробнее здесь: https://stackoverflow.com/questions/793 ... -filtering
Ответить

Быстрый ответ

Изменение регистра текста: 
Смайлики
:) :( :oops: :roll: :wink: :muza: :clever: :sorry: :angel: :read: *x)
Ещё смайлики…
   
К этому ответу прикреплено по крайней мере одно вложение.

Если вы не хотите добавлять вложения, оставьте поля пустыми.

Максимально разрешённый размер вложения: 15 МБ.

Вернуться в «Python»