dependencies
prefect.server.database.dependencies
Injected database interface dependencies
Functions
provide_database_interface
Get the current Prefect REST API database interface.
If components of the interface are not set, defaults will be inferred based on the dialect of the connection URL.
inject_db
Decorator that provides a database interface to a function.
The decorated function must take a db kwarg and if a db is passed
when called it will be used instead of creating a new one.
db_injector
Decorator to inject a PrefectDBInterface instance as the first positional argument to the decorated function.
Unlike inject_db, which injects the database connection as a keyword
argument, db_injector adds it explicitly as the first positional
argument. This change enhances type hinting by making the dependency on
PrefectDBInterface explicit in the function signature.
When decorating a coroutine function, the result will continue to pass the iscoroutinefunction() test.
Args:
func: The function or method to decorate.
Returns:
- A wrapped descriptor object which injects the PrefectDBInterface instance
- as the first argument to the function or method. This handles method
- binding transparently.
temporary_database_config
Temporarily override the Prefect REST API database configuration. When the context is closed, the existing database configuration will be restored.
Args:
tmp_database_config: Prefect REST API database configuration to inject.
temporary_query_components
Temporarily override the Prefect REST API database query components. When the context is closed, the existing query components will be restored.
Args:
tmp_queries: Prefect REST API query components to inject.
temporary_orm_config
Temporarily override the Prefect REST API ORM configuration. When the context is closed, the existing orm configuration will be restored.
Args:
tmp_orm_config: Prefect REST API ORM configuration to inject.
temporary_interface_class
Temporarily override the Prefect REST API interface class When the context is closed, the existing interface will be restored.
Args:
tmp_interface_class: Prefect REST API interface class to inject.
temporary_database_interface
Temporarily override the Prefect REST API database interface.
Any interface components that are not explicitly provided will be cleared and inferred from the Prefect REST API database connection string dialect.
When the context is closed, the existing database interface will be restored.
Args:
tmp_database_config: An optional Prefect REST API database configuration to inject.tmp_orm_config: An optional Prefect REST API ORM configuration to inject.tmp_queries: Optional Prefect REST API query components to inject.tmp_interface_class: Optional database interface class to inject
set_database_config
Set Prefect REST API database configuration.
set_query_components
Set Prefect REST API query components.
set_orm_config
Set Prefect REST API orm configuration.
set_interface_class
Set Prefect REST API interface class.