prefect.server.models.saved_searches

Functions for interacting with saved search ORM objects. Intended for internal use by the Prefect REST API.

create_saved_search(db, session, saved_search) async

Upserts a SavedSearch.

If a SavedSearch with the same name exists, all properties will be updated.

Parameters:

Name	Type	Description	Default
session	AsyncSession	a database session	required
saved_search	SavedSearch	a SavedSearch model	required

Returns:

Type	Description
SavedSearch	orm_models.SavedSearch: the newly-created or updated SavedSearch

Source code in src/prefect/server/models/saved_searches.py

@db_injector
async def create_saved_search(
    db: PrefectDBInterface,
    session: AsyncSession,
    saved_search: schemas.core.SavedSearch,
) -> orm_models.SavedSearch:
    """
    Upserts a SavedSearch.

    If a SavedSearch with the same name exists, all properties will be updated.

    Args:
        session (AsyncSession): a database session
        saved_search (schemas.core.SavedSearch): a SavedSearch model

    Returns:
        orm_models.SavedSearch: the newly-created or updated SavedSearch
    """

    insert_stmt = (
        db.insert(orm_models.SavedSearch)
        .values(**saved_search.model_dump_for_orm(exclude_unset=True))
        .on_conflict_do_update(
            index_elements=db.saved_search_unique_upsert_columns,
            set_=saved_search.model_dump_for_orm(include={"filters"}),
        )
    )

    await session.execute(insert_stmt)

    query = (
        sa.select(orm_models.SavedSearch)
        .where(
            orm_models.SavedSearch.name == saved_search.name,
        )
        .execution_options(populate_existing=True)
    )
    result = await session.execute(query)
    model = result.scalar_one()

    return model

delete_saved_search(session, saved_search_id) async

Delete a SavedSearch by id.

Parameters:

Name	Type	Description	Default
session	AsyncSession	A database session	required
saved_search_id	str	a SavedSearch id	required

Returns:

Name	Type	Description
bool	bool	whether or not the SavedSearch was deleted

Source code in src/prefect/server/models/saved_searches.py

async def delete_saved_search(session: AsyncSession, saved_search_id: UUID) -> bool:
    """
    Delete a SavedSearch by id.

    Args:
        session (AsyncSession): A database session
        saved_search_id (str): a SavedSearch id

    Returns:
        bool: whether or not the SavedSearch was deleted
    """

    result = await session.execute(
        delete(orm_models.SavedSearch).where(
            orm_models.SavedSearch.id == saved_search_id
        )
    )
    return result.rowcount > 0

read_saved_search(session, saved_search_id) async

Reads a SavedSearch by id.

Parameters:

Name	Type	Description	Default
session	AsyncSession	A database session	required
saved_search_id	str	a SavedSearch id	required

Returns:

Type	Description
Union[SavedSearch, None]	orm_models.SavedSearch: the SavedSearch

Source code in src/prefect/server/models/saved_searches.py

async def read_saved_search(
    session: AsyncSession, saved_search_id: UUID
) -> Union[orm_models.SavedSearch, None]:
    """
    Reads a SavedSearch by id.

    Args:
        session (AsyncSession): A database session
        saved_search_id (str): a SavedSearch id

    Returns:
        orm_models.SavedSearch: the SavedSearch
    """

    return await session.get(orm_models.SavedSearch, saved_search_id)

read_saved_search_by_name(session, name) async

Reads a SavedSearch by name.

Parameters:

Name	Type	Description	Default
session	AsyncSession	A database session	required
name	str	a SavedSearch name	required

Returns:

Type	Description
Union[SavedSearch, None]	orm_models.SavedSearch: the SavedSearch

Source code in src/prefect/server/models/saved_searches.py

async def read_saved_search_by_name(
    session: AsyncSession, name: str
) -> Union[orm_models.SavedSearch, None]:
    """
    Reads a SavedSearch by name.

    Args:
        session (AsyncSession): A database session
        name (str): a SavedSearch name

    Returns:
        orm_models.SavedSearch: the SavedSearch
    """
    result = await session.execute(
        select(orm_models.SavedSearch)
        .where(orm_models.SavedSearch.name == name)
        .limit(1)
    )
    return result.scalar()

read_saved_searches(session, offset=None, limit=None) async

Read SavedSearches.

Parameters:

Name	Type	Description	Default
session	AsyncSession	A database session	required
offset	int	Query offset	None
limit(int)		Query limit	required

Returns:

Type	Description
Sequence[SavedSearch]	List[orm_models.SavedSearch]: SavedSearches

Source code in src/prefect/server/models/saved_searches.py

async def read_saved_searches(
    session: AsyncSession,
    offset: Optional[int] = None,
    limit: Optional[int] = None,
) -> Sequence[orm_models.SavedSearch]:
    """
    Read SavedSearches.

    Args:
        session (AsyncSession): A database session
        offset (int): Query offset
        limit(int): Query limit

    Returns:
        List[orm_models.SavedSearch]: SavedSearches
    """

    query = select(orm_models.SavedSearch).order_by(orm_models.SavedSearch.name)

    if offset is not None:
        query = query.offset(offset)
    if limit is not None:
        query = query.limit(limit)

    result = await session.execute(query)
    return result.scalars().unique().all()