У меня плавные отношения 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
Фильтрация SQLAlchemy M2M ⇐ Python
Программы на Python
1736866659
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
Подробнее здесь: [url]https://stackoverflow.com/questions/79355424/sqlalchemy-m2m-filtering[/url]
Ответить
1 сообщение
• Страница 1 из 1
Перейти
- Кемерово-IT
- ↳ Javascript
- ↳ C#
- ↳ JAVA
- ↳ Elasticsearch aggregation
- ↳ Python
- ↳ Php
- ↳ Android
- ↳ Html
- ↳ Jquery
- ↳ C++
- ↳ IOS
- ↳ CSS
- ↳ Excel
- ↳ Linux
- ↳ Apache
- ↳ MySql
- Детский мир
- Для души
- ↳ Музыкальные инструменты даром
- ↳ Печатная продукция даром
- Внешняя красота и здоровье
- ↳ Одежда и обувь для взрослых даром
- ↳ Товары для здоровья
- ↳ Физкультура и спорт
- Техника - даром!
- ↳ Автомобилистам
- ↳ Компьютерная техника
- ↳ Плиты: газовые и электрические
- ↳ Холодильники
- ↳ Стиральные машины
- ↳ Телевизоры
- ↳ Телефоны, смартфоны, плашеты
- ↳ Швейные машинки
- ↳ Прочая электроника и техника
- ↳ Фототехника
- Ремонт и интерьер
- ↳ Стройматериалы, инструмент
- ↳ Мебель и предметы интерьера даром
- ↳ Cантехника
- Другие темы
- ↳ Разное даром
- ↳ Давай меняться!
- ↳ Отдам\возьму за копеечку
- ↳ Работа и подработка в Кемерове
- ↳ Давай с тобой поговорим...
Мобильная версия